@waron97/prbot 2.2.0 → 2.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@waron97/prbot",
-    "version": "2.2.0",
+    "version": "2.3.0",
     "description": "",
     "main": "index.js",
     "scripts": {

package/src/commands/export.js

@@ -0,0 +1,7 @@
+import { exportPb } from './exportPb.js';
+
+function exportRip() {
+    console.log('Not implemented yet.');
+}
+
+export { exportPb, exportRip };

package/src/commands/exportPb.js

@@ -0,0 +1,156 @@
+import fs from 'fs/promises';
+import path from 'path';
+import fetch from 'node-fetch';
+import search from '@inquirer/search';
+import { getToken } from '../lib/auth.js';
+import { execGit } from '../lib/git.js';
+import { resolveAddonsPath } from '../lib/addons.js';
+
+async function getProcessList(token) {
+    const url = `${process.env.IMPORTEXPORT_URL}/object/process_builder?addLanguageParam=true`;
+    const response = await fetch(url, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${token}`,
+        },
+        body: JSON.stringify({ page: 1, size: 999999, filters: [] }),
+    });
+    if (!response.ok) throw new Error(await response.text());
+    const json = await response.json();
+    return json.data;
+}
+
+async function initiateExport(guid, token) {
+    const url = `${process.env.IMPORTEXPORT_URL}/symphony/export`;
+    const response = await fetch(url, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${token}`,
+        },
+        body: JSON.stringify([{ object_guid: guid, object_type: 'process_builder' }]),
+    });
+    if (!response.ok) throw new Error(await response.text());
+}
+
+async function pollExportResult(guid, requestTime, token) {
+    const url = `${process.env.IMPORTEXPORT_URL}/export/info/processKey=ExportElement&subProcess=true&status=FAILED,COMPLETED&referenceId=process_builder`;
+    // Server createDate is offset -1hr from system time; subtract 1hr+5s buffer
+    const cutoff = requestTime - 3_605_000;
+
+    while (true) {
+        await new Promise((r) => setTimeout(r, 3000));
+
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${token}`,
+            },
+            body: JSON.stringify({ page: 1, size: 7, sorters: [] }),
+        });
+        if (!response.ok) throw new Error(await response.text());
+
+        const { data } = await response.json();
+        const match = data.find(
+            (item) =>
+                item.customResponse?.guid === guid &&
+                new Date(item.createdDate).getTime() >= cutoff
+        );
+
+        if (!match) continue;
+        if (match.status === 'FAILED') throw new Error(`Export failed for guid ${guid}`);
+        return match.requestId;
+    }
+}
+
+async function downloadZip(requestId, token) {
+    const url = `${process.env.IMPORTEXPORT_URL}/export/${requestId}`;
+    const response = await fetch(url, {
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${token}`,
+        },
+    });
+    if (!response.ok) throw new Error(await response.text());
+    return Buffer.from(await response.arrayBuffer());
+}
+
+async function findExistingZip(baseDir, filename) {
+    async function walk(dir) {
+        let entries;
+        try {
+            entries = await fs.readdir(dir, { withFileTypes: true });
+        } catch {
+            return null;
+        }
+        for (const entry of entries) {
+            const full = path.join(dir, entry.name);
+            if (entry.isDirectory()) {
+                const found = await walk(full);
+                if (found) return found;
+            } else if (entry.name === filename) {
+                return full;
+            }
+        }
+        return null;
+    }
+    return walk(baseDir);
+}
+
+async function exportPb(opts) {
+    const token = await getToken();
+
+    const processes = await getProcessList(token);
+    const choices = processes.map((p) => ({
+        name: `${p.process_name} (${p.document_id})`,
+        value: { guid: p.guid, document_id: p.document_id },
+    }));
+
+    const selected = await search({
+        message: 'Select PB process to export:',
+        source: async (input) => {
+            if (!input) return choices;
+            const q = input.toLowerCase();
+            return choices.filter((c) => c.name.toLowerCase().includes(q));
+        },
+    });
+
+    const { guid, document_id } = selected;
+    const filename = `${document_id}.zip`;
+
+    console.log(`Initiating export for ${document_id}...`);
+    const requestTime = Date.now();
+    await initiateExport(guid, token);
+
+    console.log('Waiting for export to complete...');
+    const requestId = await pollExportResult(guid, requestTime, token);
+
+    console.log(`Downloading ${filename}...`);
+    const zipBuffer = await downloadZip(requestId, token);
+
+    const ADDONS_PATH = resolveAddonsPath(process.env.ADDONS_PATH);
+    const processesDir = path.join(ADDONS_PATH, '.cloudbuild', 'pb', 'B2WA', 'processes');
+
+    const existing = await findExistingZip(processesDir, filename);
+    let savePath;
+    if (existing) {
+        savePath = existing;
+        await fs.writeFile(savePath, zipBuffer);
+        console.log(`Updated existing file at ${savePath}`);
+    } else {
+        savePath = path.join(processesDir, 'all', filename);
+        await fs.mkdir(path.dirname(savePath), { recursive: true });
+        await fs.writeFile(savePath, zipBuffer);
+        console.log(`Created new file at ${savePath}`);
+    }
+
+    if (opts.commit !== false) {
+        await execGit(['add', savePath], ADDONS_PATH);
+        await execGit(['commit', '-m', '[IMP][.cloudbuild] Update wizard'], ADDONS_PATH);
+        console.log('Committed.');
+    }
+}
+
+export { exportPb };

