Merge remote-tracking branch 'sabonerune/refactor/engineInfo' into 指定…

…されているVVPPがなかったら確認後にインストールし、使えるようにするようにする

src/backend/browser/contract.ts

@@ -1,11 +1,18 @@
 import { loadEnvEngineInfos } from "../common/envEngineInfoSchema";
-import { EngineInfo } from "@/type/preload";
+import { type EngineInfo } from "@/type/preload";
 
 const baseEngineInfo = loadEnvEngineInfos()[0];
 
-export const defaultEngine: EngineInfo = {
-  ...baseEngineInfo,
-  type: "path", // FIXME: ダミーで"path"にしているので、エンジンAPIのURLを設定できるようにし、type: "URL"にする
-  isDefault: true,
-};
+export const defaultEngine: EngineInfo = (() => {
+  const { protocol, hostname, port, pathname } = new URL(baseEngineInfo.host);
+  return {
+    ...baseEngineInfo,
+    protocol,
+    hostname,
+    defaultPort: port,
+    pathname: pathname === "/" ? "" : pathname,
+    type: "path", // FIXME: ダミーで"path"にしているので、エンジンAPIのURLを設定できるようにし、type: "URL"にする
+    isDefault: true,
+  };
+})();
 export const directoryHandleStoreKey = "directoryHandle";

src/backend/electron/engineAndVvppController.ts

@@ -147,7 +147,10 @@ export class EngineAndVvppController {
     this.configManager.set("engineSettings", engineSettings);
 
     await this.engineProcessManager.runEngineAll();
-    this.runtimeInfoManager.setEngineInfos(engineInfos);
+    this.runtimeInfoManager.setEngineInfos(
+      engineInfos,
+      this.engineInfoManager.altPortInfos,
+    );
     await this.runtimeInfoManager.exportFile();
   }
 
@@ -162,6 +165,7 @@ export class EngineAndVvppController {
     // TODO: setからexportの処理は排他処理にしたほうがより良い
     this.runtimeInfoManager.setEngineInfos(
       this.engineInfoManager.fetchEngineInfos(),
+      this.engineInfoManager.altPortInfos,
     );
     await this.runtimeInfoManager.exportFile();
   }

src/backend/electron/manager/RuntimeInfoManager.ts

@@ -6,14 +6,15 @@
 import fs from "fs";
 import AsyncLock from "async-lock";
 import log from "electron-log/main";
+import type { AltPortInfos } from "@/store/type";
 import { EngineId, EngineInfo } from "@/type/preload";
 
 /**
  * ランタイム情報書き出しに必要なEngineInfo
  */
 export type EngineInfoForRuntimeInfo = Pick<
   EngineInfo,
-  "uuid" | "host" | "name"
+  "uuid" | "protocol" | "hostname" | "defaultPort" | "pathname" | "name"
 >;
 
 /**
@@ -58,12 +59,17 @@ export class RuntimeInfoManager {
    * エンジン情報（書き出し用に記憶）
    */
   private engineInfos: EngineInfoForRuntimeInfo[] = [];
+  private altportInfos: AltPortInfos = {};
 
   /**
    * エンジン情報を登録する
    */
