chore: wip

Author: glennmichael123

storage/framework/core/buddy/src/commands/search.ts

@@ -1,4 +1,4 @@
-import type { CLI, SearchOptions } from '@stacksjs/types'
+import type { CLI, SearchCommandOptions } from '@stacksjs/types'
 import process from 'node:process'
 import { runAction } from '@stacksjs/actions'
 import { intro, log, outro } from '@stacksjs/cli'
@@ -23,7 +23,7 @@ export function search(buddy: CLI): void {
     .option('-s, --settings [settings]', descriptions.settings, { default: false })
     .option('--verbose', descriptions.verbose, { default: false })
     .option('-p, --project [project]', descriptions.project, { default: false })
-    .action(async (options: SearchOptions) => {
+    .action(async (options: SearchCommandOptions) => {
       log.debug('Running `search-engine:update` ...', options)
 
       let actionString = Action.SearchEngineImport
@@ -68,7 +68,7 @@ export function search(buddy: CLI): void {
     .command('search-engine:settings', descriptions.list)
     .option('-m, --model [model]', descriptions.model, { default: false })
     .option('--verbose', descriptions.verbose, { default: false })
-    .action(async (options: SearchOptions) => {
+    .action(async (options: SearchCommandOptions) => {
       if (!options.model)
         log.error('Missing required option --model')
 

storage/framework/core/orm/src/utils.ts

@@ -279,6 +279,55 @@ export async function writeModelNames(): Promise<void> {
   await fs.promises.writeFile(typeFilePath, fileString, 'utf8')
 }
 
+export async function writeTableNames(): Promise<void> {
+  const models = globSync([path.userModelsPath('*.ts')], { absolute: true })
+  const coreModelFiles = globSync([path.storagePath('framework/database/models/generated/*.ts')], { absolute: true })
+
+  let fileString = `export type TableNames = `
+
+  for (let i = 0; i < models.length; i++) {
+    const modelPath = models[i] as string
+    const model = (await import(modelPath)).default as Model
+    const tableName = getTableName(model, modelPath)
+
+    const pivotTables = await getPivotTables(model, modelPath)
+
+    for (const pivot of pivotTables) {
+      fileString += `'${pivot.table}'`
+      fileString += ' | '
+    }
+
+    fileString += `'${tableName}'`
+
+    if (i < models.length - 1) {
+      fileString += ' | '
+    }
+  }
+
+  fileString += ' | '
+
+  for (let j = 0; j < coreModelFiles.length; j++) {
+    const modelPath = coreModelFiles[j] as string
+
+    const model = (await import(modelPath)).default as Model
+    const tableName = getTableName(model, modelPath)
+
+    fileString += `'${tableName}'`
+
+    if (j < coreModelFiles.length - 1) {
+      fileString += ' | '
+    }
+  }
+
+  // Ensure the directory exists
+  const typesDir = path.dirname(path.typesPath(`src/table-names.ts`))
+  await fs.promises.mkdir(typesDir, { recursive: true })
+
+  // Write to the file
+  const typeFilePath = path.typesPath(`src/table-names.ts`)
+  await fs.promises.writeFile(typeFilePath, fileString, 'utf8')
+}
+
 export async function writeModelRequest(): Promise<void> {
   const modelFiles = globSync([path.userModelsPath('*.ts')], { absolute: true })
   const requestD = Bun.file(path.frameworkPath('types/requests.d.ts'))
@@ -2116,6 +2165,15 @@ export async function generateModelFiles(modelStringFile?: string): Promise<void
       handleError('Error while writing Model Names', error)
     }
 
+    try {
+      log.info('Writing Table Names...')
+      await writeTableNames()
+      log.success('Wrote Table Names')
+    }
+    catch (error) {
+      handleError('Error while writing Table Names', error)
+    }
+
     try {
       log.info('Writing Model Requests...')
       await writeModelRequest()

storage/framework/core/types/src/cli.ts

@@ -365,7 +365,7 @@ export interface CleanOptions extends CliOptions {}
 
 export interface SaasOptions extends CliOptions {}
 
-export interface SearchOptions extends CliOptions {
+export interface SearchCommandOptions extends CliOptions {
   model: string
   settings: boolean
   flush: boolean

storage/framework/core/types/src/index.ts

@@ -57,6 +57,7 @@ export * from './sms'
 export * from './ssg'
 export * from './stacks'
 export * from './storage'
+export * from './table-names'
 export * from './tables'
 export * from './team'
 export * from './ui'

storage/framework/core/types/src/model.ts

@@ -1,4 +1,4 @@
-import type { ModelNames } from '@stacksjs/types'
+import type { ModelNames, TableNames } from '@stacksjs/types'
 import type { VineBoolean, VineNumber, VineString } from '@vinejs/vine'
 import type { DeepPartial, Nullable } from '.'
 import type { SearchOptions } from './search-engine'
@@ -183,13 +183,13 @@ export interface Attributes {
 export interface RelationConfig {
   relationship: string
   model: string
-  table: string
+  table: TableNames
   relationModel?: string
   relationTable?: string
   foreignKey: string
   modelKey: string
   relationName?: string
   throughModel?: string
   throughForeignKey?: string
-  pivotTable: string
+  pivotTable: TableNames
 }

storage/framework/core/types/src/table-names.ts

@@ -0,0 +1 @@
+export type TableNames = 'projects' | 'subscriber_emails' | 'personal_access_tokens' | 'personal_access_token_teams' | 'team_users' | 'teams' | 'subscribers' | 'deployments' | 'releases' | 'team_users' | 'users' | 'posts' | 'subscriptions' | 'errors'