wehub-resource-sync
75 lines
2.3 KiB
TypeScript
75 lines
2.3 KiB
TypeScript
import type { Page } from '@stablyai/playwright-test'
|
|
import { writeFileSync } from 'node:fs'
|
|
|
|
async function setCustomGenerator(page: Page, scriptPath: string): Promise<void> {
|
|
await page.evaluate(async (scriptPath) => {
|
|
const store =
|
|
window.__store ??
|
|
(() => {
|
|
throw new Error('window.__store is not available')
|
|
})()
|
|
const currentSettings = store.getState().settings
|
|
if (!currentSettings) {
|
|
throw new Error('Settings were not loaded')
|
|
}
|
|
await store.getState().updateSettings({
|
|
activeRuntimeEnvironmentId: null,
|
|
commitMessageAi: {
|
|
...currentSettings.commitMessageAi,
|
|
enabled: true,
|
|
agentId: 'custom' as const,
|
|
selectedModelByAgent: {},
|
|
selectedThinkingByModel: {},
|
|
customPrompt: '',
|
|
customAgentCommand: `node ${JSON.stringify(scriptPath)}`
|
|
}
|
|
})
|
|
}, scriptPath)
|
|
}
|
|
|
|
export async function installDelayedPrGenerator(
|
|
page: Page,
|
|
generatorScriptPath: string,
|
|
callLogPath: string,
|
|
base: string
|
|
): Promise<void> {
|
|
writeFileSync(
|
|
generatorScriptPath,
|
|
[
|
|
"const fs = require('fs')",
|
|
`fs.appendFileSync(${JSON.stringify(callLogPath)}, 'start\\n')`,
|
|
'setTimeout(() => {',
|
|
' console.log(JSON.stringify({',
|
|
` base: ${JSON.stringify(base)},`,
|
|
" title: 'Generated PR title after switch',",
|
|
" body: 'Generated PR body after switch',",
|
|
' draft: false',
|
|
' }))',
|
|
` fs.appendFileSync(${JSON.stringify(callLogPath)}, 'finish\\n')`,
|
|
'}, 1500)'
|
|
].join('\n')
|
|
)
|
|
await setCustomGenerator(page, generatorScriptPath)
|
|
}
|
|
|
|
export async function installDelayedCommitMessageGenerator(
|
|
page: Page,
|
|
generatorScriptPath: string,
|
|
callLogPath: string
|
|
): Promise<void> {
|
|
writeFileSync(
|
|
generatorScriptPath,
|
|
[
|
|
"const fs = require('fs')",
|
|
`fs.appendFileSync(${JSON.stringify(callLogPath)}, 'start\\n')`,
|
|
'setTimeout(() => {',
|
|
" console.log('Generated commit message after switch')",
|
|
" console.log('')",
|
|
" console.log('Generated from staged e2e-commit-message-generation.txt after switching worktrees')",
|
|
` fs.appendFileSync(${JSON.stringify(callLogPath)}, 'finish\\n')`,
|
|
'}, 1500)'
|
|
].join('\n')
|
|
)
|
|
await setCustomGenerator(page, generatorScriptPath)
|
|
}
|