diegosouzapw
115 lines
3.1 KiB
TypeScript
115 lines
3.1 KiB
TypeScript
import crypto from "crypto";
|
|
import { BaseExecutor } from "./base.ts";
|
|
import { PROVIDERS } from "../config/constants.ts";
|
|
|
|
type QoderCredentials = {
|
|
apiKey?: string;
|
|
accessToken?: string;
|
|
};
|
|
|
|
/**
|
|
* QoderExecutor - Executor for Qoder API with HMAC-SHA256 signature.
|
|
*
|
|
* Qoder requires custom headers including a session ID, timestamp,
|
|
* and an HMAC-SHA256 signature for request authentication.
|
|
* Without these headers, the API returns a 406 error.
|
|
*
|
|
* Fixes: https://github.com/diegosouzapw/OmniRoute/issues/114
|
|
*/
|
|
export class QoderExecutor extends BaseExecutor {
|
|
constructor() {
|
|
super("qoder", PROVIDERS.qoder);
|
|
}
|
|
|
|
/**
|
|
* Create Qoder signature using HMAC-SHA256
|
|
* @param userAgent - User agent string
|
|
* @param sessionID - Session ID
|
|
* @param timestamp - Unix timestamp in milliseconds
|
|
* @param apiKey - API key for signing
|
|
* @returns Hex-encoded signature
|
|
*/
|
|
createQoderSignature(
|
|
userAgent: string,
|
|
sessionID: string,
|
|
timestamp: number,
|
|
apiKey: string
|
|
): string {
|
|
if (!apiKey) return "";
|
|
const payload = `${userAgent}:${sessionID}:${timestamp}`;
|
|
const hmac = crypto.createHmac("sha256", apiKey);
|
|
hmac.update(payload);
|
|
return hmac.digest("hex");
|
|
}
|
|
|
|
/**
|
|
* Build headers with Qoder-specific HMAC-SHA256 signature.
|
|
* Includes session-id, x-qoder-timestamp, and x-qoder-signature.
|
|
*/
|
|
buildHeaders(credentials: QoderCredentials, stream = true) {
|
|
// Generate session ID and timestamp
|
|
const sessionID = `session-${crypto.randomUUID()}`;
|
|
const timestamp = Date.now();
|
|
|
|
// Get user agent from config
|
|
const userAgent = this.config.headers?.["User-Agent"] || "Qoder-Cli";
|
|
|
|
// Get API key (prefer apiKey, fallback to accessToken)
|
|
const apiKey = credentials.apiKey || credentials.accessToken || "";
|
|
|
|
// Create HMAC-SHA256 signature
|
|
const signature = this.createQoderSignature(userAgent, sessionID, timestamp, apiKey);
|
|
|
|
// Build headers
|
|
const headers: Record<string, string> = {
|
|
"Content-Type": "application/json",
|
|
...this.config.headers,
|
|
"session-id": sessionID,
|
|
"x-qoder-timestamp": timestamp.toString(),
|
|
"x-qoder-signature": signature,
|
|
};
|
|
|
|
// Add authorization
|
|
if (credentials.apiKey) {
|
|
headers["Authorization"] = `Bearer ${credentials.apiKey}`;
|
|
} else if (credentials.accessToken) {
|
|
headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
}
|
|
|
|
// Add streaming header
|
|
if (stream) {
|
|
headers["Accept"] = "text/event-stream";
|
|
}
|
|
|
|
return headers;
|
|
}
|
|
|
|
/**
|
|
* Build URL for Qoder API — uses baseUrl directly.
|
|
*/
|
|
buildUrl(
|
|
model: string,
|
|
stream: boolean,
|
|
urlIndex = 0,
|
|
credentials: QoderCredentials | null = null
|
|
) {
|
|
void model;
|
|
void stream;
|
|
void urlIndex;
|
|
void credentials;
|
|
return this.config.baseUrl;
|
|
}
|
|
|
|
/**
|
|
* Transform request body (passthrough for Qoder).
|
|
*/
|
|
transformRequest(model: string, body: unknown, stream: boolean, credentials: QoderCredentials) {
|
|
void model;
|
|
void stream;
|
|
void credentials;
|
|
return body;
|
|
}
|
|
}
|
|
|
|
export default QoderExecutor;
|