migrate podsizes

Author: mms-gianni

…ations/20250702090708_init/migration.sql …ations/20250715160026_init/migration.sqlserver/prisma/migrations/20250702090708_init/migration.sql renamed to server/prisma/migrations/20250715160026_init/migration.sql server/prisma/migrations/20250702090708_init/migration.sql renamed to server/prisma/migrations/20250715160026_init/migration.sql

@@ -100,6 +100,19 @@ CREATE TABLE "Runpack" (
     CONSTRAINT "Runpack_runId_fkey" FOREIGN KEY ("runId") REFERENCES "RunpackPhase" ("id") ON DELETE RESTRICT ON UPDATE CASCADE
 );
 
+-- CreateTable
+CREATE TABLE "PodSize" (
+    "id" TEXT NOT NULL PRIMARY KEY,
+    "name" TEXT NOT NULL,
+    "cpuLimit" TEXT NOT NULL,
+    "memoryLimit" TEXT NOT NULL,
+    "cpuRequest" TEXT NOT NULL,
+    "memoryRequest" TEXT NOT NULL,
+    "description" TEXT,
+    "createdAt" DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updatedAt" DATETIME NOT NULL
+);
+
 -- CreateTable
 CREATE TABLE "RunpackPhase" (
     "id" TEXT NOT NULL PRIMARY KEY,

server/prisma/schema.prisma

@@ -174,6 +174,18 @@ model Runpack {
   updatedAt   DateTime       @updatedAt
 }
 
+model PodSize {
+  id          String   @id @default(cuid())
+  name        String
+  cpuLimit    String
+  memoryLimit String
+  cpuRequest  String
+  memoryRequest String
+  description String?
+  createdAt   DateTime @default(now())
+  updatedAt   DateTime @updatedAt
+}
+
 model RunpackPhase {
   id                   String                @id @default(cuid())
   repository           String

server/src/config/config.service.ts

@@ -635,17 +635,23 @@ export class ConfigService {
   }
 
   public async getPodSizes() {
-    const podSizeList: PodSize[] = [];
-
-    const namespace = process.env.KUBERO_NAMESPACE || 'kubero';
-    const kuberoes = await this.kubectl.getKuberoConfig(namespace);
-
-    for (const podSize of kuberoes.spec.kubero.config.podSizeList) {
-      const p = new PodSize(podSize);
-      podSizeList.push(p);
-    }
-
-    return podSizeList;
+    // Fetch PodSizes from the database using Prisma
+    const dbPodSizes = await this.prisma.podSize.findMany();
+    // Map DB results to PodSize class
+    return dbPodSizes.map((ps: any) => new PodSize({
+      name: ps.name,
+      description: ps.description,
+      resources: {
+        requests: {
+          memory: ps.memoryRequest || '',
+          cpu: ps.cpuLimit || '',
+        },
+        limits: {
+          memory: ps.memoryLimit || '',
+          cpu: ps.cpuLimit || '',
+        },
+      },
+    }));
   }
 
   public static getLocalauthEnabled(): boolean {

server/src/database/database.service.ts

@@ -3,6 +3,7 @@ import { PrismaClient } from '@prisma/client';
 import * as crypto from 'crypto';
 import * as bcrypt from 'bcrypt';
 import { runpacks } from './runpacks.seed'; // Assuming runpacks.seed.ts exports a Runpack type
+import { podsizes } from './podsizes.seed'; // Assuming podsizes.seed.ts exports a PodSize type
 import * as yaml from 'yaml'; // Import yaml for parsing runpacks
 
 @Injectable()
@@ -35,6 +36,7 @@ export class DatabaseService {
       });
 
     this.seedRunpacks();
+    this.seedPodSizes();
   }
 
   private async init() {
@@ -354,20 +356,6 @@ export class DatabaseService {
   }
 
   private async seedRunpacks() {
-    /*
-    // Seed runpacks from ./runpacks.seed.yaml
-    const fs = await import('fs');
-    const yaml = await import('yaml');
-    const path = require('path');
-    const configPath = path.resolve(__dirname, '../runpacks.seed.yaml');
-    if (!fs.existsSync(configPath)) {
-      this.logger.warn('runpacks.seed.yaml not found, skipping runpack seed.');
-      return;
-    }
-    const file = fs.readFileSync(configPath, 'utf8');
-    const config = yaml.parse(file);
-    */
-
     const config = yaml.parse(runpacks);
 
     const buildpacks = config || [];
@@ -424,4 +412,28 @@ export class DatabaseService {
     }
     this.logger.log('Buildpacks/Runpacks seeded successfully.');
   }
+
+  private async seedPodSizes() {
+    const config = yaml.parse(podsizes);
+    // seed pod sizes if the table is empty
+    const existingSizes = await this.prisma.podSize.count();
+    if (existingSizes > 0) {
+      this.logger.log('Pod sizes already exist. Skipping seeding.');
+      return;
+    }
+    for (const size of config) {
+      await this.prisma.podSize.create({
+        data: {
+          name: size.name,
+          description: size.description,
+          cpuLimit: size.resources.limits.cpu,
+          memoryLimit: size.resources.limits.memory,
+          cpuRequest: size.resources.requests.cpu,
+          memoryRequest: size.resources.requests.memory,
+        },
+      });
+      this.logger.log(`Pod size '${size.name}' seeded successfully.`);
+    }
+    this.logger.log('Pod sizes seeded successfully.');
+  }
 }

server/src/database/podsizes.seed.ts

@@ -0,0 +1,31 @@
+export const podsizes = `
+  - name: small
+    description: 'Small (CPU: 0.25, Memory: 0.5Gi)'
+    default: true
+    resources:
+      requests:
+        memory: 0.5Gi
+        cpu: 250m
+      limits:
+        memory: 1Gi
+        cpu: 500m
+  - name: medium
+    description: 'Medium (CPU: 1, Memory: 2Gi)'
+    resources:
+      requests:
+        memory: 2Gi
+        cpu: 1000m
+      limits:
+        memory: 4Gi
+        cpu: 2000m
+  - name: large
+    description: 'Large (CPU: 2, Memory: 4Gi)'
+    active: false
+    resources:
+      requests:
+        memory: 4Gi
+        cpu: 2000m
+      limits:
+        memory: 4Gi
+        cpu: 2000m
+`