feat: add prisma format functionality

packages/vscode/package.json

@@ -178,6 +178,11 @@
         "command": "prisma.generate",
         "title": "Generate",
         "category": "Prisma"
+      },
+      {
+        "command": "prisma.format",
+        "title": "Format",
+        "category": "Prisma"
       }
     ]
   },

packages/vscode/src/CodeLensProvider.ts

@@ -33,22 +33,18 @@ export class CodelensProvider implements vscode.CodeLensProvider {
     const generatorRanges = this.getGeneratorRange(document, token)
 
     const lenses = generatorRanges.map(
-      (range) =>
+      (range) => [
         new vscode.CodeLens(range, {
           title: 'Generate',
           command: 'prisma.generate',
           tooltip: `Run "prisma generate"`,
-          // ? (@druue) The arguments property does not seem to actually
-          // ? return an array of arguments. It would consistently
-          // ? return one singular string element, even when defined as:
-          // ? [this.scriptRunner, this.schemaPath]
-          // ?
-          // ? I've tried to understand why as there are usages in other
-          // ? codebases that do pass in multiple args so I have to imagine
-          // ? that it can work, but unsure.
-          // ? Reference: https://github.com/microsoft/vscode-extension-samples/blob/main/codelens-sample/
-          // ? arguments: [this.scriptRunner]
         }),
+        new vscode.CodeLens(range, {
+          title: 'Format',
+          command: 'prisma.format',
+          tooltip: `Run "prisma format"`,
+        })
+      ]
     )
 
     return lenses
@@ -108,3 +104,37 @@ export function generateClient(_args: string) {
 
   cp.exec(cmd, { cwd: rootPath }, (err, stdout, stderr) => handleExec(err, stdout, stderr))
 }
+
+export function formatPrismaSchema(_args: string) {
+  const prismaFormatOutputChannel = vscode.window.createOutputChannel('Prisma Format')
+  const rootPath = vscode.workspace.workspaceFolders?.[0].uri.fsPath
+
+  const scriptRunner = vscode.workspace.getConfiguration('prisma').get('scriptRunner', 'npx')
+  const schemaPath: string | undefined = vscode.workspace.getConfiguration('prisma').get('schemaPath')
+
+  const pathArgFlag = ` --schema=${schemaPath}`
+  const cmd = `${scriptRunner} prisma format${schemaPath ? pathArgFlag : ''}`
+
+  prismaFormatOutputChannel.clear()
+  prismaFormatOutputChannel.show(true)
+  prismaFormatOutputChannel.appendLine(['Running prisma format:', rootPath, cmd].join('\n- '))
+
+  const handleExec = (err: cp.ExecException | null, stdout: string, stderr: string) => {
+    try {
+      if (err) {
+        prismaFormatOutputChannel.appendLine(err.message)
+        return
+      }
+      if (stdout) {
+        prismaFormatOutputChannel.append(stdout)
+      }
+      if (stderr) {
+        prismaFormatOutputChannel.append(stderr)
+      }
+    } catch (e) {
+      prismaFormatOutputChannel.append(e as string)
+    }
+  }
+
+  cp.exec(cmd, { cwd: rootPath }, (err, stdout, stderr) => handleExec(err, stdout, stderr))
+}

packages/vscode/src/plugins/prisma-language-server/index.ts

@@ -36,7 +36,7 @@ import {
 import { PrismaVSCodePlugin } from '../types'
 import paths from 'env-paths'
 import FileWatcher from 'watcher'
-import { CodelensProvider, generateClient } from '../../CodeLensProvider'
+import { CodelensProvider, generateClient, formatPrismaSchema } from '../../CodeLensProvider'
 
 const packageJson = require('../../../../package.json') // eslint-disable-line
 
@@ -219,6 +219,7 @@ const plugin: PrismaVSCodePlugin = {
       }),
 
       commands.registerCommand('prisma.generate', (args: string) => generateClient(args)),
+      commands.registerCommand('prisma.format', (args: string) => formatPrismaSchema(args)),
 
       commands.registerCommand('prisma.restartLanguageServer', async () => {
         client = await restartClient(context, client, serverOptions, clientOptions)