-  public setEngineInfos(engineInfos: EngineInfoForRuntimeInfo[]) {
+  public setEngineInfos(
+    engineInfos: EngineInfoForRuntimeInfo[],
+    altportInfos: AltPortInfos,
+  ) {
     this.engineInfos = engineInfos;
+    this.altportInfos = altportInfos;
   }
 
   /**
@@ -76,9 +82,15 @@ export class RuntimeInfoManager {
         formatVersion: this.fileFormatVersion,
         appVersion: this.appVersion,
         engineInfos: this.engineInfos.map((engineInfo) => {
+          const altPort: string | undefined =
+            this.altportInfos[engineInfo.uuid];
+          const port = altPort ?? engineInfo.defaultPort;
+          const url = new URL(
+            `${engineInfo.protocol}//${engineInfo.hostname}:${port}`,
+          );
           return {
             uuid: engineInfo.uuid,
-            url: engineInfo.host, // NOTE: 元のEngineInfo.hostにURLが入っている
+            url: `${url.origin}${engineInfo.pathname}`,
             name: engineInfo.name,
           };
         }),

src/backend/electron/manager/engineInfoManager.ts

@@ -33,8 +33,13 @@ function fetchDefaultEngineInfos(
     .filter((engineInfo) => engineInfo.type != "downloadVvpp")
     .map((engineInfo) => {
       if (engineInfo.type == "downloadVvpp") throw new UnreachableError();
+      const { protocol, hostname, port, pathname } = new URL(engineInfo.host);
       return {
         ...engineInfo,
+        protocol,
+        hostname,
+        defaultPort: port,
+        pathname: pathname === "/" ? "" : pathname,
         isDefault: true,
         type: engineInfo.type,
         executionFilePath: path.resolve(engineInfo.executionFilePath),
@@ -91,7 +96,10 @@ export class EngineInfoManager {
 
       engines.push({
         uuid: manifest.uuid,
-        host: `http://127.0.0.1:${manifest.port}`,
+        protocol: "http:",
+        hostname: "127.0.0.1",
+        defaultPort: manifest.port.toString(),
+        pathname: "",
         name: manifest.name,
         path: engineDir,
         executionEnabled: true,
@@ -146,6 +154,7 @@ export class EngineInfoManager {
       ...fetchDefaultEngineInfos(this.envEngineInfos, this.defaultEngineDir),
       ...this.fetchAdditionalEngineInfos(),
     ];
+
     // 追加エンジンがダウンロードしたデフォルトエンジンと同じなら、デフォルトエンジンとして扱う
     const targetEngineUuids = this.envEngineInfos
       .filter((e) => e.type == "downloadVvpp")
@@ -161,15 +170,6 @@ export class EngineInfoManager {
       }
     }
 
-    // 代替ポートに置き換える
-    engineInfos.forEach((engineInfo) => {
-      const altPortInfo = this.altPortInfos[engineInfo.uuid];
-      if (altPortInfo) {
-        const url = new URL(engineInfo.host);
-        url.port = altPortInfo.to.toString();
-        engineInfo.host = url.origin;
-      }
-    });
     return engineInfos;
   }
 
@@ -220,12 +220,7 @@ export class EngineInfoManager {
    * エンジン起動時にポートが競合して代替ポートを使う場合に使用する。
    */
   updateAltPort(engineId: EngineId, port: number) {
-    const engineInfo = this.fetchEngineInfo(engineId);
-    const url = new URL(engineInfo.host);
-    this.altPortInfos[engineId] = {
-      from: Number(url.port),
-      to: port,
-    };
+    this.altPortInfos[engineId] = port.toString();
   }
 
   /**

src/backend/electron/manager/engineProcessManager.ts

@@ -9,8 +9,8 @@ import {
   findAltPort,
   getPidFromPort,
   getProcessNameFromPid,
+  type HostInfo,
   isAssignablePort,
-  url2HostInfo,
 } from "../portHelper";
 
 import { EngineInfo, EngineId, EngineSettings } from "@/type/preload";
@@ -86,7 +86,11 @@ export class EngineProcessManager {
     }
 
     // { hostname (localhost), port (50021) } <- url (http://localhost:50021)
-    const engineHostInfo = url2HostInfo(new URL(engineInfo.host));
+    const engineHostInfo: HostInfo = {
+      protocol: engineInfo.protocol,
+      hostname: engineInfo.hostname,
+      port: Number(engineInfo.defaultPort),
+    };
 
     // ポートが塞がっていれば代替ポートを探す
     let port = engineHostInfo.port;
@@ -152,7 +156,7 @@ export class EngineProcessManager {
     const enginePath = engineInfo.executionFilePath;
     const args = engineInfo.executionArgs.concat(useGpu ? ["--use_gpu"] : [], [
       "--host",
-      new URL(engineInfo.host).hostname,
+      engineHostInfo.hostname,
       "--port",
       port.toString(),
     ]);

src/backend/electron/portHelper.ts

@@ -17,14 +17,6 @@ const portWarn = (port: number, message: string, isNested = false) => {
   log.warn(`${isNested ? "| " : ""}PORT ${port}: ${message}`);
 };
 
-export function url2HostInfo(url: URL): HostInfo {
-  return {
-    protocol: url.protocol,
-    hostname: url.hostname,
-    port: Number(url.port),
-  };
-}
-
 /**
  * "netstat -ano" の stdout から, 指定したポートを LISTENING しているプロセスの id を取得します.
  *

src/components/Menu/MenuBar/MenuBar.vue

@@ -54,7 +54,7 @@ const currentVersion = ref("");
 const audioKeys = computed(() => store.state.audioKeys);
 
 // デフォルトエンジンの代替先ポート
-const defaultEngineAltPortTo = computed<number | undefined>(() => {
+const defaultEngineAltPortTo = computed<string | undefined>(() => {
   const altPortInfos = store.state.altPortInfos;
 
   // ref: https://github.com/VOICEVOX/voicevox/blob/32940eab36f4f729dd0390dca98f18656240d60d/src/views/EditorHome.vue#L522-L528
@@ -65,7 +65,7 @@ const defaultEngineAltPortTo = computed<number | undefined>(() => {
 
   // <defaultEngineId>: { from: number, to: number } -> to (代替先ポート)
   if (defaultEngineInfo.uuid in altPortInfos) {
-    return altPortInfos[defaultEngineInfo.uuid].to;
+    return altPortInfos[defaultEngineInfo.uuid];
   } else {
     return undefined;
   }

src/components/Talk/TalkEditor.vue

@@ -554,11 +554,12 @@ watch(
     // 代替ポートをトースト通知する
     for (const engineId of store.state.engineIds) {
       const engineName = store.state.engineInfos[engineId].name;
+      const defaultPort = store.state.engineInfos[engineId].defaultPort;
       const altPort = store.state.altPortInfos[engineId];
       if (!altPort) return;
 
       void store.actions.SHOW_NOTIFY_AND_NOT_SHOW_AGAIN_BUTTON({
-        message: `${altPort.from}番ポートが使用中であるため ${engineName} は、${altPort.to}番ポートで起動しました`,
+        message: `${defaultPort}番ポートが使用中であるため ${engineName} は、${altPort}番ポートで起動しました`,
         icon: "compare_arrows",
         tipName: "engineStartedOnAltPort",
       });

src/store/proxy.ts

@@ -20,7 +20,14 @@ const proxyStoreCreator = (_engineFactory: IEngineConnectorFactory) => {
             new Error(`No such engineInfo registered: engineId == ${engineId}`),
           );
 
-        const instance = _engineFactory.instance(engineInfo.host);
+        const altPort: string | undefined = state.altPortInfos[engineId];
+        const port = altPort ?? engineInfo.defaultPort;
+        const url = new URL(
+          `${engineInfo.protocol}//${engineInfo.hostname}:${port}`,
+        );
+        const instance = _engineFactory.instance(
+          `${url.origin}${engineInfo.pathname}`,
+        );
         return Promise.resolve({
           invoke: (v) => (arg) =>
             // FIXME: anyを使わないようにする

src/store/type.ts

@@ -105,7 +105,7 @@ export type Command = {
 export type EngineState = "STARTING" | "FAILED_STARTING" | "ERROR" | "READY";
 
 // ポートが塞がれていたときの代替ポート情報
-export type AltPortInfos = Record<EngineId, { from: number; to: number }>;
+export type AltPortInfos = Record<EngineId, string>;
 
 export type SaveResult =
   | "SUCCESS"

src/type/preload.ts

@@ -402,7 +402,10 @@ export type MinimumEngineManifestType = z.infer<
 
 export type EngineInfo = {
   uuid: EngineId;
-  host: string; // NOTE: 実際はorigin（プロトコルとhostnameとport）が入る
+  protocol: string;
+  hostname: string;
+  defaultPort: string;
+  pathname: string;
   name: string;
   path?: string; // エンジンディレクトリのパス
   executionEnabled: boolean;

tests/unit/backend/electron/RuntimeInfo.spec.ts

@@ -14,18 +14,35 @@ test("想定通りのラインタイム情報が保存されている", async ()
   const runtimeInfoManager = new RuntimeInfoManager(tempFilePath, appVersion);
 
   // エンジン情報
-  runtimeInfoManager.setEngineInfos([
-    {
-      uuid: EngineId("00000000-0000-0000-0000-000000000001"),
-      host: "https://example.com/engine1",
-      name: "engine1",
-    },
-    {
-      uuid: EngineId("00000000-0000-0000-0000-000000000002"),
-      host: "https://example.com/engine2",
-      name: "engine2",
-    },
-  ]);
+  runtimeInfoManager.setEngineInfos(
+    [
+      {
+        uuid: EngineId("00000000-0000-0000-0000-000000000001"),
+        protocol: "https:",
+        hostname: "example.com",
+        defaultPort: "",
+        pathname: "/engine1",
+        name: "engine1",
+      },
+      {
+        uuid: EngineId("00000000-0000-0000-0000-000000000002"),
+        protocol: "https:",
+        hostname: "example.com",
+        defaultPort: "",
+        pathname: "/engine2",
+        name: "engine2",
+      },
+      {
+        uuid: EngineId("00000000-0000-0000-0000-000000000003"),
+        protocol: "http:",
+        hostname: "127.0.0.1",
+        defaultPort: "8080",
+        pathname: "",
+        name: "engine3",
+      },
+    ],
+    { [EngineId("00000000-0000-0000-0000-000000000003")]: "8081" },
+  );
 
   // ファイル書き出し
   await runtimeInfoManager.exportFile();
@@ -49,6 +66,11 @@ test("想定通りのラインタイム情報が保存されている", async ()
           "url": "https://example.com/engine2",
           "uuid": "00000000-0000-0000-0000-000000000002",
         },
+        {
+          "name": "engine3",
+          "url": "http://127.0.0.1:8081",
+          "uuid": "00000000-0000-0000-0000-000000000003",
+        },
       ],
       "formatVersion": 1,
     }