Updated main function to handle generics

examples/stdin/index.ts

@@ -1,13 +1,16 @@
-import { InputProps, entryMain } from '@blockless/sdk-ts'
+import { main, readInput } from '../../lib'
 import { AbiCoder } from 'ethers'
 
 interface Arguments {
 	n: number
 	v: string
 }
 
-entryMain(async (input: InputProps<Arguments>) => {
-	console.log('\n Example: Stdin')
+main(async() => {
+	console.log('\nExample: Stdin')
+
+	// Read input arguments
+	const input = readInput<Arguments>()
 
 	if (Object.keys(input.args).length === 0) {
 		console.log('Missing args.')

examples/stdin/package.json

@@ -4,12 +4,9 @@
   "main": "index.ts",
   "private": true,
   "scripts": {
-    "build": "bls-sdk-ts build ./index.ts",
+    "build": "../../dist/bundler/index.js build ./index.ts",
     "invoke": "echo '{ \"n\": 2, \"v\": \"abc\" }' | bls-runtime-deno build/index.wasm"
   },
-  "devDependencies": {
-    "@blockless/sdk-ts": "file:../.."
-  },
   "dependencies": {
     "ethers": "^6.11.1"
   }

lib/entry.ts

@@ -1,9 +1,24 @@
-import { InputProps, readInput, writeOutput } from './stdin'
+import { writeOutput } from './stdin'
 
-export async function entryMain<T>(
-	cb: (input: InputProps<T>) => Promise<object>
-): Promise<void> {
-	const input = readInput<T>()
-	const result = await cb(input)
+type EntryCallback<T extends object> = () => T
+type EntryCallbackAsync<T extends object> = () => Promise<T>
+
+export function main<T extends object>(cb: EntryCallback<T>): T
+export async function main<T extends object>(cb: EntryCallbackAsync<T>): Promise<T>
+export async function main<T extends object>(cb: EntryCallback<T> | EntryCallbackAsync<T>): Promise<T> {
+	if (isPromiseCallback(cb)) {
+		const result = await cb()
+		writeOutput(result)
+
+		return result
+	}
+
+	const result = cb()
 	writeOutput(result)
+
+	return result
 }
+
+function isPromiseCallback<T extends object>(cb: EntryCallback<T> | EntryCallbackAsync<T>): cb is EntryCallbackAsync<T> {
+	return typeof cb === 'function' && cb.length === 0;
+}

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@blockless/sdk-ts",
-	"version": "1.0.0",
+	"version": "1.0.1",
 	"type": "module",
 	"module": "dist/lib/index.js",
 	"types": "dist/lib/index.d.ts",