probablycorey/shrimp
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

/save

Browse Source
This commit is contained in:
Chris Wanstrath 2025-10-25 10:42:49 -07:00
parent 664ba82199
commit ad1d7266b8
1 changed files with 42 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
bin/repl
View File

@ -2,7 +2,8 @@
import { Compiler } from '../src/compiler/compiler' import { Compiler } from '../src/compiler/compiler'
import { VM, type Value, Scope, bytecodeToString } from 'reefvm' import { VM, type Value, Scope, bytecodeToString } from 'reefvm'
import * as readline from 'node:readline' import * as readline from 'readline'
import * as fs from 'fs'
const colors = { const colors = {
reset: '\x1b[0m', reset: '\x1b[0m',
@ -18,7 +19,7 @@ const colors = {
} }
async function repl() { async function repl() {
const commands = ['/clear', '/reset', '/vars', '/funcs', '/history', '/bytecode', '/exit', '/quit'] const commands = ['/clear', '/reset', '/vars', '/funcs', '/history', '/bytecode', '/exit', '/save', '/quit']
function completer(line: string): [string[], string] { function completer(line: string): [string[], string] {
if (line.startsWith('/')) { if (line.startsWith('/')) {
@ -115,6 +116,37 @@ async function repl() {
return return
} }
if (trimmed.startsWith('/save')) {
const parts = trimmed.split(/\s+/)
const filename = parts[1]
if (!filename) {
console.log(`\n${colors.red}Usage:${colors.reset} /save <filename>`)
rl.prompt()
return
}
if (codeHistory.length === 0) {
console.log(`\n${colors.dim}No history to save${colors.reset}`)
rl.prompt()
return
}
// Add .shrimp extension if no extension provided
const finalFilename = filename.includes('.') ? filename : `${filename}.shrimp`
const content = codeHistory.join('\n') + '\n'
try {
fs.writeFileSync(finalFilename, content, 'utf-8')
console.log(`\n${colors.green}✓${colors.reset} Saved ${codeHistory.length} line${codeHistory.length === 1 ? '' : 's'} to ${colors.bright}${finalFilename}${colors.reset}`)
} catch (error: any) {
console.log(`\n${colors.red}Error:${colors.reset} Failed to save file: ${error.message}`)
}
rl.prompt()
return
}
codeHistory.push(trimmed) codeHistory.push(trimmed)
try { try {
@ -219,13 +251,14 @@ function showWelcome() {
) )
console.log(`\nType Shrimp expressions. Press ${colors.bright}Ctrl+D${colors.reset} to exit.`) console.log(`\nType Shrimp expressions. Press ${colors.bright}Ctrl+D${colors.reset} to exit.`)
console.log(`\nCommands:`) console.log(`\nCommands:`)
console.log(` ${colors.bright}/clear${colors.reset} - Clear screen and reset state`) console.log(` ${colors.bright}/clear${colors.reset} - Clear screen and reset state`)
console.log(` ${colors.bright}/reset${colors.reset} - Reset state (keep history visible)`) console.log(` ${colors.bright}/reset${colors.reset} - Reset state (keep history visible)`)
console.log(` ${colors.bright}/vars${colors.reset} - Show all variables`) console.log(` ${colors.bright}/vars${colors.reset} - Show all variables`)
console.log(` ${colors.bright}/funcs${colors.reset} - Show all functions`) console.log(` ${colors.bright}/funcs${colors.reset} - Show all functions`)
console.log(` ${colors.bright}/history${colors.reset} - Show code history`) console.log(` ${colors.bright}/history${colors.reset} - Show code history`)
console.log(` ${colors.bright}/bytecode${colors.reset} - Show compiled bytecode`) console.log(` ${colors.bright}/bytecode${colors.reset} - Show compiled bytecode`)
console.log(` ${colors.bright}/exit${colors.reset} - Quit REPL`) console.log(` ${colors.bright}/save <file>${colors.reset} - Save history to file`)
console.log(` ${colors.bright}/exit${colors.reset} - Quit REPL`)
console.log(`\nExamples:`) console.log(`\nExamples:`)
console.log(` ${colors.cyan}5 + 10${colors.reset}`) console.log(` ${colors.cyan}5 + 10${colors.reset}`)
console.log(` ${colors.cyan}x = 42${colors.reset}`) console.log(` ${colors.cyan}x = 42${colors.reset}`)