875c7f9b91
- Add full Telegram bot functionality with Z.AI API integration
- Implement 4 tools: Bash, FileEdit, WebSearch, Git
- Add 3 agents: Code Reviewer, Architect, DevOps Engineer
- Add 6 skills for common coding tasks
- Add systemd service file for 24/7 operation
- Add nginx configuration for HTTPS webhook
- Add comprehensive documentation
- Implement WebSocket server for real-time updates
- Add logging system with Winston
- Add environment validation
🤖 zCode CLI X - Agentic coder with Z.AI + Telegram integration
30 lines
1.2 KiB
JavaScript
30 lines
1.2 KiB
JavaScript
import { InvokeStore } from "@aws/lambda-invoke-store";
|
|
import { HttpRequest } from "@smithy/protocol-http";
|
|
const TRACE_ID_HEADER_NAME = "X-Amzn-Trace-Id";
|
|
const ENV_LAMBDA_FUNCTION_NAME = "AWS_LAMBDA_FUNCTION_NAME";
|
|
const ENV_TRACE_ID = "_X_AMZN_TRACE_ID";
|
|
export const recursionDetectionMiddleware = () => (next) => async (args) => {
|
|
const { request } = args;
|
|
if (!HttpRequest.isInstance(request)) {
|
|
return next(args);
|
|
}
|
|
const traceIdHeader = Object.keys(request.headers ?? {}).find((h) => h.toLowerCase() === TRACE_ID_HEADER_NAME.toLowerCase()) ??
|
|
TRACE_ID_HEADER_NAME;
|
|
if (request.headers.hasOwnProperty(traceIdHeader)) {
|
|
return next(args);
|
|
}
|
|
const functionName = process.env[ENV_LAMBDA_FUNCTION_NAME];
|
|
const traceIdFromEnv = process.env[ENV_TRACE_ID];
|
|
const invokeStore = await InvokeStore.getInstanceAsync();
|
|
const traceIdFromInvokeStore = invokeStore?.getXRayTraceId();
|
|
const traceId = traceIdFromInvokeStore ?? traceIdFromEnv;
|
|
const nonEmptyString = (str) => typeof str === "string" && str.length > 0;
|
|
if (nonEmptyString(functionName) && nonEmptyString(traceId)) {
|
|
request.headers[TRACE_ID_HEADER_NAME] = traceId;
|
|
}
|
|
return next({
|
|
...args,
|
|
request,
|
|
});
|
|
};
|