File size: 2,641 Bytes
e089492 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | package shellconfig
import (
"context"
"fmt"
"log/slog"
"os"
"path/filepath"
"time"
"github.com/charmbracelet/crush/internal/shell"
"github.com/charmbracelet/crush/internal/version"
)
// loadTimeout bounds a single crushrc execution. Config loading runs on the
// startup and reload critical paths while the config store's write lock is
// held, so a script that blocks (a hung command substitution, a stray loop)
// must not be able to wedge the whole store. The interpreter honors context
// cancellation, so this deadline reliably interrupts a runaway script.
const loadTimeout = 30 * time.Second
// LoadShellConfig executes a crushrc script and returns its config as a
// single JSON object. The script uses config builtins (provider, model, mcp,
// etc.) that mutate a ConfigBuilder in execution order; the builder is then
// marshaled to JSON, which the config loader merges with any other config
// files.
//
// The script runs with the same shell interpreter used by the bash tool and
// hooks, so source, $VAR, $(cmd), and other shell constructs all work.
//
// Execution is bounded by loadTimeout on top of any deadline already present
// on ctx, so a misbehaving script cannot block config loading indefinitely.
func LoadShellConfig(ctx context.Context, path string, src []byte) ([]byte, error) {
slog.Info("Loading shell config", "path", path)
ctx, cancel := context.WithTimeout(ctx, loadTimeout)
defer cancel()
builder := newConfigBuilder()
runCtx := withConfigBuilder(ctx, builder)
cwd := filepath.Dir(path)
// Expose the running Crush version so scripts can feature-detect, e.g.
// [[ "$CRUSH_VERSION" == "devel" ]] or branch on the release.
env := append(os.Environ(), "CRUSH_VERSION="+version.Version)
err := shell.Run(runCtx, shell.RunOptions{
Command: string(src),
Cwd: cwd,
Env: env,
})
if err != nil {
if shell.IsInterrupt(err) {
slog.Error("Shell config execution timed out or was cancelled", "path", path, "error", err)
return nil, fmt.Errorf("shell config %s: execution timed out or was cancelled: %w", path, err)
}
slog.Error("Shell config execution failed", "path", path, "error", err)
return nil, fmt.Errorf("executing shell config %s: %w", path, err)
}
if builder.empty() {
slog.Warn("Shell config produced no config", "path", path)
return nil, nil
}
data, err := builder.JSON()
if err != nil {
slog.Error("Failed to marshal shell config", "path", path, "error", err)
return nil, fmt.Errorf("marshaling shell config %s: %w", path, err)
}
slog.Info("Shell config loaded successfully", "path", path, "bytes", len(data))
return data, nil
}
|