mirror of
https://github.com/hanzoai/numscript.git
synced 2026-10-30 18:26:57 +00:00
zeekayandClaude Opus 5
2c2f0e6bf3
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers, and the interpreter's parseArg, evaluateExprAs and evaluateOptExprAs are now generic methods on the parser and program state they take. omitzero is not applied, since it changes what the JSON says. Co-Authored-By: Claude Opus 5 <[email protected]>
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package interpreter
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/formancehq/numscript/internal/parser"
|
|
)
|
|
|
|
type opAdd interface {
|
|
evalAdd(st *programState, other parser.ValueExpr) (Value, InterpreterError)
|
|
}
|
|
|
|
var _ opAdd = (*MonetaryInt)(nil)
|
|
var _ opAdd = (*Monetary)(nil)
|
|
|
|
func (m MonetaryInt) evalAdd(st *programState, other parser.ValueExpr) (Value, InterpreterError) {
|
|
m1 := big.Int(m)
|
|
m2, err := st.evaluateExprAs(other, expectNumber)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sum := new(big.Int).Add(&m1, m2)
|
|
return MonetaryInt(*sum), nil
|
|
}
|
|
|
|
func (m Monetary) evalAdd(st *programState, other parser.ValueExpr) (Value, InterpreterError) {
|
|
m2, err := st.evaluateExprAs(other, expectMonetary)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if m.Asset != m2.Asset {
|
|
return nil, MismatchedCurrencyError{
|
|
Expected: m.Asset.String(),
|
|
Got: m2.Asset.String(),
|
|
}
|
|
}
|
|
|
|
return Monetary{
|
|
Asset: m.Asset,
|
|
Amount: m.Amount.Add(m2.Amount),
|
|
}, nil
|
|
|
|
}
|
|
|
|
type opSub interface {
|
|
evalSub(st *programState, other parser.ValueExpr) (Value, InterpreterError)
|
|
}
|
|
|
|
var _ opSub = (*MonetaryInt)(nil)
|
|
var _ opSub = (*Monetary)(nil)
|
|
|
|
func (m MonetaryInt) evalSub(st *programState, other parser.ValueExpr) (Value, InterpreterError) {
|
|
m1 := big.Int(m)
|
|
m2, err := st.evaluateExprAs(other, expectNumber)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sum := new(big.Int).Sub(&m1, m2)
|
|
return MonetaryInt(*sum), nil
|
|
}
|
|
|
|
func (m Monetary) evalSub(st *programState, other parser.ValueExpr) (Value, InterpreterError) {
|
|
m2, err := st.evaluateExprAs(other, expectMonetary)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if m.Asset != m2.Asset {
|
|
return nil, MismatchedCurrencyError{
|
|
Expected: m.Asset.String(),
|
|
Got: m2.Asset.String(),
|
|
}
|
|
}
|
|
|
|
return Monetary{
|
|
Asset: m.Asset,
|
|
Amount: m.Amount.Sub(m2.Amount),
|
|
}, nil
|
|
|
|
}
|
|
|
|
type opNeg interface {
|
|
evalNeg(st *programState) (Value, InterpreterError)
|
|
}
|
|
|
|
var _ opNeg = (*MonetaryInt)(nil)
|
|
var _ opNeg = (*Monetary)(nil)
|
|
|
|
func (m MonetaryInt) evalNeg(st *programState) (Value, InterpreterError) {
|
|
m1 := big.Int(m)
|
|
neg := new(big.Int).Neg(&m1)
|
|
return MonetaryInt(*neg), nil
|
|
}
|
|
|
|
func (m Monetary) evalNeg(st *programState) (Value, InterpreterError) {
|
|
m1 := big.Int(m.Amount)
|
|
neg := new(big.Int).Neg(&m1)
|
|
return Monetary{
|
|
Asset: m.Asset,
|
|
Amount: MonetaryInt(*neg),
|
|
}, nil
|
|
|
|
}
|