import { Service, ServiceBroker, Context } from "moleculer"; import ApiGateway from "moleculer-web"; import chokidar from "chokidar"; import path from "path"; import fs from "fs"; import { IExtractionOptions, IFolderData } from "threetwo-ui-typings"; import { SocketIOMixin } from "../mixins/socket.io.mixin"; export const io = SocketIOMixin(); export default class ApiService extends Service { public constructor(broker: ServiceBroker) { super(broker); this.parseServiceSchema({ name: "api", mixins: [ApiGateway, SocketIOMixin], // More info about settings: https://moleculer.services/docs/0.14/moleculer-web.html settings: { port: process.env.PORT || 3000, routes: [ { path: "/api", whitelist: ["**"], cors: { origin: "*", methods: [ "GET", "OPTIONS", "POST", "PUT", "DELETE", ], allowedHeaders: ["*"], exposedHeaders: [], credentials: false, maxAge: 3600, }, use: [], mergeParams: true, authentication: false, authorization: false, autoAliases: true, aliases: {}, callingOptions: {}, bodyParsers: { json: { strict: false, limit: "1MB", }, urlencoded: { extended: true, limit: "1MB", }, }, mappingPolicy: "all", // Available values: "all", "restrict" logging: true, }, { path: "/userdata", use: [ApiGateway.serveStatic("userdata")], }, { path: "/comics", use: [ApiGateway.serveStatic("comics")], }, { path: "/logs", use: [ApiGateway.serveStatic("logs")], }, ], log4XXResponses: false, logRequestParams: true, logResponseData: true, assets: { folder: "public", // Options to `server-static` module options: {}, }, }, events: { "**"(payload, sender, event) { if (io) io.emit("event", { sender, event, payload, }); }, }, methods: {}, started(): any { // Add a connect listener io.on("connection", (client) => { console.log("Client connected via websocket!"); client.on("action", async (action) => { switch (action.type) { case "LS_IMPORT": // 1. Send task to queue console.log(`Recieved ${action.type} event.`) await this.broker.call( "library.newImport", action.data, {} ); break; case "LS_TOGGLE_IMPORT_QUEUE": console.log("lol") await this.broker.call("queue.toggleImportQueue", action.data, {}); break; } }); // Add a disconnect listener client.on("disconnect", () => { console.log("Client disconnected"); }); // Filewatcher const fileWatcher = chokidar.watch( path.resolve("./comics"), { ignored: /(^|[\/\\])\../, // ignore dotfiles persistent: true, usePolling: true, ignoreInitial: true, atomic: true, depth: 10, awaitWriteFinish: { stabilityThreshold: 2000, pollInterval: 100, }, } ); const fileCopyDelaySeconds = 30; const checkFileCopyComplete = (path, previousPath) => { fs.stat(path, async (err, stat) => { if (err) { throw err; } if ( stat.mtime.getTime() === previousPath.mtime.getTime() ) { console.log( "File detected, starting import..." ); const walkedFolder: IFolderData = await broker.call("library.walkFolders", { basePathToWalk: path, }); await this.broker.call("queue.processImport", { fileObject: { filePath: walkedFolder[0].filePath, fileSize: walkedFolder[0].fileSize, }, }); } else { setTimeout( checkFileCopyComplete, fileCopyDelaySeconds * 1000, path, stat ); } }); }; fileWatcher .on("add", async (path, stats) => { console.log("Watcher detected new files."); console.log( `File ${path} has been added with stats: ${JSON.stringify( stats )}` ); console.log("File copy started..."); fs.stat(path, function (err, stat) { if (err) { console.log( "Error watching file for copy completion. ERR: " + err.message ); console.log( "Error file not processed. PATH: " + path ); throw err; } setTimeout( checkFileCopyComplete, fileCopyDelaySeconds * 1000, path, stat ); client.emit("action", { type: "LS_COMIC_ADDED", result: path, }); }); }) .on("change", (path, stats) => console.log( `File ${path} has been changed. Stats: ${stats}` ) ) .on("unlink", (path) => console.log(`File ${path} has been removed`) ) .on("addDir", (path) => console.log(`Directory ${path} has been added`) ); }); }, }); } }