Search MCP Server

'use strict' const { spawn } = require('child_process') const cliSelect = require('cli-select') const simpleGit = require('simple-git') const git = simpleGit(process.cwd()) const COMMAND = 'npm run bench' const DEFAULT_BRANCH = 'main' const PERCENT_THRESHOLD = 5 const greyColor = '\x1b[30m' const redColor = '\x1b[31m' const greenColor = '\x1b[32m' const resetColor = '\x1b[0m' async function selectBranchName (message, branches) { console.log(message) const result = await cliSelect({ type: 'list', name: 'branch', values: branches }) console.log(result.value) return result.value } async function executeCommandOnBranch (command, branch) { console.log(`${greyColor}Checking out "${branch}"${resetColor}`) await git.checkout(branch) console.log(`${greyColor}Execute "${command}"${resetColor}`) const childProcess = spawn(command, { stdio: 'pipe', shell: true }) let result = '' childProcess.stdout.on('data', (data) => { process.stdout.write(data.toString()) result += data.toString() }) await new Promise(resolve => childProcess.on('close', resolve)) console.log() return parseBenchmarksStdout(result) } function parseBenchmarksStdout (text) { const results = [] const lines = text.split('\n') for (const line of lines) { const match = /^(.+?)(\.*) x (.+) ops\/sec .*$/.exec(line) if (match !== null) { results.push({ name: match[1], alignedName: match[1] + match[2], result: parseInt(match[3].split(',').join('')) }) } } return results } function compareResults (featureBranch, mainBranch) { for (const { name, alignedName, result: mainBranchResult } of mainBranch) { const featureBranchBenchmark = featureBranch.find(result => result.name === name) if (featureBranchBenchmark) { const featureBranchResult = featureBranchBenchmark.result const percent = (featureBranchResult - mainBranchResult) * 100 / mainBranchResult const roundedPercent = Math.round(percent * 100) / 100 const percentString = roundedPercent > 0 ? `+${roundedPercent}%` : `${roundedPercent}%` const message = alignedName + percentString.padStart(7, '.') if (roundedPercent > PERCENT_THRESHOLD) { console.log(`${greenColor}${message}${resetColor}`) } else if (roundedPercent < -PERCENT_THRESHOLD) { console.log(`${redColor}${message}${resetColor}`) } else { console.log(message) } } } } (async function () { const branches = await git.branch() const currentBranch = branches.branches[branches.current] let featureBranch = null let mainBranch = null if (process.argv[2] === '--ci') { featureBranch = currentBranch.name mainBranch = DEFAULT_BRANCH } else { featureBranch = await selectBranchName('Select the branch you want to compare (feature branch):', branches.all) mainBranch = await selectBranchName('Select the branch you want to compare with (main branch):', branches.all) } try { const featureBranchResult = await executeCommandOnBranch(COMMAND, featureBranch) const mainBranchResult = await executeCommandOnBranch(COMMAND, mainBranch) compareResults(featureBranchResult, mainBranchResult) } catch (error) { console.error('Switch to origin branch due to an error', error.message) } await git.checkout(currentBranch.commit) await git.checkout(currentBranch.name) console.log(`${greyColor}Back to ${currentBranch.name} ${currentBranch.commit}${resetColor}`) })()