openAgent

1#!/usr/bin/env bun
2 
3import { $ } from "bun"
4import path from "path"
5import os from "os"
6import { ZenData } from "../src/model"
7 
8const root = path.resolve(process.cwd(), "..", "..", "..")
9const models = await $`bun sst secret list --stage frank`.cwd(root).text()
10const PARTS = 30
11 
12// read the line starting with "ZEN_MODELS"
13const lines = models.split("\n")
14const oldValues = Array.from({ length: PARTS }, (_, i) => {
15  const value = lines
16    .find((line) => line.startsWith(`ZEN_MODELS${i + 1}=`))
17    ?.split("=")
18    .slice(1)
19    .join("=")
20  if (!value) throw new Error(`ZEN_MODELS${i + 1} not found`)
21  return value
22})
23 
24// store the prettified json to a temp file
25const filename = `models-${Date.now()}.json`
26const tempFile = Bun.file(path.join(os.tmpdir(), filename))
27await tempFile.write(JSON.stringify(JSON.parse(oldValues.join("")), null, 2))
28console.log("tempFile", tempFile.name)
29 
30// open temp file in vim and read the file on close
31await $`vim ${tempFile.name}`
32const newValue = JSON.stringify(JSON.parse(await tempFile.text()))
33ZenData.validate(JSON.parse(newValue))
34 
35// update the secret
36const chunk = Math.ceil(newValue.length / PARTS)
37const newValues = Array.from({ length: PARTS }, (_, i) =>
38  newValue.slice(chunk * i, i === PARTS - 1 ? undefined : chunk * (i + 1)),
39)
40 
41const envFile = Bun.file(path.join(os.tmpdir(), `models-${Date.now()}.env`))
42await envFile.write(newValues.map((v, i) => `ZEN_MODELS${i + 1}="${v.replace(/"/g, '\\"')}"`).join("\n"))
43await $`bun sst secret load ${envFile.name} --stage frank`.cwd(root)
44