helpers.go

// path: codemode/helpers.go
package codemode

import (
	"fmt"
	"strings"
)

func preprocessUserCode(code string) string {
	code = strings.TrimSpace(code)

	if !strings.Contains(code, "__out") &&
		!strings.Contains(code, "return") {

		code += "\n__out = nil"
	}

	return code
}

func wrapIntoProgram(clean string) string {
	return fmt.Sprintf(`package main

import (
	codemode "codemode_helpers/codemode_helpers"
)

func run() any {
	var __out any

	// ----- BEGIN USER CODE -----
%s
	// ----- END USER CODE -----

	return __out
}
`, indent(clean, "\t"))
}

func indent(s, prefix string) string {
	lines := strings.Split(s, "\n")

	for i := range lines {
		lines[i] = prefix + lines[i]
	}

	return strings.Join(lines, "\n")
}

type CacheStats struct {
	Hits   int
	Misses int
}

func extractJSON(s string) string {
	start := strings.Index(s, "{")
	end := strings.LastIndex(s, "}")
	if start != -1 && end != -1 && start <= end {
		return s[start : end+1]
	}
	return ""
}

func isValidSnippet(code string) bool {
	// Simple validation, actual compilation will happen in Yaegi
	return strings.TrimSpace(code) != ""
}