Chore/move cmdk (#21936)

* init * fix build issues * move deps over * Update CommandMenuWrapper.tsx * build fixes * Update sitemap_www.xml * reset * Update sitemap_www.xml * Update tsconfig.json * Update package.json * Update package.json * Update package-lock.json * Update package.json
2026-07-07 07:30:21 +08:00 · 2024-03-11 18:31:23 +08:00
parent 7ec29d8b84
commit 07a6c5ab4e
39 changed files with 47 additions and 59 deletions
--- a/packages/ui-patterns/Cmdk/GenerateSQL/GenerateSQL.utils.ts
+++ b/packages/ui-patterns/Cmdk/GenerateSQL/GenerateSQL.utils.ts
@@ -0,0 +1,100 @@
+import { stripIndent } from 'common-tags'
+
+export function getEdgeFunctionUrl() {
+  const supabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL?.replace(/\/$/, '')
+
+  if (!supabaseUrl) return undefined
+
+  // https://github.com/supabase/supabase-js/blob/10d3423506cbd56345f7f6ab2ec2093c8db629d4/src/SupabaseClient.ts#L96
+  const isPlatform = supabaseUrl.match(/(supabase\.co)|(supabase\.in)/)
+
+  if (isPlatform) {
+    const [schemeAndProjectId, domain, tld] = supabaseUrl.split('.')
+    return `${schemeAndProjectId}.functions.${domain}.${tld}`
+  } else {
+    return `${supabaseUrl}/functions/v1`
+  }
+}
+
+export function promptDataReducer(
+  state: any[],
+  action: {
+    index?: number
+    answer?: string | undefined
+    status?: string
+    query?: string | undefined
+    type?: 'remove-last-item' | string
+  }
+) {
+  // set a standard state to use later
+  let current = [...state]
+
+  if (action.type) {
+    switch (action.type) {
+      case 'remove-last-item':
+        current.pop()
+        return [...current]
+      default:
+        break
+    }
+  }
+
+  // check that an index is present
+  if (action.index === undefined) return [...state]
+
+  if (!current[action.index]) {
+    current[action.index] = { query: '', answer: '', status: '' }
+  }
+
+  current[action.index].answer = action.answer
+
+  if (action.query) {
+    current[action.index].query = action.query
+  }
+  if (action.status) {
+    current[action.index].status = action.status
+  }
+
+  return [...current]
+}
+
+export const generatePrompt = (prompt: string, definitions?: any) => {
+  return stripIndent`
+${
+  definitions !== undefined
+    ? `
+Given the following Postgres SQL tables:
+${definitions}
+`
+    : ''
+}
+
+Generate a Postgres SQL query based on the following natural language prompt.
+- Only output valid SQL - all explanations must be SQL comments
+- SQL comments should be short
+- Your very last output should be "\`\`\`"
+- For primary keys, always use "integer primary key generated always as identity"
+
+Natural language prompt:
+${prompt}
+  
+Postgres SQL query (markdown SQL only):
+`.trim()
+}
+
+/**
+ * Formats a string for use as a title.
+ *
+ * Removes punctuation and capitalizes each word
+ */
+export function formatTitle(value: string) {
+  let words = value.replace(/\.$/, '').replace(/['"]/g, '').split(' ')
+  words = words.map((word) => {
+    // Don't capitalize code
+    if (/[._\(\)]+/.test(word)) {
+      return word
+    }
+    return word.charAt(0).toUpperCase() + word.slice(1)
+  })
+  return words.join(' ')
+}