her sey

apps/api/src/lib/daemon.ts

@@ -200,6 +200,11 @@ interface DaemonServiceClient extends grpc.Client {
     metadata: grpc.Metadata,
     callback: UnaryCallback<DaemonManagedDatabaseCredentialsRaw>,
   ): void;
+  importDatabaseSql(
+    request: { database_name: string; sql: string },
+    metadata: grpc.Metadata,
+    callback: UnaryCallback<EmptyResponse>,
+  ): void;
   updateDatabasePassword(
     request: { username: string; password: string },
     metadata: grpc.Metadata,
@@ -319,9 +324,7 @@ function buildGrpcTarget(fqdn: string, grpcPort: number): string {
 
   const withoutPath = host.replace(/\/.*$/, '');
   if (/^\[.+\](?::\d+)?$/.test(withoutPath)) {
-    const innerHost = withoutPath
-      .replace(/^\[/, '')
-      .replace(/\](?::\d+)?$/, '');
+    const innerHost = withoutPath.replace(/^\[/, '').replace(/\](?::\d+)?$/, '');
     return `[${innerHost}]:${grpcPort}`;
   }
   if (/^[^:]+:\d+$/.test(withoutPath)) {
@@ -340,14 +343,10 @@ function getMetadata(daemonToken: string): grpc.Metadata {
 
 function createClient(node: DaemonNodeConnection): DaemonServiceClient {
   const target = buildGrpcTarget(node.fqdn, node.grpcPort);
-  return new DaemonService(
-    target,
-    grpc.credentials.createInsecure(),
-    {
-      'grpc.max_send_message_length': MAX_GRPC_MESSAGE_BYTES,
-      'grpc.max_receive_message_length': MAX_GRPC_MESSAGE_BYTES,
-    },
-  ) as unknown as DaemonServiceClient;
+  return new DaemonService(target, grpc.credentials.createInsecure(), {
+    'grpc.max_send_message_length': MAX_GRPC_MESSAGE_BYTES,
+    'grpc.max_receive_message_length': MAX_GRPC_MESSAGE_BYTES,
+  }) as unknown as DaemonServiceClient;
 }
 
 function waitForReady(client: grpc.Client, timeoutMs: number): Promise<void> {
@@ -444,9 +443,7 @@ interface DaemonRequestTimeoutOptions {
   rpcTimeoutMs?: number;
 }
 
-export async function daemonGetNodeStatus(
-  node: DaemonNodeConnection,
-): Promise<DaemonNodeStatus> {
+export async function daemonGetNodeStatus(node: DaemonNodeConnection): Promise<DaemonNodeStatus> {
   const client = createClient(node);
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
@@ -466,9 +463,7 @@ export async function daemonGetNodeStatus(
   }
 }
 
-export async function daemonGetNodeStats(
-  node: DaemonNodeConnection,
-): Promise<DaemonNodeStats> {
+export async function daemonGetNodeStats(node: DaemonNodeConnection): Promise<DaemonNodeStats> {
   const client = createClient(node);
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
@@ -511,7 +506,8 @@ export async function daemonDeleteServer(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     await callUnary<EmptyResponse>(
-      (callback) => client.deleteServer({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.deleteServer({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );
   } finally {
@@ -593,6 +589,30 @@ export async function daemonUpdateDatabasePassword(
   }
 }
 
+export async function daemonImportDatabaseSql(
+  node: DaemonNodeConnection,
+  request: { databaseName: string; sql: string },
+): Promise<void> {
+  const client = createClient(node);
+  try {
+    await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
+    await callUnary<EmptyResponse>(
+      (callback) =>
+        client.importDatabaseSql(
+          {
+            database_name: request.databaseName,
+            sql: request.sql,
+          },
+          getMetadata(node.daemonToken),
+          callback,
+        ),
+      DEFAULT_RPC_TIMEOUT_MS,
+    );
+  } finally {
+    client.close();
+  }
+}
+
 export async function daemonDeleteDatabase(
   node: DaemonNodeConnection,
   request: { databaseName: string; username: string },
@@ -626,7 +646,12 @@ export async function daemonSetPowerState(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     await callUnary<EmptyResponse>(
-      (callback) => client.setPowerState({ uuid: serverUuid, action: POWER_ACTIONS[action] }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.setPowerState(
+          { uuid: serverUuid, action: POWER_ACTIONS[action] },
+          getMetadata(node.daemonToken),
+          callback,
+        ),
       POWER_RPC_TIMEOUT_MS,
     );
   } finally {
@@ -643,7 +668,8 @@ export async function daemonGetServerStatus(
   try {
     await waitForReady(client, timeouts.connectTimeoutMs ?? DEFAULT_CONNECT_TIMEOUT_MS);
     return await callUnary<DaemonStatusResponse>(
-      (callback) => client.getServerStatus({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.getServerStatus({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
       timeouts.rpcTimeoutMs ?? DEFAULT_RPC_TIMEOUT_MS,
     );
   } finally {
@@ -688,7 +714,8 @@ export async function daemonSendCommand(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     await callUnary<EmptyResponse>(
-      (callback) => client.sendCommand({ uuid: serverUuid, command }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.sendCommand({ uuid: serverUuid, command }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );
   } finally {
@@ -705,7 +732,8 @@ export async function daemonListFiles(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     const response = await callUnary<DaemonFileListResponseRaw>(
-      (callback) => client.listFiles({ uuid: serverUuid, path }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.listFiles({ uuid: serverUuid, path }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );
 
@@ -731,7 +759,8 @@ export async function daemonReadFile(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     const response = await callUnary<DaemonFileContentRaw>(
-      (callback) => client.readFile({ uuid: serverUuid, path }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.readFile({ uuid: serverUuid, path }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );
 
@@ -756,7 +785,11 @@ export async function daemonWriteFile(
     await callUnary<EmptyResponse>(
       (callback) =>
         client.writeFile(
-          { uuid: serverUuid, path, data: typeof data === 'string' ? Buffer.from(data, 'utf8') : data },
+          {
+            uuid: serverUuid,
+            path,
+            data: typeof data === 'string' ? Buffer.from(data, 'utf8') : data,
+          },
           getMetadata(node.daemonToken),
           callback,
         ),
@@ -776,7 +809,8 @@ export async function daemonDeleteFiles(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     await callUnary<EmptyResponse>(
-      (callback) => client.deleteFiles({ uuid: serverUuid, paths }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.deleteFiles({ uuid: serverUuid, paths }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );
   } finally {
@@ -870,7 +904,8 @@ export async function daemonGetActivePlayers(
   try {
     await waitForReady(client, DEFAULT_CONNECT_TIMEOUT_MS);
     const response = await callUnary<DaemonPlayerListRaw>(
-      (callback) => client.getActivePlayers({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
+      (callback) =>
+        client.getActivePlayers({ uuid: serverUuid }, getMetadata(node.daemonToken), callback),
       DEFAULT_RPC_TIMEOUT_MS,
     );