package/src/commands/init.js

@@ -116,6 +116,14 @@ async function init(completion) {
             message: 'AutoPR target branch:',
             default: existing.AUTOPR_TARGET_BRANCH ?? '15.0-dev',
         },
+        {
+            type: 'input',
+            name: 'IMPORTEXPORT_URL',
+            message: 'ImportExport URL:',
+            default:
+                existing.IMPORTEXPORT_URL ??
+                'https://sorgenia-test-02.symple.cloud/api/importexport/v1/',
+        },
     ]);
 
     if (!answers.KC_PASSWORD && existing.KC_PASSWORD) {

package/src/commands/pr.js

@@ -3,28 +3,7 @@ import fs from 'fs/promises';
 import path from 'path';
 import fetch from 'node-fetch';
 import { resolveAddonsPath } from '../lib/addons.js';
-
-async function getToken() {
-    const url = process.env.KC_URL;
-    const payload = new URLSearchParams();
-
-    payload.append('username', process.env.KC_USER);
-    payload.append('password', process.env.KC_PASSWORD);
-    payload.append('client_id', process.env.KC_ID);
-    payload.append('client_secret', process.env.KC_SECRET);
-    payload.append('grant_type', 'password');
-
-    const response = await fetch(url, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/x-www-form-urlencoded',
-        },
-        body: payload.toString(),
-    });
-
-    const json = await response.json();
-    return json.access_token;
-}
+import { getToken } from '../lib/auth.js';
 
 async function getFiles(module_name, token) {
     const url = `${process.env.RIP_URL}/ir.model/xml_prbot`;

package/src/index.js

@@ -7,6 +7,7 @@ import omelette from 'omelette';
 import { autopr } from './commands/autopr.js';
 import { changelog } from './commands/changelog.js';
 import { commit } from './commands/commit.js';
+import { exportPb, exportRip } from './commands/export.js';
 import { init } from './commands/init.js';
 import { main as prMain } from './commands/pr.js';
 import { verbot } from './commands/ver.js';
@@ -108,4 +109,32 @@ program.command('commit').action((opts) => {
     });
 });
 
+const exportCmd = program.command('export');
+
+exportCmd
+    .command('workflow <module>')
+    .option('-b, --bump <level>')
+    .action((module, opts) => {
+        prMain(module)
+            .then(() => {
+                if (opts.bump) {
+                    return verbot(module, opts.bump);
+                }
+            })
+            .catch((err) => {
+                throw err;
+            });
+    });
+
+exportCmd.command('rip').action(() => exportRip());
+
+exportCmd
+    .command('pb')
+    .option('--no-commit')
+    .action((opts) => {
+        exportPb(opts).catch((err) => {
+            throw err;
+        });
+    });
+
 program.parse();

package/src/lib/auth.js

@@ -0,0 +1,25 @@
+import fetch from 'node-fetch';
+
+async function getToken() {
+    const url = process.env.KC_URL;
+    const payload = new URLSearchParams();
+
+    payload.append('username', process.env.KC_USER);
+    payload.append('password', process.env.KC_PASSWORD);
+    payload.append('client_id', process.env.KC_ID);
+    payload.append('client_secret', process.env.KC_SECRET);
+    payload.append('grant_type', 'password');
+
+    const response = await fetch(url, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/x-www-form-urlencoded',
+        },
+        body: payload.toString(),
+    });
+
+    const json = await response.json();
+    return json.access_token;
+}
+
+export { getToken };