feat(lockup): add linearTimelock shape inference

Author: PaulRBerg

envio/lockup/store/entity-stream.ts

@@ -41,7 +41,7 @@ export function createLinear(
 
   const cliff = addCliff(baseStream, params);
   const initial = addInitial(params);
-  const shape = addLinearShape(baseStream, cliff.cliff ?? false);
+  const shape = addLinearShape(baseStream, cliff.cliff ?? false, cliff.cliffTime, params.endTime);
 
   const stream: Entity.Stream = {
     ...baseStream,
@@ -217,11 +217,16 @@ function addInitial(
  * Older versions of Lockup did not have a shape field, but it can be inferred.
  * @see https://github.com/sablier-labs/interfaces/blob/30fffc0/packages/constants/src/stream/shape.ts#L12
  */
-function addLinearShape(stream: Entity.Stream, cliff: boolean): ShapeResult {
+function addLinearShape(
+  stream: Entity.Stream,
+  cliff: boolean,
+  cliffTime: bigint | undefined,
+  endTime: bigint
+): ShapeResult {
   if (stream.shape) {
     return { shape: stream.shape, shapeSource: "Event" };
   }
-  return { shape: inferLinearShape(cliff), shapeSource: "Inferred" };
+  return { shape: inferLinearShape(cliff, cliffTime, endTime), shapeSource: "Inferred" };
 }
 
 function addDynamicShape(stream: Entity.Stream, segments: Segment[]): ShapeResult {

envio/lockup/store/shape-inference.ts

@@ -63,9 +63,21 @@ export function normalizeEventShape(shape: string): LockupShape | undefined {
 
 /**
  * Infer the shape of a linear stream based on whether it has a cliff.
+ * Detects linearTimelock when cliff is followed by immediate unlock (≤ 1 second).
  */
-export function inferLinearShape(cliff: boolean): LockupShape {
-  return cliff ? Shape.Lockup.Cliff : Shape.Lockup.Linear;
+export function inferLinearShape(
+  cliff: boolean,
+  cliffTime: bigint | undefined,
+  endTime: bigint
+): LockupShape {
+  if (!cliff) {
+    return Shape.Lockup.Linear;
+  }
+  // Detect linearTimelock: cliff followed by immediate unlock (≤ 1 second)
+  if (cliffTime !== undefined && endTime - cliffTime <= TIMELOCK_MAX_DURATION) {
+    return Shape.Lockup.LinearTimelock;
+  }
+  return Shape.Lockup.Cliff;
 }
 
 /**

graph/lockup/store/entity-stream.ts

@@ -66,7 +66,7 @@ export function createStreamLinear(
   }
 
   stream = addCliff(stream, commonParams, linearParams);
-  stream = addLinearShape(stream, stream.cliff);
+  stream = addLinearShape(stream, stream.cliff, stream.cliffTime, stream.endTime);
   stream.save();
 }
 
@@ -235,12 +235,17 @@ function addCliff(
  * Older versions of Lockup did not have a shape field, but it can be inferred.
  * @see https://github.com/sablier-labs/interfaces/blob/30fffc0/packages/constants/src/stream/shape.ts#L12
  */
-function addLinearShape(stream: Entity.Stream, cliff: boolean): Entity.Stream {
+function addLinearShape(
+  stream: Entity.Stream,
+  cliff: boolean,
+  cliffTime: BigInt | null,
+  endTime: BigInt
+): Entity.Stream {
   if (stream.shape !== null) {
     stream.shapeSource = "Event";
     return stream;
   }
-  stream.shape = inferLinearShape(cliff);
+  stream.shape = inferLinearShape(cliff, cliffTime, endTime);
   stream.shapeSource = "Inferred";
   return stream;
 }

graph/lockup/store/shape-inference.ts

@@ -8,9 +8,21 @@ const TIMELOCK_MAX_DURATION = ONE;
 
 /**
  * Infer the shape of a linear stream based on whether it has a cliff.
+ * Detects linearTimelock when cliff is followed by immediate unlock (≤ 1 second).
  */
-export function inferLinearShape(cliff: boolean): string {
-  return cliff ? LockupShape.Cliff : LockupShape.Linear;
+export function inferLinearShape(
+  cliff: boolean,
+  cliffTime: BigInt | null,
+  endTime: BigInt
+): string {
+  if (!cliff) {
+    return LockupShape.Linear;
+  }
+  // Detect linearTimelock: cliff followed by immediate unlock (≤ 1 second)
+  if (cliffTime !== null && endTime.minus(cliffTime).le(TIMELOCK_MAX_DURATION)) {
+    return LockupShape.LinearTimelock;
+  }
+  return LockupShape.Cliff;
 }
 
 /**