mvhutz/lambda
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: moved cli to registry

Browse Source
This commit is contained in:
Maxim Hutz
2026-02-06 23:42:19 -05:00
parent 58d0823069
commit a2355fcd56
9 changed files with 53 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/lambda/lambda_reduce.go
View File

@@ -5,8 +5,8 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
"git.maximhutz.com/max/lambda/internal/cli"
"git.maximhutz.com/max/lambda/internal/config" "git.maximhutz.com/max/lambda/internal/config"
"git.maximhutz.com/max/lambda/internal/registry"
) )
func LambdaReduce() *cobra.Command { func LambdaReduce() *cobra.Command {
@@ -53,7 +53,7 @@ func LambdaReduce() *cobra.Command {
} }
// Find engine. // Find engine.
var engine cli.Engine var engine registry.Engine
if engineFlag == "" { if engineFlag == "" {
if engine, err = r.GetDefaultEngine(inputRepr); err != nil { if engine, err = r.GetDefaultEngine(inputRepr); err != nil {
return err return err

9
cmd/lambda/registry.go
View File

@@ -1,7 +1,6 @@
package main package main
import ( import (
"git.maximhutz.com/max/lambda/internal/cli"
"git.maximhutz.com/max/lambda/internal/registry" "git.maximhutz.com/max/lambda/internal/registry"
"git.maximhutz.com/max/lambda/pkg/convert" "git.maximhutz.com/max/lambda/pkg/convert"
"git.maximhutz.com/max/lambda/pkg/engine/normalorder" "git.maximhutz.com/max/lambda/pkg/engine/normalorder"
@@ -13,14 +12,14 @@ func GetRegistry() *registry.Registry {
r := registry.New() r := registry.New()
// Codecs // Codecs
r.MustAddConversions(cli.ConvertCodec(convert.Saccharine2Lambda{}, "saccharine", "lambda")...) (registry.RegisterCodec(r, convert.Saccharine2Lambda{}, "saccharine", "lambda"))
// Engines // Engines
r.MustAddEngine(cli.ConvertEngine(normalorder.Engine{}, "normalorder", "lambda")) (registry.RegisterEngine(r, normalorder.Engine{}, "normalorder", "lambda"))
// Marshalers // Marshalers
r.MustAddMarshaler(cli.ConvertMarshaler(lambda.Marshaler{}, "lambda")) (registry.RegisterMarshaler(r, lambda.Marshaler{}, "lambda"))
r.MustAddMarshaler(cli.ConvertMarshaler(saccharine.Marshaler{}, "saccharine")) (registry.RegisterMarshaler(r, saccharine.Marshaler{}, "saccharine"))
return r return r
} }

12
internal/cli/conversion.go → internal/registry/conversion.go
View File

@@ -1,4 +1,4 @@
package cli package registry
import ( import (
"fmt" "fmt"
@@ -59,9 +59,9 @@ func (c backwardCodec[T, U]) InType() string { return c.outType }
func (c backwardCodec[T, U]) OutType() string { return c.inType } func (c backwardCodec[T, U]) OutType() string { return c.inType }
func ConvertCodec[T, U any](e codec.Codec[T, U], inType, outType string) []Conversion { func RegisterCodec[T, U any](registry *Registry, c codec.Codec[T, U], inType, outType string) error {
return []Conversion{ registry.converter.Add(forwardCodec[T, U]{c, inType, outType})
forwardCodec[T, U]{e, inType, outType}, registry.converter.Add(backwardCodec[T, U]{c, inType, outType})
backwardCodec[T, U]{e, inType, outType},
} return nil
} }

14
internal/registry/converter.go
View File

@@ -1,27 +1,23 @@
package registry package registry
import (
"git.maximhutz.com/max/lambda/internal/cli"
)
type Converter struct { type Converter struct {
data map[string][]cli.Conversion data map[string][]Conversion
} }
func NewConverter() *Converter { func NewConverter() *Converter {
return &Converter{data: map[string][]cli.Conversion{}} return &Converter{data: map[string][]Conversion{}}
} }
func (g *Converter) Add(c cli.Conversion) { func (g *Converter) Add(c Conversion) {
conversionsFromIn, ok := g.data[c.InType()] conversionsFromIn, ok := g.data[c.InType()]
if !ok { if !ok {
conversionsFromIn = []cli.Conversion{} conversionsFromIn = []Conversion{}
} }
conversionsFromIn = append(conversionsFromIn, c) conversionsFromIn = append(conversionsFromIn, c)
g.data[c.InType()] = conversionsFromIn g.data[c.InType()] = conversionsFromIn
} }
func (g *Converter) ConversionsFrom(t string) []cli.Conversion { func (g *Converter) ConversionsFrom(t string) []Conversion {
return g.data[t] return g.data[t]
} }

17
internal/cli/engine.go → internal/registry/engine.go
View File

@@ -1,6 +1,10 @@
package cli package registry
import "git.maximhutz.com/max/lambda/pkg/engine" import (
"fmt"
"git.maximhutz.com/max/lambda/pkg/engine"
)
type Engine interface { type Engine interface {
Load() Process Load() Process
@@ -22,6 +26,11 @@ func (e convertedEngine[T]) Load() Process {
return convertedProcess[T]{e.engine.Load(), e.inType} return convertedProcess[T]{e.engine.Load(), e.inType}
} }
func ConvertEngine[T any](e engine.Engine[T], name, inType string) Engine { func RegisterEngine[T any](registry *Registry, e engine.Engine[T], name, inType string) error {
return &convertedEngine[T]{e, name, inType} if _, ok := registry.engines[name]; ok {
return fmt.Errorf("engine '%s' already registered", name)
}
registry.engines[name] = &convertedEngine[T]{e, name, inType}
return nil
} }

11
internal/cli/marshaler.go → internal/registry/marshaler.go
View File

@@ -1,4 +1,4 @@
package cli package registry
import ( import (
"fmt" "fmt"
@@ -40,6 +40,11 @@ func (c convertedMarshaler[T]) Encode(r Repr) (string, error) {
func (c convertedMarshaler[T]) InType() string { return c.inType } func (c convertedMarshaler[T]) InType() string { return c.inType }
func ConvertMarshaler[T any](e codec.Marshaler[T], inType string) Marshaler { func RegisterMarshaler[T any](registry *Registry, m codec.Marshaler[T], inType string) error {
return convertedMarshaler[T]{e, inType} if _, ok := registry.marshalers[inType]; ok {
return fmt.Errorf("marshaler for '%s' already registered", inType)
}
registry.marshalers[inType] = convertedMarshaler[T]{m, inType}
return nil
} }

2
internal/cli/process.go → internal/registry/process.go
View File

@@ -1,4 +1,4 @@
package cli package registry
import ( import (
"fmt" "fmt"

71
internal/registry/registry.go
View File

@@ -4,68 +4,23 @@ import (
"fmt" "fmt"
"iter" "iter"
"maps" "maps"
"git.maximhutz.com/max/lambda/internal/cli"
) )
type Registry struct { type Registry struct {
marshalers map[string]cli.Marshaler marshalers map[string]Marshaler
converter *Converter converter *Converter
engines map[string]cli.Engine engines map[string]Engine
} }
func New() *Registry { func New() *Registry {
return &Registry{ return &Registry{
marshalers: map[string]cli.Marshaler{}, marshalers: map[string]Marshaler{},
converter: NewConverter(), converter: NewConverter(),
engines: map[string]cli.Engine{}, engines: map[string]Engine{},
} }
} }
func (r *Registry) AddConversions(conversions ...cli.Conversion) error { func (r Registry) GetEngine(name string) (Engine, error) {
for _, conversion := range conversions {
r.converter.Add(conversion)
}
return nil
}
func (r *Registry) MustAddConversions(conversions ...cli.Conversion) {
if err := r.AddConversions(conversions...); err != nil {
panic(err)
}
}
func (r *Registry) AddMarshaler(c cli.Marshaler) error {
if _, ok := r.marshalers[c.InType()]; ok {
return fmt.Errorf("marshaler for '%s' already registered", c.InType())
}
r.marshalers[c.InType()] = c
return nil
}
func (r *Registry) MustAddMarshaler(c cli.Marshaler) {
if err := r.AddMarshaler(c); err != nil {
panic(err)
}
}
func (r *Registry) AddEngine(e cli.Engine) error {
if _, ok := r.engines[e.Name()]; ok {
return fmt.Errorf("engine '%s' already registered", e.Name())
}
r.engines[e.Name()] = e
return nil
}
func (r *Registry) MustAddEngine(e cli.Engine) {
if err := r.AddEngine(e); err != nil {
panic(err)
}
}
func (r Registry) GetEngine(name string) (cli.Engine, error) {
e, ok := r.engines[name] e, ok := r.engines[name]
if !ok { if !ok {
return nil, fmt.Errorf("engine '%s' not found", name) return nil, fmt.Errorf("engine '%s' not found", name)
@@ -74,11 +29,11 @@ func (r Registry) GetEngine(name string) (cli.Engine, error) {
return e, nil return e, nil
} }
func (r Registry) ListEngines() iter.Seq[cli.Engine] { func (r Registry) ListEngines() iter.Seq[Engine] {
return maps.Values(r.engines) return maps.Values(r.engines)
} }
func (r *Registry) GetDefaultEngine(id string) (cli.Engine, error) { func (r *Registry) GetDefaultEngine(id string) (Engine, error) {
for _, engine := range r.engines { for _, engine := range r.engines {
if engine.InType() == id { if engine.InType() == id {
return engine, nil return engine, nil
@@ -88,7 +43,7 @@ func (r *Registry) GetDefaultEngine(id string) (cli.Engine, error) {
return nil, fmt.Errorf("no engine for '%s'", id) return nil, fmt.Errorf("no engine for '%s'", id)
} }
func (r *Registry) ConvertTo(repr cli.Repr, outType string) (cli.Repr, error) { func (r *Registry) ConvertTo(repr Repr, outType string) (Repr, error) {
path, err := r.ConversionPath(repr.Id(), outType) path, err := r.ConversionPath(repr.Id(), outType)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -105,7 +60,7 @@ func (r *Registry) ConvertTo(repr cli.Repr, outType string) (cli.Repr, error) {
return result, err return result, err
} }
func (r *Registry) Marshal(repr cli.Repr) (string, error) { func (r *Registry) Marshal(repr Repr) (string, error) {
m, ok := r.marshalers[repr.Id()] m, ok := r.marshalers[repr.Id()]
if !ok { if !ok {
return "", fmt.Errorf("no marshaler for '%s'", repr.Id()) return "", fmt.Errorf("no marshaler for '%s'", repr.Id())
@@ -114,7 +69,7 @@ func (r *Registry) Marshal(repr cli.Repr) (string, error) {
return m.Encode(repr) return m.Encode(repr)
} }
func (r *Registry) Unmarshal(s string, outType string) (cli.Repr, error) { func (r *Registry) Unmarshal(s string, outType string) (Repr, error) {
m, ok := r.marshalers[outType] m, ok := r.marshalers[outType]
if !ok { if !ok {
return nil, fmt.Errorf("no marshaler for '%s'", outType) return nil, fmt.Errorf("no marshaler for '%s'", outType)
@@ -137,8 +92,8 @@ func reverse[T any](list []T) []T {
return reversed return reversed
} }
func (r *Registry) ConversionPath(from, to string) ([]cli.Conversion, error) { func (r *Registry) ConversionPath(from, to string) ([]Conversion, error) {
backtrack := map[string]cli.Conversion{} backtrack := map[string]Conversion{}
iteration := []string{from} iteration := []string{from}
for len(iteration) > 0 { for len(iteration) > 0 {
nextIteration := []string{} nextIteration := []string{}
@@ -157,7 +112,7 @@ func (r *Registry) ConversionPath(from, to string) ([]cli.Conversion, error) {
iteration = nextIteration iteration = nextIteration
} }
reversedPath := []cli.Conversion{} reversedPath := []Conversion{}
current := to current := to
for current != from { for current != from {
conversion, ok := backtrack[current] conversion, ok := backtrack[current]

2
internal/cli/repr.go → internal/registry/repr.go
View File

@@ -1,4 +1,4 @@
package cli package registry
type Repr interface { type Repr interface {
// Id returns to name of the objects underlying representation. If is // Id returns to name of the objects underlying representation. If is