feat: better code formatting

publish.ts

@@ -0,0 +1,100 @@
+import * as dotenv from 'dotenv';
+import * as fs from 'node:fs';
+import { z } from 'zod';
+
+import path = require('node:path');
+
+const EnvVariables = z.object({
+    DIST_PATH: z.string().default('./dist'),
+    SCREEPS_SERVER: z.url().default('https://screeps.com'),
+    SCREEPS_TOKEN: z.string(),
+    SCREEPS_BRANCH: z.string().default('default'),
+});
+type EnvVariables = z.infer<typeof EnvVariables>;
+
+const readEnv = (): EnvVariables => {
+    return EnvVariables.parse(process.env);
+};
+
+type ScriptPayload = {
+    branch: string;
+    modules: Record<string, string>;
+};
+
+type ScreepsHeaders = {
+    'X-Token': string;
+} & object;
+
+const publishScript = async (
+    screeps_server: string,
+    payload: ScriptPayload,
+    headers: ScreepsHeaders
+) => {
+    const resp = await fetch(screeps_server + '/api/user/code', {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            ...headers,
+        },
+        body: JSON.stringify(payload),
+    });
+    if (!resp.ok) {
+        const text = await resp.text();
+        throw new Error(`Upload failed: ${resp.status} ${text}`);
+    }
+};
+
+const genModuleKey = (filePath: string): string => {
+    filePath = filePath.split(path.sep).slice(1).join(path.sep);
+    return filePath.split('.')[0].split(path.sep).join('.');
+};
+
+const parseDist = (dist_path: string): Record<string, string> => {
+    let modules: Record<string, string> = {};
+
+    const files = fs.readdirSync(dist_path, { withFileTypes: true });
+    for (const file of files) {
+        if (file.isDirectory()) {
+            const subModules = parseDist(path.join(dist_path, file.name));
+            modules = {
+                ...modules,
+                ...subModules,
+            };
+            continue;
+        }
+
+        if (!file.name.endsWith('.js')) {
+            continue;
+        }
+
+        const filePath = path.join(dist_path, file.name);
+        const module_key = genModuleKey(filePath);
+        modules[module_key] = fs.readFileSync(filePath, 'utf-8');
+    }
+    return modules;
+};
+
+const main = async () => {
+    dotenv.config();
+
+    const env = readEnv();
+
+    const payload: ScriptPayload = {
+        branch: env.SCREEPS_BRANCH,
+        modules: parseDist(env.DIST_PATH),
+    };
+    console.log(JSON.stringify(payload));
+    const headers: ScreepsHeaders = {
+        'X-Token': env.SCREEPS_TOKEN,
+    };
+
+    await publishScript(env.SCREEPS_SERVER, payload, headers);
+};
+
+main()
+    .then(() => {
+        console.log('Script published successfully!');
+    })
+    .catch((error) => {
+        console.error(error);
+    });