Hookup disk and nic creation on instance create

Author: just-be-dev

libs/api-mocks/msw/handlers.ts

@@ -31,6 +31,7 @@ import {
 import {
   NotImplemented,
   errIfExists,
+  errIfInvalidDiskSize,
   getStartAndEndTime,
   getTimestamps,
   paginated,
@@ -156,6 +157,7 @@ export const handlers = makeHandlers({
     const project = lookupProject(params.path)
 
     errIfExists(db.disks, { name: body.name, project_id: project.id })
+    errIfInvalidDiskSize(params.path, body)
 
     const { name, description, size, disk_source } = body
     const newDisk: Json<Api.Disk> = {
@@ -243,8 +245,69 @@ export const handlers = makeHandlers({
 
     errIfExists(db.instances, { name: body.name, project_id: project.id })
 
+    const instanceId = uuid()
+
+    for (const diskParams of body.disks || []) {
+      if (diskParams.type === 'create') {
+        errIfExists(db.disks, { name: diskParams.name, project_id: project.id })
+        errIfInvalidDiskSize(params.path, diskParams)
+        const { size, name, description, disk_source } = diskParams
+        const newDisk: Json<Api.Disk> = {
+          id: uuid(),
+          name,
+          description,
+          size,
+          project_id: project.id,
+          state: { state: 'attached', instance: instanceId },
+          device_path: '/mnt/disk',
+          block_size: disk_source.type === 'blank' ? disk_source.block_size : 4096,
+          ...getTimestamps(),
+        }
+        db.disks.push(newDisk)
+      } else {
+        const disk = lookupDisk({ ...params.path, diskName: diskParams.name })
+        disk.state = { state: 'attached', instance: instanceId }
+      }
+    }
+
+    if (body.network_interfaces?.type === 'default') {
+      db.networkInterfaces.push({
+        id: uuid(),
+        description: 'The default network interface',
+        instance_id: instanceId,
+        primary: true,
+        mac: '00:00:00:00:00:00',
+        ip: '127.0.0.1',
+        name: 'default',
+        vpc_id: uuid(),
+        subnet_id: uuid(),
+        ...getTimestamps(),
+      })
+    } else if (body.network_interfaces?.type === 'create') {
+      body.network_interfaces.params.forEach(
+        ({ name, description, ip, subnet_name, vpc_name }, i) => {
+          db.networkInterfaces.push({
+            id: uuid(),
+            name,
+            description,
+            instance_id: instanceId,
+            primary: i === 0 ? true : false,
+            mac: '00:00:00:00:00:00',
+            ip: ip || '127.0.0.1',
+            vpc_id: lookupVpc({ ...params.path, vpcName: vpc_name }).id,
+            subnet_id: lookupVpcSubnet({
+              ...params.path,
+              vpcName: vpc_name,
+              subnetName: subnet_name,
+            }).id,
+            ...getTimestamps(),
+          })
+        }
+      )
+    }
+
     const newInstance: Json<Api.Instance> = {
-      id: uuid(),
+      id: instanceId,
       project_id: project.id,
       ...pick(body, 'name', 'description', 'hostname', 'memory', 'ncpus'),
       ...getTimestamps(),

libs/api-mocks/msw/util.ts

@@ -1,6 +1,11 @@
 import { subHours } from 'date-fns'
 
+import type { DiskCreate, DiskCreatePathParams } from '@oxide/gen/Api'
+import type { Json } from '@oxide/gen/msw-handlers'
 import { json } from '@oxide/gen/msw-handlers'
+import { GiB } from '@oxide/util'
+
+import { db } from './db'
 
 export { json } from '@oxide/gen/msw-handlers'
 
@@ -86,3 +91,30 @@ export const errIfExists = <T extends Record<string, unknown>>(
     throw json({ error_code: 'ObjectAlreadyExists' }, { status: 400 })
   }
 }
+
+export const errIfInvalidDiskSize = (
+  params: DiskCreatePathParams,
+  disk: Json<DiskCreate>
+) => {
+  const source = disk.disk_source
+  if (source.type === 'snapshot') {
+    const snapshotSize = db.snapshots.find((s) => source.snapshot_id === s.id)?.size ?? 0
+    if (disk.size >= snapshotSize) return
+    throw 'Disk size must be greater than or equal to the snapshot size'
+  }
+  if (source.type === 'image') {
+    const imageSize = db.images.find((i) => source.image_id === i.id)?.size ?? 0
+    if (disk.size >= imageSize) return
+    throw 'Disk size must be greater than or equal to the image size'
+  }
+  if (source.type === 'global_image') {
+    const globalImageSize = db.globalImages.find((i) => source.image_id === i.id)?.size ?? 0
+    if (disk.size >= globalImageSize) return
+    throw 'Disk size must be greater than or equal to the global image size'
+  }
+  if (source.type === 'blank') {
+    if (disk.size >= 1 * GiB) return
+    // TODO: this is a bit arbitrary, should match whatever the API does
+    throw 'Minimum disk size is 1 GiB'
+  }
+}