mirror of
https://github.com/Nezumi-2711/9router.git
synced 2026-09-22 13:38:31 +00:00
feat(iflow): add IFlowExecutor with HMAC-SHA256 signature and enable models
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
committed by
decolua
co-authored by
Cursor
parent
e3dbd448af
commit
bd23ab41ee
@@ -0,0 +1,104 @@
|
||||
import crypto from "crypto";
|
||||
import { BaseExecutor } from "./base.js";
|
||||
import { PROVIDERS } from "../config/constants.js";
|
||||
|
||||
/**
|
||||
* IFlowExecutor - Executor for iFlow API with HMAC-SHA256 signature
|
||||
*/
|
||||
export class IFlowExecutor extends BaseExecutor {
|
||||
constructor() {
|
||||
super("iflow", PROVIDERS.iflow);
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate UUID v4
|
||||
* @returns {string} UUID v4 string
|
||||
*/
|
||||
generateUUID() {
|
||||
return crypto.randomUUID();
|
||||
}
|
||||
|
||||
/**
|
||||
* Create iFlow signature using HMAC-SHA256
|
||||
* @param {string} userAgent - User agent string
|
||||
* @param {string} sessionID - Session ID
|
||||
* @param {number} timestamp - Unix timestamp in milliseconds
|
||||
* @param {string} apiKey - API key for signing
|
||||
* @returns {string} Hex-encoded signature
|
||||
*/
|
||||
createIFlowSignature(userAgent, sessionID, timestamp, apiKey) {
|
||||
if (!apiKey) return "";
|
||||
const payload = `${userAgent}:${sessionID}:${timestamp}`;
|
||||
const hmac = crypto.createHmac("sha256", apiKey);
|
||||
hmac.update(payload);
|
||||
return hmac.digest("hex");
|
||||
}
|
||||
|
||||
/**
|
||||
* Build headers with iFlow-specific signature
|
||||
* @param {object} credentials - Provider credentials
|
||||
* @param {boolean} stream - Whether streaming is enabled
|
||||
* @returns {object} Headers object
|
||||
*/
|
||||
buildHeaders(credentials, stream = true) {
|
||||
// Generate session ID and timestamp
|
||||
const sessionID = `session-${this.generateUUID()}`;
|
||||
const timestamp = Date.now();
|
||||
|
||||
// Get user agent from config
|
||||
const userAgent = this.config.headers["User-Agent"] || "iFlow-Cli";
|
||||
|
||||
// Get API key (prefer apiKey, fallback to accessToken)
|
||||
const apiKey = credentials.apiKey || credentials.accessToken || "";
|
||||
|
||||
// Create signature
|
||||
const signature = this.createIFlowSignature(userAgent, sessionID, timestamp, apiKey);
|
||||
|
||||
// Build headers
|
||||
const headers = {
|
||||
"Content-Type": "application/json",
|
||||
...this.config.headers,
|
||||
"session-id": sessionID,
|
||||
"x-iflow-timestamp": timestamp.toString(),
|
||||
"x-iflow-signature": signature
|
||||
};
|
||||
|
||||
// Add authorization
|
||||
if (credentials.apiKey) {
|
||||
headers["Authorization"] = `Bearer ${credentials.apiKey}`;
|
||||
}
|
||||
|
||||
// Add streaming header
|
||||
if (stream) {
|
||||
headers["Accept"] = "text/event-stream";
|
||||
}
|
||||
|
||||
return headers;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build URL for iFlow API
|
||||
* @param {string} model - Model name
|
||||
* @param {boolean} stream - Whether streaming is enabled
|
||||
* @param {number} urlIndex - URL index for fallback
|
||||
* @param {object} credentials - Provider credentials
|
||||
* @returns {string} API URL
|
||||
*/
|
||||
buildUrl(model, stream, urlIndex = 0, credentials = null) {
|
||||
return this.config.baseUrl;
|
||||
}
|
||||
|
||||
/**
|
||||
* Transform request body (passthrough for iFlow)
|
||||
* @param {string} model - Model name
|
||||
* @param {object} body - Request body
|
||||
* @param {boolean} stream - Whether streaming is enabled
|
||||
* @param {object} credentials - Provider credentials
|
||||
* @returns {object} Transformed body
|
||||
*/
|
||||
transformRequest(model, body, stream, credentials) {
|
||||
return body;
|
||||
}
|
||||
}
|
||||
|
||||
export default IFlowExecutor;
|
||||
Reference in New Issue
Block a user