Dbschema migr -- command side

examples/hello/dbos-config.yaml

@@ -12,4 +12,6 @@ database:
   system_database: 'hello_systemdb'
   connectionTimeoutMillis: 3000
   user_dbclient: 'knex'
+  migrate: ['migrate:latest']
+  rollback: ['migrate:rollback']
 version: "1.0"

src/cloud-cli/cli.ts

@@ -12,7 +12,7 @@ import {
 import { Command } from 'commander';
 import { login } from "./login";
 import { registerUser } from "./register";
-import { createUserDb, getUserDb, deleteUserDb } from "./userdb";
+import { createUserDb, getUserDb, deleteUserDb, migrate } from "./userdb";
 import { credentialsExist } from "./utils";
 
 const program = new Command();
@@ -174,6 +174,12 @@ userdb
     await deleteUserDb(host, port, dbname, options.sync)
   }))
 
+  userdb
+  .command('migrate')
+  .action((() => {
+    migrate()
+  }))
+
 program.parse(process.argv);
 
 // If no arguments provided, display help by default

src/cloud-cli/userdb.ts

@@ -2,6 +2,8 @@ import axios from "axios";
 import { createGlobalLogger } from "../telemetry/logs";
 import { getCloudCredentials } from "./utils";
 import { sleep } from "../utils"
+import { ConfigFile, loadConfigFile, dbosConfigFilePath } from "../dbos-runtime/config";
+import { execSync } from "child_process";
 
 export async function createUserDb(host: string, port: string, dbName: string, adminName: string, adminPassword: string, sync: boolean) {
   const logger = createGlobalLogger();
@@ -42,7 +44,7 @@ export async function createUserDb(host: string, port: string, dbName: string, a
       logger.info("Saving db state to cloud db");
       await axios.put(`http://${host}:${port}/${userCredentials.userName}/databases/userdb/info`, 
       // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
-      {"Name": dbName,"Status": status, "HostName": dbhostname, "Port": dbport},
+      {"DBName": dbName,"Status": status, "HostName": dbhostname, "Port": dbport},
       {
         headers: {
         "Content-Type": "application/json",
@@ -133,6 +135,59 @@ export async function getUserDb(host: string, port: string, dbName: string) {
   }
 }
 
+export function migrate() {
+  const logger = createGlobalLogger();
+
+  // read the yaml file
+  const configFile: ConfigFile | undefined = loadConfigFile(dbosConfigFilePath);
+  if (!configFile) {
+    logger.error(`failed to parse ${dbosConfigFilePath}`);
+    return;
+  }
+
+  let create_db = configFile.database.create_db;
+  if (create_db == undefined) {
+    create_db = false;
+  }
+
+  const userdbname = configFile.database.user_database
+
+  if (create_db) {
+    logger.info(`Creating database ${userdbname}`);
+    let cmd = `createdb -h ${configFile.database.hostname} -p ${configFile.database.port} ${userdbname} -U ${configFile.database.username} -w ${configFile.database.password} -e`;
+    logger.info(cmd);
+    try {
+      execSync(cmd);
+    } catch(e) {
+      logger.error("Database already exists or could not be created.");
+    }
+  }
+
+  let dbType = configFile.database.user_dbclient;
+  if (dbType == undefined) {
+    dbType = 'knex';
+  }
+
+  const migratecommands = configFile.database.migrate;
+  const rollbackcommands = configFile.database.rollback;
+
+  try {
+    migratecommands?.forEach((cmd) => {
+        const command = "npx "+ dbType + " " + cmd; 
+        logger.info("Executing " + command);
+        execSync(command);
+    })
+  } catch(e) {
+    logger.error("Error running migration");
+    rollbackcommands?.forEach((cmd) => {
+      const command = "npx " + dbType + " " + cmd; 
+        logger.info("Executing " + command);
+        execSync(command);
+    })
+
+  }
+}
+
 // eslint-disable-next-line @typescript-eslint/no-unsafe-return, @typescript-eslint/no-explicit-any
 async function getDb(host: string, port: string, dbName: string) : Promise<any> {
 

src/dbos-runtime/config.ts

@@ -25,6 +25,9 @@ export interface ConfigFile {
     observability_database: string;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     user_dbclient?: UserDatabaseName;
+    create_db: boolean;
+    migrate?: string[];
+    rollback?: string[];
   };
   telemetry?: TelemetryConfig;
   // eslint-disable-next-line @typescript-eslint/no-explicit-any