jx/internal/resource/declaration.go

// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.

package resource

import (
	"context"
	"encoding/json"
	"fmt"
	"gopkg.in/yaml.v3"
	"log/slog"
)

type DeclarationType struct {
	Type TypeName `json:"type" yaml:"type"`
}

type Declaration struct {
	Type TypeName `json:"type" yaml:"type"`
	Attributes Resource `json:"attributes" yaml:"attributes"`
}

type ResourceLoader interface {
	LoadDecl(string) error
}

type StateTransformer interface {
	Apply() error
}

type YamlLoader func(string, any) error

func YamlLoadDecl(yamlFileResourceDeclaration string, resource any) error {
	if err := yaml.Unmarshal([]byte(yamlFileResourceDeclaration), resource); err != nil {
		return err
	}
	return nil
}

func NewDeclaration() *Declaration {
	return &Declaration{}
}

func (d *Declaration) LoadDecl(yamlResourceDeclaration string) error {
	return YamlLoadDecl(yamlResourceDeclaration, d)
}

func (d *Declaration) NewResource() error {
	uri := fmt.Sprintf("%s://", d.Type)
	newResource, err := ResourceTypes.New(uri)
	d.Attributes = newResource
	return err
}

func (d *Declaration) Resource() Resource {
	return d.Attributes
}

func (d *Declaration) SetURI(uri string) error {
	slog.Info("SetURI()", "uri", uri)
	d.Attributes = NewResource(uri)
	if d.Attributes == nil {
		panic("unknown resource")
	}
	d.Type = TypeName(d.Attributes.Type())
	d.Attributes.Read(context.Background()) // fix context
	return nil
}

func (d *Declaration) UnmarshalYAML(value *yaml.Node) error {
	t := &DeclarationType{}
	if unmarshalResourceTypeErr := value.Decode(t); unmarshalResourceTypeErr != nil {
		return unmarshalResourceTypeErr
	}

	d.Type = t.Type
	newResource, resourceErr := ResourceTypes.New(fmt.Sprintf("%s://", d.Type))
	if resourceErr != nil {
		return resourceErr
	}
	d.Attributes = newResource
	resourceAttrs := struct {
		Attributes yaml.Node `json:"attributes"`
	}{}
	if unmarshalAttributesErr := value.Decode(&resourceAttrs); unmarshalAttributesErr != nil {
		return unmarshalAttributesErr
	}
	if unmarshalResourceErr := resourceAttrs.Attributes.Decode(d.Attributes); unmarshalResourceErr != nil {
		return unmarshalResourceErr
	}
	return nil
}

func (d *Declaration) UnmarshalJSON(data []byte) error {
	t := &DeclarationType{}
	if unmarshalResourceTypeErr := json.Unmarshal(data, t); unmarshalResourceTypeErr != nil {
		return unmarshalResourceTypeErr
	}

	d.Type = t.Type

	newResource, resourceErr := ResourceTypes.New(fmt.Sprintf("%s://", d.Type))
	if resourceErr != nil {
		return resourceErr
	}
	d.Attributes = newResource

	resourceAttrs := struct {
		Attributes Resource `json:"attributes"`
	}{Attributes: newResource}
	if unmarshalAttributesErr := json.Unmarshal(data, &resourceAttrs); unmarshalAttributesErr != nil {
		return unmarshalAttributesErr
	}

	return nil
}

/*
func (d *Declaration) MarshalJSON() ([]byte, error) {
  buf := new(bytes.Buffer)
  buf.WriteByte('"')
  buf.WriteString("value"))
  buf.WriteByte('"')
  return buf.Bytes(), nil
}
*/

func (d *Declaration) MarshalYAML() (any, error) {
	return d, nil
}