96 lines
2.9 KiB
TypeScript
96 lines
2.9 KiB
TypeScript
// TypeScript类型声明
|
||
// @ts-nocheck
|
||
type LogFunction = (...args: any[]) => void;
|
||
type Handler = (input: any) => any;
|
||
|
||
// Save original console.log
|
||
const originalConsoleLog: LogFunction = console.log;
|
||
let logs: string[] = [];
|
||
|
||
// Replace console.log to capture logs
|
||
console.log = function(...args: any[]): void {
|
||
// Convert arguments to string and join them
|
||
const message = args.map(arg =>
|
||
typeof arg === 'object' && arg !== null ? JSON.stringify(arg) : String(arg)
|
||
).join(' ');
|
||
|
||
// Store log
|
||
logs.push(message);
|
||
|
||
// Also log to original console if showing logs
|
||
if ({{SHOW_LOGS}}) {
|
||
originalConsoleLog.apply(console, args);
|
||
}
|
||
};
|
||
|
||
// 异步读取输入参数的函数
|
||
async function readInputParams(): Promise<any> {
|
||
let input: any = {};
|
||
try {
|
||
const inputFile = Deno.env.get("INPUT_JSON_FILE");
|
||
if (inputFile) {
|
||
const inputJson = await Deno.readTextFile(inputFile);
|
||
input = JSON.parse(inputJson);
|
||
console.log("接收到的参数:", JSON.stringify(input));
|
||
} else {
|
||
// 兼容旧的环境变量方式
|
||
const inputJson = Deno.env.get("INPUT_JSON");
|
||
if (inputJson) {
|
||
input = JSON.parse(inputJson);
|
||
console.log("接收到的参数:", JSON.stringify(input));
|
||
}
|
||
}
|
||
} catch (error) {
|
||
console.error("解析输入参数失败:", error);
|
||
}
|
||
return input;
|
||
}
|
||
|
||
async function executeHandler() {
|
||
try {
|
||
// Add the original code
|
||
{{USER_CODE}}
|
||
|
||
// 读取输入参数
|
||
const input = await readInputParams();
|
||
|
||
// Execute handler function and get result
|
||
let result: any = null;
|
||
|
||
// 优先检查main函数
|
||
if (typeof main === 'function') {
|
||
// 检查main是否是异步函数
|
||
if (main.constructor.name === 'AsyncFunction') {
|
||
result = await (main as (input: any) => Promise<any>)(input);
|
||
} else {
|
||
result = (main as Handler)(input);
|
||
}
|
||
} else if (typeof handler === 'function') {
|
||
// 如果没有main函数，检查handler
|
||
if (handler.constructor.name === 'AsyncFunction') {
|
||
result = await (handler as (input: any) => Promise<any>)(input);
|
||
} else {
|
||
result = (handler as Handler)(input);
|
||
}
|
||
} else {
|
||
throw new Error("没有找到main或handler函数");
|
||
}
|
||
|
||
// Print final output as JSON
|
||
originalConsoleLog(JSON.stringify({
|
||
logs: logs,
|
||
result: result !== undefined ? (typeof result === 'object' ? JSON.stringify(result) : String(result)) : null,
|
||
error: null
|
||
}));
|
||
} catch (error) {
|
||
// Handle errors
|
||
originalConsoleLog(JSON.stringify({
|
||
logs: logs,
|
||
result: null,
|
||
error: String(error)
|
||
}));
|
||
}
|
||
}
|
||
|
||
// 执行并等待结果
|
||
executeHandler();
|