mirror of
https://github.com/ksyasuda/SubMiner.git
synced 2026-02-28 06:22:45 -08:00
refactor: extract texthooker and subtitle websocket services
This commit is contained in:
57
src/core/services/subtitle-ws-service.ts
Normal file
57
src/core/services/subtitle-ws-service.ts
Normal file
@@ -0,0 +1,57 @@
|
||||
import * as fs from "fs";
|
||||
import * as os from "os";
|
||||
import * as path from "path";
|
||||
import WebSocket from "ws";
|
||||
|
||||
export function hasMpvWebsocketPlugin(): boolean {
|
||||
const mpvWebsocketPath = path.join(
|
||||
os.homedir(),
|
||||
".config",
|
||||
"mpv",
|
||||
"mpv_websocket",
|
||||
);
|
||||
return fs.existsSync(mpvWebsocketPath);
|
||||
}
|
||||
|
||||
export class SubtitleWebSocketService {
|
||||
private server: WebSocket.Server | null = null;
|
||||
|
||||
public isRunning(): boolean {
|
||||
return this.server !== null;
|
||||
}
|
||||
|
||||
public start(port: number, getCurrentSubtitleText: () => string): void {
|
||||
this.server = new WebSocket.Server({ port, host: "127.0.0.1" });
|
||||
|
||||
this.server.on("connection", (ws: WebSocket) => {
|
||||
console.log("WebSocket client connected");
|
||||
const currentText = getCurrentSubtitleText();
|
||||
if (currentText) {
|
||||
ws.send(JSON.stringify({ sentence: currentText }));
|
||||
}
|
||||
});
|
||||
|
||||
this.server.on("error", (err: Error) => {
|
||||
console.error("WebSocket server error:", err.message);
|
||||
});
|
||||
|
||||
console.log(`Subtitle WebSocket server running on ws://127.0.0.1:${port}`);
|
||||
}
|
||||
|
||||
public broadcast(text: string): void {
|
||||
if (!this.server) return;
|
||||
const message = JSON.stringify({ sentence: text });
|
||||
for (const client of this.server.clients) {
|
||||
if (client.readyState === WebSocket.OPEN) {
|
||||
client.send(message);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public stop(): void {
|
||||
if (this.server) {
|
||||
this.server.close();
|
||||
this.server = null;
|
||||
}
|
||||
}
|
||||
}
|
||||
82
src/core/services/texthooker-service.ts
Normal file
82
src/core/services/texthooker-service.ts
Normal file
@@ -0,0 +1,82 @@
|
||||
import * as fs from "fs";
|
||||
import * as http from "http";
|
||||
import * as path from "path";
|
||||
|
||||
export class TexthookerService {
|
||||
private server: http.Server | null = null;
|
||||
|
||||
public isRunning(): boolean {
|
||||
return this.server !== null;
|
||||
}
|
||||
|
||||
public start(port: number): http.Server | null {
|
||||
const texthookerPath = this.getTexthookerPath();
|
||||
if (!texthookerPath) {
|
||||
console.error("texthooker-ui not found");
|
||||
return null;
|
||||
}
|
||||
|
||||
this.server = http.createServer((req, res) => {
|
||||
const urlPath = (req.url || "/").split("?")[0];
|
||||
const filePath = path.join(
|
||||
texthookerPath,
|
||||
urlPath === "/" ? "index.html" : urlPath,
|
||||
);
|
||||
|
||||
const ext = path.extname(filePath);
|
||||
const mimeTypes: Record<string, string> = {
|
||||
".html": "text/html",
|
||||
".js": "application/javascript",
|
||||
".css": "text/css",
|
||||
".json": "application/json",
|
||||
".png": "image/png",
|
||||
".svg": "image/svg+xml",
|
||||
".ttf": "font/ttf",
|
||||
".woff": "font/woff",
|
||||
".woff2": "font/woff2",
|
||||
};
|
||||
|
||||
fs.readFile(filePath, (err, data) => {
|
||||
if (err) {
|
||||
res.writeHead(404);
|
||||
res.end("Not found");
|
||||
return;
|
||||
}
|
||||
res.writeHead(200, { "Content-Type": mimeTypes[ext] || "text/plain" });
|
||||
res.end(data);
|
||||
});
|
||||
});
|
||||
|
||||
this.server.listen(port, "127.0.0.1", () => {
|
||||
console.log(`Texthooker server running at http://127.0.0.1:${port}`);
|
||||
});
|
||||
|
||||
return this.server;
|
||||
}
|
||||
|
||||
public stop(): void {
|
||||
if (this.server) {
|
||||
this.server.close();
|
||||
this.server = null;
|
||||
}
|
||||
}
|
||||
|
||||
private getTexthookerPath(): string | null {
|
||||
const searchPaths = [
|
||||
path.join(__dirname, "..", "..", "..", "vendor", "texthooker-ui", "docs"),
|
||||
path.join(
|
||||
process.resourcesPath,
|
||||
"app",
|
||||
"vendor",
|
||||
"texthooker-ui",
|
||||
"docs",
|
||||
),
|
||||
];
|
||||
for (const candidate of searchPaths) {
|
||||
if (fs.existsSync(path.join(candidate, "index.html"))) {
|
||||
return candidate;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user