mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-16 11:37:36 +02:00
89 lines
1.3 KiB
Go
89 lines
1.3 KiB
Go
package compiler
|
|
|
|
import (
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
)
|
|
|
|
type (
|
|
globalScope struct {
|
|
params map[string]struct{}
|
|
}
|
|
|
|
scope struct {
|
|
global *globalScope
|
|
parent *scope
|
|
vars map[string]struct{}
|
|
}
|
|
)
|
|
|
|
func newGlobalScope() *globalScope {
|
|
return &globalScope{
|
|
params: map[string]struct{}{},
|
|
}
|
|
}
|
|
|
|
func newRootScope(global *globalScope) *scope {
|
|
return &scope{
|
|
global: global,
|
|
vars: make(map[string]struct{}),
|
|
}
|
|
}
|
|
|
|
func newScope(parent *scope) *scope {
|
|
s := newRootScope(parent.global)
|
|
s.parent = parent
|
|
|
|
return s
|
|
}
|
|
|
|
func (s *scope) AddParam(name string) {
|
|
s.global.params[name] = struct{}{}
|
|
}
|
|
|
|
func (s *scope) HasVariable(name string) bool {
|
|
_, exists := s.vars[name]
|
|
|
|
if exists {
|
|
return true
|
|
}
|
|
|
|
if s.parent != nil {
|
|
return s.parent.HasVariable(name)
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (s *scope) SetVariable(name string) error {
|
|
_, exists := s.vars[name]
|
|
|
|
if exists {
|
|
return core.Error(ErrVariableNotUnique, name)
|
|
}
|
|
|
|
// TODO: add type detection
|
|
s.vars[name] = struct{}{}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *scope) RemoveVariable(name string) error {
|
|
_, exists := s.vars[name]
|
|
|
|
if !exists {
|
|
return core.Error(ErrVariableNotFound, name)
|
|
}
|
|
|
|
delete(s.vars, name)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *scope) ClearVariables() {
|
|
s.vars = make(map[string]struct{})
|
|
}
|
|
|
|
func (s *scope) Fork() *scope {
|
|
return newScope(s)
|
|
}
|