feat: refactor some shit
This commit is contained in:
118
parser/blocks/template.go
Normal file
118
parser/blocks/template.go
Normal file
@@ -0,0 +1,118 @@
|
||||
package blocks
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"go.yaml.in/yaml/v4"
|
||||
)
|
||||
|
||||
func cleanTemplate(input string) string {
|
||||
s := strings.TrimSpace(input)
|
||||
s = strings.TrimPrefix(s, "{")
|
||||
s = strings.TrimSuffix(s, "}")
|
||||
return s
|
||||
}
|
||||
|
||||
func parseShortTemplate(input string) (TemplateBlock, error) {
|
||||
|
||||
var split = strings.Split(cleanTemplate(input), "|")
|
||||
if len(split) < 1 {
|
||||
return TemplateBlock{}, fmt.Errorf("Invalid Short Template")
|
||||
}
|
||||
|
||||
block := TemplateBlock{
|
||||
Type: DataBlock,
|
||||
Path: strings.TrimSpace(split[0]),
|
||||
Codec: CodecText,
|
||||
content: input,
|
||||
}
|
||||
|
||||
if len(split) > 1 {
|
||||
var optionSplit = strings.Split(split[1], ",")
|
||||
for _, option := range optionSplit {
|
||||
switch strings.TrimSpace(option) {
|
||||
case "required":
|
||||
block.Required = true
|
||||
case "number":
|
||||
block.Codec = CodecNumber
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return block, nil
|
||||
}
|
||||
|
||||
type yamlBlock struct {
|
||||
Path string `yaml:"path"`
|
||||
Codec string `yaml:"codec"`
|
||||
Required bool `yaml:"required,omitempty"`
|
||||
Fields []yamlField `yaml:"fields"`
|
||||
Item *struct {
|
||||
Template string `yaml:"template,omitempty"`
|
||||
} `yaml:"item,omitempty"`
|
||||
Template string `yaml:"template,omitempty"`
|
||||
}
|
||||
|
||||
type yamlField struct {
|
||||
Path string `yaml:"path"`
|
||||
Value any `yaml:"value,omitempty"`
|
||||
Codec string `yaml:"codec"`
|
||||
Required bool `yaml:"required"`
|
||||
}
|
||||
|
||||
func parseYamlTemplate(input string) (block TemplateBlock, err error) {
|
||||
|
||||
var blk yamlBlock
|
||||
|
||||
cleaned := cleanTemplate(input)
|
||||
|
||||
dec := yaml.NewDecoder(strings.NewReader(cleaned))
|
||||
dec.KnownFields(true)
|
||||
|
||||
if err := dec.Decode(&blk); err != nil {
|
||||
fmt.Printf("Failed to parse:\n---\n%s\n---\n", cleaned)
|
||||
return block, err
|
||||
}
|
||||
|
||||
if blk.Path == "" {
|
||||
return block, fmt.Errorf("missing top-level 'path'")
|
||||
}
|
||||
|
||||
codec, err := parseCodecType(blk.Codec)
|
||||
if err != nil {
|
||||
return block, fmt.Errorf("failed to parse codec: %w", err)
|
||||
}
|
||||
|
||||
return TemplateBlock{
|
||||
Type: DataBlock,
|
||||
Path: blk.Path,
|
||||
Codec: codec,
|
||||
content: input,
|
||||
}, nil
|
||||
|
||||
}
|
||||
|
||||
func ParseTemplateBlock(template string, blockType BlockType) (block TemplateBlock, err error) {
|
||||
|
||||
if blockType == MatchingBlock {
|
||||
return TemplateBlock{
|
||||
Type: MatchingBlock,
|
||||
content: template,
|
||||
}, nil
|
||||
}
|
||||
|
||||
block.Type = DataBlock
|
||||
block.content = template
|
||||
|
||||
templateType := DetectTemplateType(template)
|
||||
if templateType == InvalidTemplate {
|
||||
return block, fmt.Errorf("Invalid Template")
|
||||
}
|
||||
|
||||
if templateType == ShortTemplate {
|
||||
return parseShortTemplate(template)
|
||||
}
|
||||
|
||||
return parseYamlTemplate(template)
|
||||
}
|
Reference in New Issue
Block a user