spiceflow 0.0.7 → 1.0.0

package/src/stream.test.ts

@@ -0,0 +1,342 @@
+import { describe, it, expect } from 'vitest'
+
+import { createParser } from 'eventsource-parser'
+
+import { Elysia } from './spiceflow'
+
+import { req, sleep } from './utils'
+
+function textEventStream(items: string[]) {
+	return items
+		.map((item) => `event: message\ndata: ${JSON.stringify(item)}\n\n`)
+		.join('')
+}
+
+function parseTextEventStreamItem(item: string) {
+	const data = item.split('data: ')[1].split('\n')[0]
+	return JSON.parse(data)
+}
+
+describe('Stream', () => {
+	it('handle stream', async () => {
+		const expected = ['a', 'b', 'c']
+
+		const app = new Elysia().get('/', async function* () {
+			yield 'a'
+			await sleep(10)
+
+			yield 'b'
+			await sleep(10)
+
+			yield 'c'
+		})
+
+		const response = await app
+			.handle(req('/'))
+			.then((x) => x.body)
+			.then((x) => {
+				if (!x) return
+
+				const reader = x?.getReader()
+
+				let acc = ''
+				const { promise, resolve } = Promise.withResolvers()
+
+				reader.read().then(function pump({ done, value }): unknown {
+					if (done) return resolve(acc)
+
+					expect(parseTextEventStreamItem(value.toString())).toBe(
+						expected.shift()!
+					)
+
+					acc += value.toString()
+					return reader.read().then(pump)
+				})
+
+				return promise
+			})
+
+		expect(expected).toHaveLength(0)
+		expect(response).toBe(textEventStream(['a', 'b', 'c']))
+	})
+	it('handle errors after yield', async () => {
+		const app = new Elysia().get('/', async function* () {
+			yield 'a'
+			await sleep(10)
+
+			throw new Error('an error')
+		})
+
+		const response = await app.handle(req('/')).then((x) => x.text())
+
+		expect(response).toBe(
+			'event: message\ndata: "a"\n\nevent: error\ndata: "an error"\n\n'
+		)
+	})
+
+	it('handle errors before yield when aot is false', async () => {
+		const app = new Elysia()
+			.onError(({ error }) => {
+				return new Response(error.message)
+			})
+			.get('/', async function* () {
+				throw new Error('an error xxxx')
+			})
+
+		const response = await app.handle(req('/')).then((x) => x.text())
+
+		expect(response).toContain('an error')
+	})
+
+	it.todo('handle errors before yield when aot is true', async () => {
+		const app = new Elysia()
+			.onError(({ error }) => {
+				return new Response(error.message)
+			})
+			.get('/', async function* () {
+				throw new Error('an error')
+			})
+
+		const response = await app.handle(req('/')).then((x) => x.text())
+
+		expect(response).toContain('an error')
+	})
+
+	it.todo('handle errors before yield with onError', async () => {
+		const expected = 'error expected'
+		const app = new Elysia()
+			.onError(({}) => {
+				return new Response(expected)
+			})
+			.get('/', async function* () {
+				throw new Error('an error')
+			})
+
+		const response = await app.handle(req('/')).then((x) => x.text())
+
+		expect(response).toBe(expected)
+	})
+
+	it('stop stream on canceled request', async () => {
+		const expected = ['a', 'b']
+
+		const app = new Elysia().get('/', async function* () {
+			yield 'a'
+			await sleep(10)
+
+			yield 'b'
+			await sleep(10)
+
+			yield 'c'
+		})
+
+		const controller = new AbortController()
+
+		setTimeout(() => {
+			controller.abort()
+		}, 15)
+
+		const response = await app
+			.handle(
+				new Request('http://e.ly', {
+					signal: controller.signal
+				})
+			)
+			.then((x) => x.body)
+			.then((x) => {
+				if (!x) return
+
+				const reader = x?.getReader()
+
+				let acc = ''
+				const { promise, resolve } = Promise.withResolvers()
+
+				reader.read().then(function pump({ done, value }): unknown {
+					if (done) {
+						return resolve(acc)
+					}
+
+					expect(parseTextEventStreamItem(value.toString())).toBe(
+						expected.shift()!
+					)
+
+					acc += value.toString()
+					return reader.read().then(pump)
+				})
+
+				return promise
+			})
+
+		expect(expected).toHaveLength(0)
+		expect(response).toBe(textEventStream(['a', 'b']))
+	})
+
+	// it('mutate set before yield is called', async () => {
+	// 	const expected = ['a', 'b', 'c']
+
+	// 	const app = new Elysia().get('/', function* () {
+	// 		set.headers['access-control-allow-origin'] = 'http://saltyaom.com'
+
+	// 		yield 'a'
+	// 		yield 'b'
+	// 		yield 'c'
+	// 	})
+
+	// 	const response = await app.handle(req('/')).then((x) => x.headers)
+
+	// 	expect(response.get('access-control-allow-origin')).toBe(
+	// 		'http://saltyaom.com'
+	// 	)
+	// })
+	it('handle stream with objects', async () => {
+		const objects = [
+			{ message: 'hello' },
+			{ response: 'world' },
+			{ data: [1, 2, 3] },
+			{ result: [4, 5, 6] }
+		]
+		const app = new Elysia().get('/', async function* ({}) {
+			for (const obj of objects) {
+				yield obj
+			}
+		})
+
+		const body = await app.handle(req('/')).then((x) => x.body)
+
+		let events = [] as any[]
+		const parser = createParser((event) => {
+			events.push(event)
+		})
+		const { promise, resolve } = Promise.withResolvers<void>()
+		const reader = body?.getReader()!
+
+		reader.read().then(function pump({ done, value }): unknown {
+			if (done) {
+				return resolve()
+			}
+			const text = value.toString()
+			parser.feed(text)
+			return reader.read().then(pump)
+		})
+		await promise
+
+		expect(events.map((x) => x.data)).toEqual(
+			objects.map((x) => JSON.stringify(x))
+		)
+	})
+
+	// it('mutate set before yield is called', async () => {
+	// 	const expected = ['a', 'b', 'c']
+
+	// 	const app = new Elysia().get('/', function* () {
+	// 		set.headers['access-control-allow-origin'] = 'http://saltyaom.com'
+
+	// 		yield 'a'
+	// 		yield 'b'
+	// 		yield 'c'
+	// 	})
+
+	// 	const response = await app.handle(req('/')).then((x) => x.headers)
+
+	// 	expect(response.get('access-control-allow-origin')).toBe(
+	// 		'http://saltyaom.com'
+	// 	)
+	// })
+
+	// it('async mutate set before yield is called', async () => {
+	// 	const expected = ['a', 'b', 'c']
+
+	// 	const app = new Elysia().get('/', async function* () {
+	// 		set.headers['access-control-allow-origin'] = 'http://saltyaom.com'
+
+	// 		yield 'a'
+	// 		yield 'b'
+	// 		yield 'c'
+	// 	})
+
+	// 	const response = await app.handle(req('/')).then((x) => x.headers)
+
+	// 	expect(response.get('access-control-allow-origin')).toBe(
+	// 		'http://saltyaom.com'
+	// 	)
+	// })
+
+	it('return value if not yield', async () => {
+		const app = new Elysia()
+			.get('/', function* () {
+				return 'hello'
+			})
+			.get('/json', function* () {
+				return { hello: 'world' }
+			})
+
+		const response = await Promise.all([
+			app.handle(req('/')),
+			app.handle(req('/json'))
+		])
+
+		expect(await response[0].text()).toBe('"hello"')
+		expect(await response[1].json()).toEqual({
+			hello: 'world'
+		})
+	})
+
+	it('return async value if not yield', async () => {
+		const app = new Elysia()
+			.get('/', function* () {
+				return 'hello'
+			})
+			.get('/json', function* () {
+				return { hello: 'world' }
+			})
+
+		const response = await Promise.all([
+			app.handle(req('/')),
+			app.handle(req('/json'))
+		])
+
+		expect(await response[0].text()).toBe('"hello"')
+		expect(await response[1].json()).toEqual({
+			hello: 'world'
+		})
+	})
+
+	it('handle object and array', async () => {
+		const expected = [{ a: 'b' }, ['a'], ['a', 1, { a: 'b' }]]
+		const expectedResponse = JSON.stringify([...expected])
+		let i = 0
+
+		const app = new Elysia().get('/', async function* () {
+			yield expected[0]
+			await sleep(10)
+
+			yield expected[1]
+			await sleep(10)
+
+			yield expected[2]
+		})
+
+		const response = await app
+			.handle(req('/'))
+			.then((x) => x.body)
+			.then((x) => {
+				if (!x) return
+
+				const reader = x?.getReader()
+
+				const { promise, resolve } = Promise.withResolvers<void>()
+
+				reader.read().then(function pump({ done, value }): unknown {
+					if (done) return resolve()
+
+					expect(parseTextEventStreamItem(value.toString())).toEqual(
+						expected[i++]
+					)
+
+					return reader.read().then(pump)
+				})
+
+				return promise
+			})
+	})
+})

