goscript 0.1.0 → 0.1.2

package/compiler/tsworkspace/owner-process-unix_test.go

@@ -0,0 +1,72 @@
+//go:build unix
+
+package tsworkspace
+
+import (
+	"context"
+	"errors"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"syscall"
+	"testing"
+	"time"
+)
+
+func TestOwnerRunToolCancelsProcessGroup(t *testing.T) {
+	dir := t.TempDir()
+	pidFile := filepath.Join(dir, "child.pid")
+	tool := filepath.Join(dir, "spawn-child.sh")
+	script := "#!/bin/sh\n" +
+		"sleep 30 &\n" +
+		"echo $! > " + strconv.Quote(pidFile) + "\n" +
+		"wait\n"
+	if err := os.WriteFile(tool, []byte(script), 0o755); err != nil {
+		t.Fatal(err.Error())
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+	result := NewOwner(dir, dir).RunTool(ctx, PhaseRuntime, dir, tool)
+	if !result.Failed() {
+		t.Fatalf("expected canceled tool to fail")
+	}
+
+	pid := readChildPID(t, pidFile)
+	deadline := time.Now().Add(2 * time.Second)
+	for processExists(pid) && time.Now().Before(deadline) {
+		time.Sleep(25 * time.Millisecond)
+	}
+	if processExists(pid) {
+		_ = syscall.Kill(pid, syscall.SIGKILL)
+		t.Fatalf("child process %d survived canceled tool", pid)
+	}
+}
+
+func readChildPID(t *testing.T, path string) int {
+	t.Helper()
+	deadline := time.Now().Add(2 * time.Second)
+	var data []byte
+	var err error
+	for time.Now().Before(deadline) {
+		data, err = os.ReadFile(path)
+		if err == nil {
+			break
+		}
+		time.Sleep(25 * time.Millisecond)
+	}
+	if err != nil {
+		t.Fatalf("read child pid: %v", err)
+	}
+	pid, err := strconv.Atoi(strings.TrimSpace(string(data)))
+	if err != nil {
+		t.Fatalf("parse child pid %q: %v", data, err)
+	}
+	return pid
+}
+
+func processExists(pid int) bool {
+	err := syscall.Kill(pid, 0)
+	return err == nil || errors.Is(err, syscall.EPERM)
+}

package/compiler/tsworkspace/owner.go

@@ -0,0 +1,342 @@
+package tsworkspace
+
+import (
+	"bytes"
+	"context"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"slices"
+	"strings"
+	"time"
+
+	jsoniter "github.com/aperturerobotics/json-iterator-lite"
+	"github.com/pkg/errors"
+)
+
+// Phase names a TypeScript workspace execution phase.
+type Phase string
+
+const (
+	// PhaseWorkspace covers workspace file preparation and tool discovery.
+	PhaseWorkspace Phase = "workspace"
+	// PhaseTypeCheck covers TypeScript type checking.
+	PhaseTypeCheck Phase = "typecheck"
+	// PhaseRuntime covers TypeScript runtime execution.
+	PhaseRuntime Phase = "runtime"
+
+	// NodeAmbientTypesFile names the GoScript runtime ambient declarations.
+	NodeAmbientTypesFile = "goscript-node.d.ts"
+)
+
+// Owner owns TypeScript test workspace files, tool discovery, and execution.
+type Owner struct {
+	workDir string
+	toolDir string
+}
+
+// NewOwner creates a TypeScript workspace owner.
+func NewOwner(workDir, toolDir string) *Owner {
+	return &Owner{
+		workDir: workDir,
+		toolDir: toolDir,
+	}
+}
+
+// WorkDir returns the workspace directory.
+func (o *Owner) WorkDir() string {
+	if o == nil {
+		return ""
+	}
+	return o.workDir
+}
+
+// EnsurePackageJSON writes the module package metadata used by Bun.
+func (o *Owner) EnsurePackageJSON() Result {
+	return o.WriteFile(PhaseWorkspace, "package.json", "{\"type\":\"module\"}\n")
+}
+
+// EnsureNodeAmbientTypes writes host runtime declarations needed by emitted tests.
+func (o *Owner) EnsureNodeAmbientTypes() Result {
+	if NodeTypesPresent(o.workDir, o.toolDir) {
+		return o.WriteFile(PhaseWorkspace, NodeAmbientTypesFile, "")
+	}
+	return o.WriteFile(PhaseWorkspace, NodeAmbientTypesFile, nodeAmbientTypes)
+}
+
+// WriteFile writes a workspace-relative file.
+func (o *Owner) WriteFile(phase Phase, name string, data string) Result {
+	if o == nil || o.workDir == "" {
+		return Result{Phase: phase, Error: "TypeScript workspace directory is empty"}
+	}
+	if err := os.MkdirAll(o.workDir, 0o755); err != nil {
+		return Result{Phase: phase, Error: errors.Wrap(err, "create TypeScript workspace").Error()}
+	}
+	path, err := o.workspacePath(name)
+	if err != nil {
+		return Result{Phase: phase, Error: err.Error()}
+	}
+	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
+		return Result{Phase: phase, Error: errors.Wrap(err, "create TypeScript workspace file parent").Error()}
+	}
+	if err := os.WriteFile(path, []byte(data), 0o644); err != nil {
+		return Result{Phase: phase, Error: errors.Wrap(err, "write TypeScript workspace file").Error()}
+	}
+	return Result{Phase: phase}
+}
+
+// WriteJSON writes an indented JSON workspace file.
+func (o *Owner) WriteJSON(phase Phase, name string, value any) Result {
+	data, err := renderJSON(value)
+	if err != nil {
+		return Result{Phase: phase, Error: err.Error()}
+	}
+	return o.WriteFile(phase, name, data)
+}
+
+// RunTool finds and executes a TypeScript workspace tool.
+func (o *Owner) RunTool(ctx context.Context, phase Phase, dir string, name string, args ...string) Result {
+	tool, err := o.FindTool(name)
+	if err != nil {
+		return Result{Phase: phase, Error: err.Error()}
+	}
+	if dir == "" {
+		dir = o.workDir
+	}
+	start := time.Now()
+	cmd := exec.CommandContext(ctx, tool, args...)
+	configureToolCommand(cmd)
+	cmd.Cancel = func() error {
+		killToolCommand(cmd)
+		return nil
+	}
+	cmd.WaitDelay = 5 * time.Second
+	cmd.Dir = dir
+	var output bytes.Buffer
+	cmd.Stdout = &output
+	cmd.Stderr = &output
+	err = cmd.Run()
+	result := Result{
+		Phase:   phase,
+		Command: append([]string{tool}, args...),
+		Output:  output.String(),
+		Elapsed: time.Since(start),
+	}
+	if err != nil {
+		result.Error = err.Error()
+	}
+	return result
+}
+
+// FindTool locates a TypeScript workspace tool.
+func (o *Owner) FindTool(name string) (string, error) {
+	if filepath.IsAbs(name) {
+		return name, nil
+	}
+	starts := []string{o.toolDir, o.workDir, sourceDirectory(), currentWorkingDirectory()}
+	for _, start := range starts {
+		if start == "" {
+			continue
+		}
+		for current := start; current != ""; current = filepath.Dir(current) {
+			candidate := filepath.Join(current, "node_modules", ".bin", name)
+			info, err := os.Stat(candidate)
+			if err == nil && !info.IsDir() {
+				return candidate, nil
+			}
+			if parent := filepath.Dir(current); parent == current {
+				break
+			}
+		}
+	}
+	if path, err := exec.LookPath(name); err == nil {
+		return path, nil
+	}
+	return "", errors.New(name + " not found in PATH or ancestor node_modules/.bin")
+}
+
+func (o *Owner) workspacePath(name string) (string, error) {
+	if filepath.IsAbs(name) {
+		return "", errors.Errorf("TypeScript workspace path must be relative: %s", name)
+	}
+	clean := filepath.Clean(name)
+	if clean == "." || clean == ".." || strings.HasPrefix(clean, ".."+string(filepath.Separator)) {
+		return "", errors.Errorf("TypeScript workspace path escapes workspace: %s", name)
+	}
+	return filepath.Join(o.workDir, clean), nil
+}
+
+// NodeTypeRoots returns existing @types roots visible from the provided dirs.
+func NodeTypeRoots(dirs ...string) []string {
+	seen := make(map[string]bool)
+	var roots []string
+	for _, start := range append(dirs, currentWorkingDirectory()) {
+		if start == "" {
+			continue
+		}
+		for current := start; current != ""; current = filepath.Dir(current) {
+			candidate := filepath.Join(current, "node_modules", "@types")
+			info, err := os.Stat(candidate)
+			if err == nil && info.IsDir() && !seen[candidate] {
+				seen[candidate] = true
+				roots = append(roots, candidate)
+				break
+			}
+			if parent := filepath.Dir(current); parent == current {
+				break
+			}
+		}
+	}
+	slices.Sort(roots)
+	return roots
+}
+
+// NodeTypesPresent reports whether @types/node is visible from any provided directory.
+func NodeTypesPresent(dirs ...string) bool {
+	seen := make(map[string]bool)
+	for _, start := range dirs {
+		if start == "" {
+			continue
+		}
+		for current := start; current != ""; current = filepath.Dir(current) {
+			if seen[current] {
+				if parent := filepath.Dir(current); parent == current {
+					break
+				}
+				continue
+			}
+			seen[current] = true
+			candidate := filepath.Join(current, "node_modules", "@types", "node")
+			info, err := os.Stat(candidate)
+			if err == nil && info.IsDir() {
+				return true
+			}
+			if parent := filepath.Dir(current); parent == current {
+				break
+			}
+		}
+	}
+	return false
+}
+
+func currentWorkingDirectory() string {
+	wd, err := os.Getwd()
+	if err != nil {
+		return ""
+	}
+	return wd
+}
+
+func sourceDirectory() string {
+	_, file, _, ok := runtime.Caller(0)
+	if !ok {
+		return ""
+	}
+	return filepath.Dir(file)
+}
+
+const nodeAmbientTypes = `declare const process: {
+  env?: Record<string, string | undefined>
+  chdir?: (directory: string) => void
+  cwd?: () => string
+  exit?: (code?: number) => never
+}
+
+declare var gc: (() => void) | undefined
+
+declare namespace NodeJS {
+  type Timeout = ReturnType<typeof setTimeout>
+}
+
+declare module 'node:fs' {
+  export function mkdirSync(...args: unknown[]): unknown
+}
+
+declare module 'node:os' {
+  export function tmpdir(): string
+}
+
+declare module 'node:path' {
+  export function join(...parts: string[]): string
+}
+`
+
+func renderJSON(value any) (string, error) {
+	stream := jsoniter.NewStream(nil, 512, 2)
+	if err := writeJSONValue(stream, value); err != nil {
+		return "", errors.Wrap(err, "marshal TypeScript workspace JSON")
+	}
+	stream.WriteRaw("\n")
+	if stream.Error != nil {
+		return "", errors.Wrap(stream.Error, "marshal TypeScript workspace JSON")
+	}
+	return string(stream.Buffer()), nil
+}
+
+func writeJSONValue(stream *jsoniter.Stream, value any) error {
+	switch typed := value.(type) {
+	case nil:
+		stream.WriteNil()
+	case bool:
+		stream.WriteBool(typed)
+	case string:
+		stream.WriteString(typed)
+	case []string:
+		stream.WriteArrayStart()
+		for idx, item := range typed {
+			if idx != 0 {
+				stream.WriteMore()
+			}
+			stream.WriteString(item)
+		}
+		stream.WriteArrayEnd()
+	case []any:
+		stream.WriteArrayStart()
+		for idx, item := range typed {
+			if idx != 0 {
+				stream.WriteMore()
+			}
+			if err := writeJSONValue(stream, item); err != nil {
+				return err
+			}
+		}
+		stream.WriteArrayEnd()
+	case map[string]string:
+		return writeJSONObject(stream, typed, func(value string) error {
+			stream.WriteString(value)
+			return nil
+		})
+	case map[string][]string:
+		return writeJSONObject(stream, typed, func(value []string) error {
+			return writeJSONValue(stream, value)
+		})
+	case map[string]any:
+		return writeJSONObject(stream, typed, func(value any) error {
+			return writeJSONValue(stream, value)
+		})
+	default:
+		return errors.Errorf("unsupported TypeScript workspace JSON value %T", value)
+	}
+	return nil
+}
+
+func writeJSONObject[T any](stream *jsoniter.Stream, fields map[string]T, writeValue func(T) error) error {
+	stream.WriteObjectStart()
+	keys := make([]string, 0, len(fields))
+	for key := range fields {
+		keys = append(keys, key)
+	}
+	slices.Sort(keys)
+	for idx, key := range keys {
+		if idx != 0 {
+			stream.WriteMore()
+		}
+		stream.WriteObjectField(key)
+		if err := writeValue(fields[key]); err != nil {
+			return err
+		}
+	}
+	stream.WriteObjectEnd()
+	return nil
+}

