feat: add invocation mode to manifest (#1421)

Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>

src/function.ts

@@ -2,6 +2,14 @@ import { Stats } from 'fs'
 
 import type { FunctionConfig } from './config.js'
 import type { Runtime, ZipFunctionResult } from './runtimes/runtime.js'
+import { ObjectValues } from './types/utils.js'
+
+export const INVOCATION_MODE = {
+  Buffer: 'buffer',
+  Stream: 'stream',
+} as const
+
+export type InvocationMode = ObjectValues<typeof INVOCATION_MODE>
 
 // A function that has been processed and turned into an archive.
 export type FunctionArchive = ZipFunctionResult & {

src/manifest.ts

@@ -3,9 +3,11 @@ import { resolve } from 'path'
 import { arch, platform } from 'process'
 
 import type { FeatureFlags } from './feature_flags.js'
+import type { InvocationMode } from './function.js'
 import type { FunctionResult } from './utils/format_result.js'
 
 interface ManifestFunction {
+  invocationMode?: InvocationMode
   mainFile: string
   name: string
   path: string
@@ -50,12 +52,24 @@ export const createManifest = async ({
 }
 
 const formatFunctionForManifest = (
-  { bundler, displayName, generator, mainFile, name, path, runtime, runtimeVersion, schedule }: FunctionResult,
+  {
+    bundler,
+    displayName,
+    generator,
+    invocationMode,
+    mainFile,
+    name,
+    path,
+    runtime,
+    runtimeVersion,
+    schedule,
+  }: FunctionResult,
   featureFlags: FeatureFlags,
 ): ManifestFunction => ({
   bundler,
   displayName,
   generator,
+  invocationMode,
   mainFile,
   name,
   runtimeVersion: featureFlags.functions_inherit_build_nodejs_version ? runtimeVersion : undefined,

src/runtimes/node/index.ts

@@ -2,6 +2,7 @@ import { join } from 'path'
 
 import { copyFile } from 'cp-file'
 
+import { INVOCATION_MODE } from '../../function.js'
 import getInternalValue from '../../utils/get_internal_value.js'
 import { GetSrcFilesFunction, Runtime, RUNTIME, ZipFunction } from '../runtime.js'
 
@@ -117,13 +118,16 @@ const zipFunction: ZipFunction = async function ({
 
   await cleanupFunction?.()
 
+  const invocationMode = featureFlags.zisi_functions_api_v2 ? INVOCATION_MODE.Stream : INVOCATION_MODE.Buffer
+
   return {
     bundler: bundlerName,
     bundlerWarnings,
     config,
     inputs,
     includedFiles,
     inSourceConfig,
+    invocationMode,
     nativeNodeModules,
     nodeModulesWithDynamicImports,
     path: zipPath,

src/runtimes/runtime.ts

@@ -1,7 +1,7 @@
 import type { ArchiveFormat } from '../archive.js'
 import type { FunctionConfig } from '../config.js'
 import type { FeatureFlags } from '../feature_flags.js'
-import type { FunctionSource, SourceFile } from '../function.js'
+import type { FunctionSource, InvocationMode, SourceFile } from '../function.js'
 import { ObjectValues } from '../types/utils.js'
 import type { RuntimeCache } from '../utils/cache.js'
 
@@ -47,6 +47,7 @@ export interface ZipFunctionResult {
   inputs?: string[]
   includedFiles?: string[]
   inSourceConfig?: ISCValues
+  invocationMode?: InvocationMode
   nativeNodeModules?: object
   nodeModulesWithDynamicImports?: string[]
   path: string