lib/strfry.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

import { createServer, connect } from "net";
import { randomUUID } from "crypto";

const exec = async (cmd) =>
  new Promise((resolve, reject) => {
    const c = `/app/strfry ${cmd} 2>/dev/null`;
    const id = randomUUID();
    const resultSocketPath = `/sockets/result_${id}`;

    const resultServer = createServer((socket) => {
      let resultBuffer = "";

      const results = [];
      socket.on("data", (data) => {
        resultBuffer += data.toString();

        const parts = resultBuffer.split("\n");

        for (let i = 0; i < parts.length - 1; i++) {
          try {
            const parsedObject = JSON.parse(parts[i]);
            results.push(parsedObject);
          } catch (e) {
            console.error(parts[i]);
          }
        }

        resultBuffer = parts[parts.length - 1];
      });

      socket.on("end", () => {
        socket.write("DONE");
        resolve(results);
        socket.end();

        resultServer.close();
      });
    });

    resultServer.listen(resultSocketPath, () => {
      const controlClient = connect("/sockets/ctrl", () => {
        const message = `${resultSocketPath} ${c}\n`;
        controlClient.write(message);
        controlClient.end();
      });

      controlClient.on("error", (e) =>
        reject(`Control socket error: ${e.message}`),
      );
    });

    resultServer.on("error", (e) =>
      reject(`Result server error: ${e.message}`),
    );
  });

export const scan = (f) => exec(`scan '${JSON.stringify(f)}'\n`);
export const sync = (r, f) =>
  exec(`sync ${r} --dir down --filter '${JSON.stringify(f)}'\n`);