wip: profile

.gitignore

@@ -9,3 +9,4 @@ explorations
 .idea
 *.local
 /packages/vite/LICENSE
+*.cpuprofile

packages/plugin-react-refresh/index.js

@@ -1,5 +1,6 @@
 // @ts-check
 const fs = require('fs')
+const { transformSync } = require('@babel/core')
 
 const runtimePublicPath = '/@react-refresh'
 const runtimeFilePath = require.resolve(
@@ -77,7 +78,7 @@ const transform = {
     }
 
     const isReasonReact = id.endsWith('.bs.js')
-    const result = require('@babel/core').transformSync(code, {
+    const result = transformSync(code, {
       plugins: [
         require('@babel/plugin-syntax-import-meta'),
         require('react-refresh/babel')

packages/vite/bin/vite.js

@@ -4,6 +4,7 @@ global.__vite_start_time = Date.now()
 // check debug mode first before requiring the CLI.
 const debugIndex = process.argv.indexOf('--debug')
 const filterIndex = process.argv.indexOf('--filter')
+const profileIndex = process.argv.indexOf('--profile')
 
 if (debugIndex > 0) {
   let value = process.argv[debugIndex + 1]
@@ -31,4 +32,22 @@ if (debugIndex > 0) {
   } catch (e) {}
 }
 
-require('../dist/node/cli')
+function start() {
+  require('../dist/node/cli')
+}
+
+if (profileIndex > 0) {
+  process.argv.splice(profileIndex, 1)
+  const next = process.argv[profileIndex]
+  if (next && !next.startsWith('-')) {
+    process.argv.splice(profileIndex, 1)
+  }
+  const inspector = require('inspector')
+  const session = (global.__vite_profile_session = new inspector.Session())
+  session.connect()
+  session.post('Profiler.enable', () => {
+    session.post('Profiler.start', start)
+  })
+} else {
+  start()
+}

packages/vite/rollup.config.js

@@ -33,7 +33,8 @@ const nodeConfig = {
   input: {
     index: path.resolve(__dirname, 'src/node/index.ts'),
     cli: path.resolve(__dirname, 'src/node/cli.ts'),
-    server: path.resolve(__dirname, 'src/node/server/index.ts')
+    server: path.resolve(__dirname, 'src/node/server/index.ts'),
+    build: path.resolve(__dirname, 'src/node/build/index.ts')
   },
   external: [
     'fsevents',

packages/vite/src/node/cli.ts

@@ -35,9 +35,9 @@ function cleanOptions(options: GlobalCLIOptions) {
 
 cli
   .option('-c, --config <file>', `[string] use specified config file`)
-  .option('-d, --debug [feat]', `[string | boolean] show debug logs`)
-  .option('--filter [filter]', `[string] filter debug logs`)
   .option('--root <path>', `[string] use specified config file`)
+  .option('--debug [feat]', `[string | boolean] show debug logs`)
+  .option('--filter [filter]', `[string] filter debug logs`)
 
 // dev
 cli

packages/vite/src/node/server/index.ts

@@ -1,4 +1,5 @@
 import os from 'os'
+import fs from 'fs'
 import path from 'path'
 import * as http from 'http'
 import * as https from 'https'
@@ -343,12 +344,31 @@ export async function startServer(
           console.log(`  > ${type} ${chalk.cyan(url)}`)
         })
     )
+
     console.log()
     console.log(
       // @ts-ignore
       chalk.cyan(`  ready in ${Date.now() - global.__vite_start_time}ms.`)
     )
     console.log()
+
+    // @ts-ignore
+    const profileSession = global.__vite_profile_session
+    if (profileSession) {
+      profileSession.post('Profiler.stop', (err: any, { profile }: any) => {
+        // Write profile to disk, upload, etc.
+        if (!err) {
+          const outPath = path.resolve('./vite-profile.cpuprofile')
+          fs.writeFileSync(outPath, JSON.stringify(profile))
+          console.log(
+            chalk.yellow(`  CPU profile written to ${chalk.white.dim(outPath)}`)
+          )
+          console.log()
+        } else {
+          throw err
+        }
+      })
+    }
   })
 
   if (options.open) {