99 lines
1.6 KiB
Go
99 lines
1.6 KiB
Go
package object
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"code.jmug.me/jmug/interpreter-in-go/pkg/ast"
|
|
)
|
|
|
|
type ObjectType string
|
|
|
|
const (
|
|
INTEGER_OBJ = "INTEGER"
|
|
BOOLEAN_OBJ = "BOOLEAN"
|
|
NULL_OBJ = "NULL"
|
|
RETURN_VALUE_OBJ = "RETURN"
|
|
ERROR_OBJ = "ERROR"
|
|
FUNCTION_OBJ = "FUNCTION"
|
|
)
|
|
|
|
type Object interface {
|
|
Type() ObjectType
|
|
Inspect() string
|
|
}
|
|
|
|
type Integer struct {
|
|
Value int64
|
|
}
|
|
|
|
func (i *Integer) Type() ObjectType {
|
|
return INTEGER_OBJ
|
|
}
|
|
func (i *Integer) Inspect() string {
|
|
return fmt.Sprintf("%d", i.Value)
|
|
}
|
|
|
|
type Boolean struct {
|
|
Value bool
|
|
}
|
|
|
|
func (b *Boolean) Type() ObjectType {
|
|
return BOOLEAN_OBJ
|
|
}
|
|
func (b *Boolean) Inspect() string {
|
|
return fmt.Sprintf("%t", b.Value)
|
|
}
|
|
|
|
type Null struct{}
|
|
|
|
func (n *Null) Type() ObjectType {
|
|
return NULL_OBJ
|
|
}
|
|
func (n *Null) Inspect() string {
|
|
return "null"
|
|
}
|
|
|
|
type ReturnValue struct {
|
|
Value Object
|
|
}
|
|
|
|
func (rv *ReturnValue) Type() ObjectType {
|
|
return RETURN_VALUE_OBJ
|
|
}
|
|
func (rv *ReturnValue) Inspect() string {
|
|
return rv.Value.Inspect()
|
|
}
|
|
|
|
type Error struct {
|
|
Message string
|
|
}
|
|
|
|
func (e *Error) Type() ObjectType {
|
|
return ERROR_OBJ
|
|
}
|
|
func (e *Error) Inspect() string {
|
|
return "ERROR: " + e.Message
|
|
}
|
|
|
|
type Function struct {
|
|
Parameters []*ast.Identifier
|
|
Body *ast.BlockStatement
|
|
Env *Environment
|
|
}
|
|
|
|
func (f *Function) Type() ObjectType {
|
|
return FUNCTION_OBJ
|
|
}
|
|
func (f *Function) Inspect() string {
|
|
var out bytes.Buffer
|
|
params := []string{}
|
|
for _, p := range f.Parameters {
|
|
params = append(params, p.Value)
|
|
}
|
|
out.WriteString("fn")
|
|
out.WriteString("(" + strings.Join(params, ", ") + ")")
|
|
out.WriteString(" {\n" + f.Body.String() + "\n}")
|
|
return out.String()
|
|
}
|