Create server.mjs

server.mjs

@@ -0,0 +1,115 @@
+import http from 'node:http';
+import {Client} from 'itdog-web-api';
+
+const port = Number.parseInt(process.env.PORT || '7860', 10);
+
+const client = new Client();
+
+const sendJson = (res, statusCode, payload) => {
+    const body = JSON.stringify(payload, null, 2);
+    res.writeHead(statusCode, {
+        'content-type': 'application/json; charset=utf-8',
+        'access-control-allow-origin': '*',
+        'access-control-allow-methods': 'GET,POST,OPTIONS',
+        'access-control-allow-headers': 'content-type',
+    });
+    res.end(body);
+};
+
+const readJsonBody = (req) => new Promise((resolve, reject) => {
+    let data = '';
+    req.on('data', (chunk) => {
+        data += chunk;
+        if (data.length > 2 * 1024 * 1024) {
+            reject(new Error('Request body too large'));
+            req.destroy();
+        }
+    });
+    req.on('end', () => {                                                          if (!data) {
+            resolve({});
+            return;
+        }
+        try {
+            resolve(JSON.parse(data));
+        } catch {
+            reject(new Error('Invalid JSON body'));
+        }
+    });
+    req.on('error', (err) => reject(err));
+});                                                                                                                                                   const pickAPIResult = (result) => ({
+    task_id: result.task_id,
+    wss_url: result.wss_url,
+    taskToken: result.taskToken,
+});
+
+const runWS = async (runner) => {
+    const messages = [];
+    const final = await runner((data) => {
+        messages.push(data);
+    });
+    return {
+        result: pickAPIResult(final.result),
+        messages,
+    };
+};
+                                                                           const routes = {
+    'POST /ping': (body) => runWS((onMessage) => client.ping(body, onMessage)),
+    'POST /pingv6': (body) => runWS((onMessage) => client.pingV6(body, onMessage)),
+    'POST /tcping': (body) => runWS((onMessage) => client.tcping(body, onMessage)),
+    'POST /tcpingv6': (body) => runWS((onMessage) => client.tcpingV6(body, onMessage)),
+    'POST /http': (body) => runWS((onMessage) => client.http(body, onMessage)),
+    'POST /httpv6': (body) => runWS((onMessage) => client.httpV6(body, onMessage)),
+    'POST /httping': (body) => runWS((onMessage) => client.httping(body, onMessage)),
+    'POST /httpingv6': (body) => runWS((onMessage) => client.httpingV6(body, onMessage)),
+    'POST /traceroute': (body) => runWS((onMessage) => client.traceRoute(body, onMessage)),
+    'POST /traceroutev6': (body) => runWS((onMessage) => client.traceRouteV6(body, onMessage)),
+    'POST /dns': (body) => runWS((onMessage) => client.dns(body, onMessage)),
+    'POST /batch-tcping': (body) => runWS((onMessage) => client.batchTCPing(body, onMessage)),
+    'POST /generic': async (body) => {
+        if (!body?.endpoint) {
+            throw new Error('Missing "endpoint" in body');
+        }
+        const method = body.method || 'POST';
+        const params = body.params || {};
+        return runWS((onMessage) => client.generic(body.endpoint, method, params, onMessage));
+    },
+};
+
+const server = http.createServer(async (req, res) => {
+    if (!req.url || !req.method) {
+        sendJson(res, 400, {error: 'Invalid request'});
+        return;
+    }
+
+    if (req.method === 'OPTIONS') {
+        sendJson(res, 204, {});
+        return;
+    }
+
+    if (req.method === 'GET' && req.url === '/') {
+        sendJson(res, 200, {
+            ok: true,
+            service: 'itdog-web-api',
+            endpoints: Object.keys(routes).map((key) => key.replace('POST ', '')),
+        });
+        return;
+    }
+
+    const routeKey = `${req.method} ${req.url}`;
+    const handler = routes[routeKey];
+    if (!handler) {
+        sendJson(res, 404, {error: 'Not found'});
+        return;
+    }
+
+    try {
+        const body = await readJsonBody(req);
+        const data = await handler(body);
+        sendJson(res, 200, data);
+    } catch (error) {
+        sendJson(res, 400, {error: error instanceof Error ? error.message : 'Unknown error'});
+    }
+});
+
+server.listen(port, () => {                                                    console.log(`Server listening on http://0.0.0.0:${port}`);
+});