package/compiler/tsworkspace/owner_test.go

@@ -0,0 +1,93 @@
+package tsworkspace
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+)
+
+func TestOwnerWritesWorkspaceFiles(t *testing.T) {
+	dir := t.TempDir()
+	owner := NewOwner(dir, dir)
+
+	if result := owner.EnsurePackageJSON(); result.Failed() {
+		t.Fatalf("write package json: %s", result.Error)
+	}
+	if result := owner.WriteJSON(PhaseWorkspace, "tsconfig.json", map[string]any{
+		"compilerOptions": map[string]any{
+			"strict": true,
+		},
+	}); result.Failed() {
+		t.Fatalf("write tsconfig: %s", result.Error)
+	}
+
+	for _, name := range []string{"package.json", "tsconfig.json"} {
+		if _, err := os.Stat(filepath.Join(dir, name)); err != nil {
+			t.Fatalf("expected %s: %v", name, err)
+		}
+	}
+}
+
+func TestOwnerRejectsWorkspacePathEscapes(t *testing.T) {
+	dir := t.TempDir()
+	owner := NewOwner(dir, dir)
+
+	for _, name := range []string{"../escape.ts", filepath.Join("nested", "..", "..", "escape.ts"), filepath.Join(dir, "escape.ts")} {
+		if result := owner.WriteFile(PhaseWorkspace, name, "escape"); !result.Failed() {
+			t.Fatalf("expected %s to fail", name)
+		}
+	}
+}
+
+func TestOwnerRunsToolsWithCapturedOutput(t *testing.T) {
+	dir := t.TempDir()
+	owner := NewOwner(dir, dir)
+
+	result := owner.RunTool(context.Background(), PhaseRuntime, dir, "go", "env", "GOVERSION")
+	if result.Failed() {
+		t.Fatalf("run tool: %s\n%s", result.Error, result.Output)
+	}
+	if result.Output == "" {
+		t.Fatalf("expected captured output")
+	}
+	if len(result.Command) == 0 {
+		t.Fatalf("expected command capture")
+	}
+}
+
+func TestOwnerWritesNodeAmbientTypes(t *testing.T) {
+	dir := t.TempDir()
+	owner := NewOwner(dir, dir)
+
+	if result := owner.EnsureNodeAmbientTypes(); result.Failed() {
+		t.Fatalf("write node ambient types: %s", result.Error)
+	}
+	data, err := os.ReadFile(filepath.Join(dir, NodeAmbientTypesFile))
+	if err != nil {
+		t.Fatalf("read node ambient types: %v", err)
+	}
+	if !strings.Contains(string(data), "declare module 'node:fs'") {
+		t.Fatalf("node ambient types missing node:fs declaration: %s", data)
+	}
+}
+
+func TestOwnerSkipsNodeAmbientTypesWhenNodeTypesExist(t *testing.T) {
+	dir := t.TempDir()
+	if err := os.MkdirAll(filepath.Join(dir, "node_modules", "@types", "node"), 0o755); err != nil {
+		t.Fatal(err.Error())
+	}
+	owner := NewOwner(filepath.Join(dir, "work"), dir)
+
+	if result := owner.EnsureNodeAmbientTypes(); result.Failed() {
+		t.Fatalf("write node ambient types: %s", result.Error)
+	}
+	data, err := os.ReadFile(filepath.Join(dir, "work", NodeAmbientTypesFile))
+	if err != nil {
+		t.Fatalf("read node ambient types: %v", err)
+	}
+	if strings.TrimSpace(string(data)) != "" {
+		t.Fatalf("node ambient types should be empty when @types/node is already visible: %s", data)
+	}
+}

