e2b

type APIError struct {
	Code    int    `json:"code,omitempty"` // Code is the code of the error.
	Message string `json:"message"`        // Message is the message of the error.
}

type ErrMissingRequiredArgument struct {
	ToolName string
	ArgName  string
}

type ErrToolArgument struct {
	ToolName string
	ArgName  string
}

type ErrToolNotFound struct {
	ToolName string
}

type Event struct {
	Path      string      `json:"path"`      // Path is the path of the event.
	Name      string      `json:"name"`      // Name is the name of file or directory.
	Timestamp int64       `json:"timestamp"` // Timestamp is the timestamp of the event.
	Error     string      `json:"error"`     // Error is the possible error of the event.
	Params    EventParams `json:"params"`    // Params is the parameters of the event.
}

type EventParams struct {
	Subscription string      `json:"subscription"` // Subscription is the subscription id of the event.
	Result       EventResult `json:"result"`       // Result is the result of the event.
}

type EventResult struct {
	Type        string `json:"type"`
	Line        string `json:"line"`
	Timestamp   int64  `json:"timestamp"`
	IsDirectory bool   `json:"isDirectory"`
	Error       string `json:"error"`
}

type LsResult struct {
	Name  string `json:"name"`  // Name is the name of the file or directory.
	IsDir bool   `json:"isDir"` // isDir is true if the entry is a directory.
}

type Method string

type Option func(*Sandbox)

type Params struct {
	Path    string `json:"path"`
	Data    string `json:"data"`
	Cmd     string `json:"cmd"`
	Timeout int    `json:"timeout"`
	Cwd     string `json:"cwd"`
	Name    string `json:"name"`
}

type Process struct {
	Cwd string // cwd is process's current working directory.

	Env map[string]string // env is process's environment variables.
	// contains filtered or unexported fields
}

type ProcessEvents string

type ProcessOption func(*Process)

type Request struct {
	JSONRPC string `json:"jsonrpc"` // JSONRPC is the JSON-RPC version of the request.
	Method  Method `json:"method"`  // Method is the request method.
	ID      int    `json:"id"`      // ID of the request.
	Params  []any  `json:"params"`  // Params of the request.
}

type Response[T any, Q any] struct {
	ID     int `json:"id"`     // ID of the response.
	Result T   `json:"result"` // Result of the response.
	Error  Q   `json:"error"`  // Error of the message.
}

type Sandbox struct {
	ID       string `json:"sandboxID"` // ID of the sandbox.
	ClientID string `json:"clientID"`  // ClientID of the sandbox.
	Cwd      string `json:"cwd"`       // Cwd is the sandbox's current working directory.

	Template SandboxTemplate `json:"templateID"` // Template of the sandbox.

	Metadata map[string]string `json:"metadata"` // Metadata of the sandbox.

	Map *sync.Map `json:"-"` // Map is the map of the sandbox.
	// contains filtered or unexported fields
}

type SandboxTemplate string

type SbFn func(ctx context.Context, s *Sandbox, params *Params) (pegwings.ChatCompletionMessage, error)

type ToolingWrapper struct {
	ToolMap map[*tools.Tool]SbFn
}