package/src/utils.ts

@@ -1,75 +1,26 @@
-import { PluginPass, parse, traverse } from '@babel/core';
-
-import { Node, types } from '@babel/core';
-import { JsonRpcError, JsonRpcErrorResponse } from './jsonRpc.js';
-
-const PURE_ANNOTATION = '#__PURE__';
-
-export function annotateAsPure<N extends Node>(t: typeof types, node: N): N {
-  t.addComment(node, 'leading', PURE_ANNOTATION);
-  return node;
+// deno-lint-ignore no-explicit-any
+export const deepFreeze = (value: any) => {
+	for (const key of Reflect.ownKeys(value)) {
+		if (value[key] && typeof value[key] === 'object') {
+			deepFreeze(value[key])
+		}
+	}
+	return Object.freeze(value)
 }
 
-export type Literal =
-  | number
-  | string
-  | boolean
-  | { [key: string]: Literal }
-  | Literal[];
-
-const enabled = !!process.env.DEBUG_ACTIONS;
-export const logger = {
-  log(...args) {
-    enabled && console.log('[actions]:', ...args);
-  },
-  error(...args) {
-    enabled && console.log('[actions]:', ...args);
-  },
-};
-
-export function getFileName(state: PluginPass) {
-  const { filename, cwd } = state;
-
-  if (!filename) {
-    return undefined;
-  }
-
-  if (cwd && filename.startsWith(cwd)) {
-    return filename.slice(cwd.length + 1);
-  }
-
-  return filename;
-}
-export const directive = 'use spiceflow';
-
-export function jsonRpcError({
-  id = null,
-  message,
-  code = 1,
-}): JsonRpcErrorResponse {
-  return {
-    jsonrpc: '2.0',
-    id,
-    error: {
-      code,
-      message,
-      data: null,
-    },
-  };
-}
-
-export function camelCaseCapitalized(str: string) {
-  // functionName -> FunctionName
-  // CamelCase -> CamelCase
-  // camelCase -> CamelCase
-
-  if (str.length === 0) {
-    return str;
-  }
-  const first = str[0].toUpperCase();
-  return first + str.slice(1);
+export const req = (path: string, options?: RequestInit) =>
+	new Request(`http://localhost${path}`, options)
+
+export function isAsyncIterable(obj: any): obj is AsyncGenerator<any> {
+	return (
+		typeof obj === 'object' &&
+		typeof obj.next === 'function' &&
+		typeof obj.return === 'function' &&
+		typeof obj.throw === 'function' &&
+		typeof obj.return === 'function'
+	)
 }
 
-export function removeExtension(filePath: string) {
-  return filePath.replace(/\.[j|t]sx?$/, '');
+export function sleep(ms: number) {
+	return new Promise((resolve) => setTimeout(resolve, ms))
 }

package/context.d.ts

@@ -1,2 +0,0 @@
-import * as context from './dist/context';
-export = context;

package/context.js

@@ -1 +0,0 @@
-export * from './dist/context';

package/dist/babel.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=babel.test.d.ts.map

package/dist/babel.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"babel.test.d.ts","sourceRoot":"","sources":["../src/babel.test.ts"],"names":[],"mappings":""}

package/dist/babel.test.js

@@ -1,27 +0,0 @@
-import { test, expect } from 'vitest';
-import { parseExpression } from '@babel/parser';
-import generate from '@babel/generator';
-// import { parseExpression } from './utils';
-test('parseExpression', () => {
-    {
-        const parsed = parseExpression('{ "type": "StringLiteral", "value": "/api/actions-node" }');
-        const code = generate(parsed).code;
-        expect(code).toMatchInlineSnapshot(`
-      "{
-        "type": "StringLiteral",
-        "value": "/api/actions-node"
-      }"
-    `);
-    }
-    {
-        const parsed = parseExpression('null');
-        const code = generate(parsed).code;
-        expect(code).toMatchInlineSnapshot(`"null"`);
-    }
-    {
-        const parsed = parseExpression(`typeof wrapMethod === 'function' ? wrapMethod : undefined`);
-        const code = generate(parsed).code;
-        expect(code).toMatchInlineSnapshot(`"typeof wrapMethod === 'function' ? wrapMethod : undefined"`);
-    }
-});
-//# sourceMappingURL=babel.test.js.map

package/dist/babel.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"babel.test.js","sourceRoot":"","sources":["../src/babel.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,EAAY,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,QAAQ,MAAM,kBAAkB,CAAC;AAGxC,6CAA6C;AAE7C,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC3B,CAAC;QACC,MAAM,MAAM,GAAG,eAAe,CAC5B,2DAA2D,CAC5D,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC;;;;;KAKlC,CAAC,CAAC;IACL,CAAC;IACD,CAAC;QACC,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IACD,CAAC;QACC,MAAM,MAAM,GAAG,eAAe,CAC5B,2DAA2D,CAC5D,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAChC,6DAA6D,CAC9D,CAAC;IACJ,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/babelDebugOutputs.d.ts

@@ -1,9 +0,0 @@
-import * as babel from '@babel/core';
-import * as types from '@babel/types';
-import { PluginOptions } from './babelTransformRpc.js';
-type Babel = {
-    types: typeof types;
-};
-export default function debugOutputsPlugin({ types: t }: Babel, { isServer }: PluginOptions): babel.PluginObj | undefined;
-export {};
-//# sourceMappingURL=babelDebugOutputs.d.ts.map

package/dist/babelDebugOutputs.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"babelDebugOutputs.d.ts","sourceRoot":"","sources":["../src/babelDebugOutputs.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,aAAa,CAAC;AAErC,OAAO,KAAK,KAAK,MAAM,cAAc,CAAC;AAEtC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAKvD,KAAK,KAAK,GAAG;IAAE,KAAK,EAAE,OAAO,KAAK,CAAA;CAAE,CAAC;AAIrC,MAAM,CAAC,OAAO,UAAU,kBAAkB,CACxC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,KAAK,EACnB,EAAE,QAAQ,EAAE,EAAE,aAAa,GAC1B,KAAK,CAAC,SAAS,GAAG,SAAS,CAuC7B"}

package/dist/babelDebugOutputs.js

@@ -1,34 +0,0 @@
-import generate from '@babel/generator';
-import fs from 'fs';
-import { default as nodePath, default as path } from 'path';
-import { getFileName, logger } from './utils.js';
-let deletedDir = false;
-export default function debugOutputsPlugin({ types: t }, { isServer }) {
-    const cwd = process.cwd();
-    if (!deletedDir) {
-        deletedDir = true;
-        fs.mkdirSync('./actions-outputs', { recursive: true });
-    }
-    return {
-        visitor: {
-            Program: {
-                exit(program, state) {
-                    var _a;
-                    const filePath = (_a = getFileName(state)) !== null && _a !== void 0 ? _a : nodePath.join('pages', 'Default.js');
-                    if (!process.env.DEBUG_ACTIONS) {
-                        return;
-                    }
-                    // stringify the AST and print it
-                    const output = generate(program.node, {
-                    /* options */
-                    }, this.file.code);
-                    let p = path.resolve('./actions-outputs', isServer ? 'server/' : 'client/', path.relative(cwd, path.resolve(filePath)));
-                    logger.log(`${isServer ? 'server' : 'client'} plugin output:`, p);
-                    fs.mkdirSync(path.dirname(p), { recursive: true });
-                    fs.writeFileSync(p, output.code);
-                },
-            },
-        },
-    };
-}
-//# sourceMappingURL=babelDebugOutputs.js.map

package/dist/babelDebugOutputs.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"babelDebugOutputs.js","sourceRoot":"","sources":["../src/babelDebugOutputs.ts"],"names":[],"mappings":"AACA,OAAO,QAAQ,MAAM,kBAAkB,CAAC;AAExC,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,OAAO,IAAI,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5D,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAIjD,IAAI,UAAU,GAAG,KAAK,CAAC;AAEvB,MAAM,CAAC,OAAO,UAAU,kBAAkB,CACxC,EAAE,KAAK,EAAE,CAAC,EAAS,EACnB,EAAE,QAAQ,EAAiB;IAE3B,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAE1B,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,UAAU,GAAG,IAAI,CAAC;QAElB,EAAE,CAAC,SAAS,CAAC,mBAAmB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC;IACD,OAAO;QACL,OAAO,EAAE;YACP,OAAO,EAAE;gBACP,IAAI,CAAC,OAAO,EAAE,KAAK;;oBACjB,MAAM,QAAQ,GACZ,MAAA,WAAW,CAAC,KAAK,CAAC,mCAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;oBAE7D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC;wBAC/B,OAAO;oBACT,CAAC;oBAED,iCAAiC;oBACjC,MAAM,MAAM,GAAG,QAAQ,CACrB,OAAO,CAAC,IAAI,EACZ;oBACE,aAAa;qBACd,EACD,IAAI,CAAC,IAAI,CAAC,IAAI,CACf,CAAC;oBACF,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAClB,mBAAmB,EACnB,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,EAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAC3C,CAAC;oBACF,MAAM,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,iBAAiB,EAAE,CAAC,CAAC,CAAC;oBAClE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBACnD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;gBACnC,CAAC;aACF;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/babelTransformRpc.d.ts

@@ -1,19 +0,0 @@
-import * as babel from '@babel/core';
-import type * as types from '@babel/types';
-import { WrapMethodMeta } from './server.js';
-type Babel = {
-    types: typeof types;
-};
-export declare function getConfigObject(program: babel.NodePath<babel.types.Program>): {
-    isEdge: boolean;
-} | undefined;
-export declare function isEdgeInConfig(configObject?: babel.NodePath<babel.types.ObjectExpression>): boolean;
-export interface PluginOptions {
-    isServer: boolean;
-    rootDir: string;
-    onMethod?: (method: WrapMethodMeta) => void;
-    url?: string;
-}
-export default function ({ types: t }: Babel, { rootDir, onMethod, isServer, url: rpcUrl }: PluginOptions): babel.PluginObj;
-export {};
-//# sourceMappingURL=babelTransformRpc.d.ts.map

package/dist/babelTransformRpc.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"babelTransformRpc.d.ts","sourceRoot":"","sources":["../src/babelTransformRpc.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,aAAa,CAAC;AAGrC,OAAO,KAAK,KAAK,KAAK,MAAM,cAAc,CAAC;AAG3C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG7C,KAAK,KAAK,GAAG;IAAE,KAAK,EAAE,OAAO,KAAK,CAAA;CAAE,CAAC;AA+CrC,wBAAgB,eAAe,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC;;cAe3E;AAmDD,wBAAgB,cAAc,CAC5B,YAAY,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAC1D,OAAO,CAoBT;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,IAAI,CAAC;IAC5C,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,CAAC,OAAO,WACZ,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,KAAK,EACnB,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,aAAa,GAC1D,KAAK,CAAC,SAAS,CA6OjB"}