package/compiler/tsworkspace/result.go

@@ -0,0 +1,17 @@
+package tsworkspace
+
+import "time"
+
+// Result describes one workspace file or process operation.
+type Result struct {
+	Phase   Phase
+	Command []string
+	Output  string
+	Error   string
+	Elapsed time.Duration
+}
+
+// Failed returns true when the operation failed.
+func (r Result) Failed() bool {
+	return r.Error != ""
+}

package/compiler/tsworkspace/tool-process-other.go

@@ -0,0 +1,14 @@
+//go:build !unix
+
+package tsworkspace
+
+import "os/exec"
+
+func configureToolCommand(cmd *exec.Cmd) {}
+
+func killToolCommand(cmd *exec.Cmd) {
+	if cmd == nil || cmd.Process == nil {
+		return
+	}
+	_ = cmd.Process.Kill()
+}

package/compiler/tsworkspace/tool-process-unix.go

@@ -0,0 +1,19 @@
+//go:build unix
+
+package tsworkspace
+
+import (
+	"os/exec"
+	"syscall"
+)
+
+func configureToolCommand(cmd *exec.Cmd) {
+	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
+}
+
+func killToolCommand(cmd *exec.Cmd) {
+	if cmd == nil || cmd.Process == nil {
+		return
+	}
+	_ = syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL)
+}