Compare commits
17 Commits
Author | SHA1 | Date | |
---|---|---|---|
08b2f5301f | |||
3f3278b73e | |||
9943452ff9 | |||
8feb7b8d56 | |||
b61010b99a | |||
a3c2d17c69 | |||
a73acb8b93 | |||
d88b67ea2f | |||
6925598bf2 | |||
2e84afa87f | |||
784d586f48 | |||
1117882ced | |||
37ed8bfa83 | |||
0d748fb0bf | |||
05e1539134 | |||
ea97070fea | |||
dae9c1cf45 |
2
Makefile
2
Makefile
@ -32,6 +32,8 @@ run:
|
||||
docker run -it -v $(HOME)/.git-credentials:/root/.git-credentials -v $(HOME)/.gitconfig:/root/.gitconfig -v /var/run/docker.sock:/var/run/docker.sock -v $(shell pwd):/src $(IMAGE) sh
|
||||
run-alpine:
|
||||
docker run -it -v $(HOME)/.git-credentials:/root/.git-credentials -v $(HOME)/.gitconfig:/root/.gitconfig -v /var/run/docker.sock:/var/run/docker.sock -v $(shell pwd):/src golang:1.22.6-alpine sh
|
||||
build-container:
|
||||
docker run -it -v $(HOME)/.git-credentials:/root/.git-credentials -v $(HOME)/.gitconfig:/root/.gitconfig -v /var/run/docker.sock:/var/run/docker.sock -v $(shell pwd):/src -w /src rosskeenhouse/build-golang:1.22.6-alpine sh
|
||||
clean:
|
||||
go clean -modcache
|
||||
rm jx
|
||||
|
10
README.md
10
README.md
@ -10,7 +10,9 @@ These tools work with YAML descriptions of resources (E.g. files, users, contain
|
||||
|
||||
# Releases
|
||||
|
||||
**v0 releases are unstable and changes may be made to interfaces and specifications.**
|
||||
**<span style="color:red">v0 releases are unstable and changes may be made to interfaces and specifications.</span>**
|
||||
|
||||
Use at your own risk.
|
||||
|
||||
# JX Documents
|
||||
|
||||
@ -41,14 +43,14 @@ resources:
|
||||
|
||||
# Testing
|
||||
|
||||
Testing the current version involves checking out main and building.
|
||||
Testing the current version involves checking out main and building inside of the alpine go build container.
|
||||
|
||||
```
|
||||
git clone https://gitea.rosskeen.house/doublejynx/jx.git
|
||||
|
||||
make test
|
||||
make build-container
|
||||
|
||||
make build
|
||||
make test
|
||||
```
|
||||
|
||||
# Command-line
|
||||
|
16
build/docker/container-image-build.jx.yaml
Normal file
16
build/docker/container-image-build.jx.yaml
Normal file
@ -0,0 +1,16 @@
|
||||
---
|
||||
imports:
|
||||
- /etc/jx/dockerhub.jx.yaml
|
||||
resources:
|
||||
- type: container-image
|
||||
config: dockerhub
|
||||
transition: update
|
||||
attributes:
|
||||
name: rosskeenhouse/build-golang:1.22.6-alpine
|
||||
push: true
|
||||
dockerfile: |-
|
||||
FROM golang:1.22.6-alpine
|
||||
COPY . /opt/build
|
||||
WORKDIR /opt/build
|
||||
RUN ./jx apply ./alpine.jx.yaml
|
||||
contextref: file://build/docker/golang/build
|
36
build/docker/golang/build/alpine.jx.yaml
Normal file
36
build/docker/golang/build/alpine.jx.yaml
Normal file
@ -0,0 +1,36 @@
|
||||
imports:
|
||||
- file://common.jx.yaml
|
||||
resources:
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: musl-dev
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: luajit
|
||||
verion: =~2.2
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: luajit-dev
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: protobuf
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: openjdk8
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: docker
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: openssh-client
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: golangci-lint
|
45
build/docker/golang/build/common.jx.yaml
Normal file
45
build/docker/golang/build/common.jx.yaml
Normal file
@ -0,0 +1,45 @@
|
||||
resources:
|
||||
- type: file
|
||||
transition: create
|
||||
attributes:
|
||||
path: /usr/local/bin/antlr-4.10-complete.jar
|
||||
sourceref: https://www.antlr.org/download/antlr-4.10-complete.jar
|
||||
owner: root
|
||||
group: root
|
||||
mode: 0755
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: make
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: openssl
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: curl
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: git
|
||||
- type: package
|
||||
transition: create
|
||||
attributes:
|
||||
name: gcc
|
||||
- type: exec
|
||||
transition: create
|
||||
attributes:
|
||||
create:
|
||||
path: go
|
||||
args:
|
||||
- install
|
||||
- google.golang.org/protobuf/cmd/protoc-gen-go@latest
|
||||
- type: exec
|
||||
transition: create
|
||||
attributes:
|
||||
create:
|
||||
path: go
|
||||
args:
|
||||
- install
|
||||
- golang.org/x/vuln/cmd/govulncheck@latest
|
23
cli_test.go
23
cli_test.go
@ -184,6 +184,25 @@ func TestResourcesRead(t *testing.T) {
|
||||
t.Skip("cli not built")
|
||||
}
|
||||
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||
defer req.Body.Close()
|
||||
assert.Equal(t, req.URL.String(), "/resource/user")
|
||||
_, err := io.ReadAll(req.Body)
|
||||
assert.Nil(t, err)
|
||||
userdecl := []byte(`
|
||||
type: "user"
|
||||
attributes:
|
||||
name: "foo"
|
||||
gecos: "foo user"
|
||||
`)
|
||||
|
||||
_, writeErr := rw.Write(userdecl)
|
||||
assert.Nil(t, writeErr)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
|
||||
assert.Nil(t, TempDir.CreateFile("testread", "data"))
|
||||
|
||||
resources := fmt.Sprintf(`
|
||||
@ -215,7 +234,7 @@ resources:
|
||||
- type: http
|
||||
transition: read
|
||||
attributes:
|
||||
endpoint: https://gitea.rosskeen.house
|
||||
endpoint: %s/resource/user
|
||||
- type: route
|
||||
transition: read
|
||||
attributes:
|
||||
@ -227,7 +246,7 @@ resources:
|
||||
rtid: all
|
||||
routetype: local
|
||||
metric: 100
|
||||
`, TempDir.FilePath("testread"))
|
||||
`, TempDir.FilePath("testread"), server.URL)
|
||||
|
||||
assert.Nil(t, TempDir.CreateFile("resources.jx.yaml", resources))
|
||||
|
||||
|
@ -159,7 +159,7 @@ func main() {
|
||||
|
||||
DefaultConfigurations, configErr := builtin.BuiltInDocuments()
|
||||
if configErr != nil {
|
||||
slog.Error("Failed loading default configuration", "error", configErr)
|
||||
slog.Warn("Failed loading default configuration", "error", configErr)
|
||||
}
|
||||
|
||||
ConfigDoc.AppendConfigurations(DefaultConfigurations)
|
||||
@ -167,8 +167,8 @@ func main() {
|
||||
for _, subCmd := range jxSubCommands {
|
||||
cmdFlagSet := flag.NewFlagSet(subCmd.Name, flag.ExitOnError)
|
||||
|
||||
cmdFlagSet.StringVar(&ConfigPath, "config", "/etc/jx", "Config file path")
|
||||
cmdFlagSet.StringVar(&ConfigPath, "c", "/etc/jx", "Config file path")
|
||||
cmdFlagSet.StringVar(&ConfigPath, "config", "/etc/jx/conf.d", "Config file path")
|
||||
cmdFlagSet.StringVar(&ConfigPath, "c", "/etc/jx/conf.d", "Config file path")
|
||||
|
||||
GlobalOformat = cmdFlagSet.String("oformat", "yaml", "Output serialization format")
|
||||
cmdFlagSet.StringVar(&GlobalOutput, "output", "-", "Output target (default stdout)")
|
||||
|
26
cmd/cli/main_test.go
Normal file
26
cmd/cli/main_test.go
Normal file
@ -0,0 +1,26 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
_ "github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
_ "decl/internal/folio"
|
||||
_ "decl/internal/data"
|
||||
_ "log/slog"
|
||||
)
|
||||
|
||||
func TestLoadSourceURIConverter(t *testing.T) {
|
||||
/*
|
||||
var uri folio.URI = "file://../../examples/file.jx.yaml"
|
||||
docs, err := LoadSourceURIConverter(uri)
|
||||
assert.Nil(t, err)
|
||||
assert.Greater(t, len(docs), 0)
|
||||
slog.Info("TestLoadSourceURIConverter", "doc", docs[0], "resource", docs[0].(*folio.Document).ResourceDeclarations[0].Attributes)
|
||||
resDecl := docs[0].(*folio.Document).ResourceDeclarations[0]
|
||||
assert.Equal(t, "file", resDecl.Attributes.Type())
|
||||
v, ok := docs[0].Get("file:///tmp/foo.txt")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, "/tmp/foo.txt", v.(data.Declaration).Resource().(data.FileResource).FilePath())
|
||||
*/
|
||||
}
|
@ -1,8 +1,8 @@
|
||||
resources:
|
||||
- type: file
|
||||
transition: create
|
||||
attributes:
|
||||
path: /tmp/foo.txt
|
||||
owner: nobody
|
||||
group: nobody
|
||||
mode: 0644
|
||||
state: present
|
||||
|
@ -1,7 +1,7 @@
|
||||
resources:
|
||||
- type: user
|
||||
transition: create
|
||||
attributes:
|
||||
name: "testuser"
|
||||
uid: "12001"
|
||||
home: "/home/testuser"
|
||||
state: present
|
||||
|
54
internal/builtin/documents/config.jx.yaml
Normal file
54
internal/builtin/documents/config.jx.yaml
Normal file
@ -0,0 +1,54 @@
|
||||
configurations:
|
||||
- name: confdir
|
||||
values:
|
||||
prefix: /etc/jx
|
||||
resources:
|
||||
- type: group
|
||||
transition: create
|
||||
onerror: stop
|
||||
attributes:
|
||||
name: "jx"
|
||||
- type: file
|
||||
transition: update
|
||||
attributes:
|
||||
path: "/etc/jx"
|
||||
owner: "root"
|
||||
group: "root"
|
||||
mode: "0755"
|
||||
filetype: directory
|
||||
- type: file
|
||||
transition: update
|
||||
config: confdir
|
||||
attributes:
|
||||
path: "conf.d"
|
||||
owner: "root"
|
||||
group: "jx"
|
||||
mode: "0770"
|
||||
filetype: directory
|
||||
- type: file
|
||||
transition: update
|
||||
config: confdir
|
||||
attributes:
|
||||
path: "lib"
|
||||
owner: "root"
|
||||
group: "jx"
|
||||
mode: "0770"
|
||||
filetype: directory
|
||||
- type: file
|
||||
transition: update
|
||||
config: confdir
|
||||
attributes:
|
||||
path: "pki"
|
||||
owner: "root"
|
||||
group: "jx"
|
||||
mode: "0770"
|
||||
filetype: directory
|
||||
- type: file
|
||||
transition: update
|
||||
config: confdir
|
||||
attributes:
|
||||
path: "pki/ca"
|
||||
owner: "root"
|
||||
group: "jx"
|
||||
mode: "0770"
|
||||
filetype: directory
|
16
internal/builtin/documents/install.jx.yaml
Normal file
16
internal/builtin/documents/install.jx.yaml
Normal file
@ -0,0 +1,16 @@
|
||||
imports:
|
||||
- file://documents/config.jx.yaml
|
||||
configurations:
|
||||
- name: bindir
|
||||
values:
|
||||
prefix: /usr/local/bin
|
||||
resources:
|
||||
- type: file
|
||||
transition: update
|
||||
config: bindir
|
||||
attributes:
|
||||
path: "jx"
|
||||
owner: "root"
|
||||
group: "root"
|
||||
mode: "0755"
|
||||
sourceref: file://jx
|
@ -126,7 +126,7 @@ func (a *App) ImportResource(ctx context.Context, uri string) (err error) {
|
||||
a.Documents = append(a.Documents, folio.DocumentRegistry.NewDocument(""))
|
||||
}
|
||||
resourceURI := folio.URI(uri)
|
||||
u := resourceURI.Parse()
|
||||
u := resourceURI.Parse().URL()
|
||||
if u == nil {
|
||||
return fmt.Errorf("Failed adding resource: %s", uri)
|
||||
}
|
||||
@ -147,9 +147,19 @@ func (a *App) ImportResource(ctx context.Context, uri string) (err error) {
|
||||
}
|
||||
|
||||
func (a *App) ImportSource(uri string) (loadedDocuments []data.Document, err error) {
|
||||
if loadedDocuments, err = folio.DocumentRegistry.Load(folio.URI(uri)); err == nil && loadedDocuments != nil {
|
||||
if source := folio.URI(uri).Parse().URL(); source != nil {
|
||||
if source.Scheme == "" {
|
||||
source.Scheme = "file"
|
||||
}
|
||||
|
||||
slog.Info("Client.ImportSource()", "uri", uri, "source", source, "error", err)
|
||||
if loadedDocuments, err = folio.DocumentRegistry.LoadFromParsedURI(source); err == nil && loadedDocuments != nil {
|
||||
a.Documents = append(a.Documents, loadedDocuments...)
|
||||
}
|
||||
} else {
|
||||
err = folio.ErrInvalidURI
|
||||
}
|
||||
|
||||
slog.Info("Client.ImportSource()", "uri", uri, "error", err)
|
||||
return
|
||||
}
|
||||
@ -183,7 +193,7 @@ func (a *App) Apply(ctx context.Context, deleteResources bool) (err error) {
|
||||
continue
|
||||
}
|
||||
|
||||
slog.Info("Client.Apply()", "document", d, "state", overrideState, "error", err)
|
||||
slog.Info("Client.Apply()", "uri", d.GetURI(), "document", d, "state", overrideState, "error", err)
|
||||
if e := d.(*folio.Document).Apply(overrideState); e != nil {
|
||||
slog.Info("Client.Apply() error", "error", e)
|
||||
return e
|
||||
|
@ -16,6 +16,7 @@ import (
|
||||
"strings"
|
||||
"text/template"
|
||||
"decl/internal/codec"
|
||||
"syscall"
|
||||
)
|
||||
|
||||
var ErrUnknownCommand error = errors.New("Unable to find command in path")
|
||||
@ -26,6 +27,8 @@ type CommandExists func() error
|
||||
|
||||
type CommandArg string
|
||||
|
||||
type CommandInput string
|
||||
|
||||
type Command struct {
|
||||
Path string `json:"path" yaml:"path"`
|
||||
Args []CommandArg `json:"args" yaml:"args"`
|
||||
@ -33,9 +36,13 @@ type Command struct {
|
||||
Split bool `json:"split" yaml:"split"`
|
||||
FailOnError bool `json:"failonerror" yaml:"failonerror"`
|
||||
StdinAvailable bool `json:"stdinavailable,omitempty" yaml:"stdinavailable,omitempty"`
|
||||
ExitCode int `json:"exitcode,omitempty" yaml:"exitcode,omitempty"`
|
||||
Stdout string `json:"stdout,omitempty" yaml:"stdout,omitempty"`
|
||||
Stderr string `json:"stderr,omitempty" yaml:"stderr,omitempty"`
|
||||
Executor CommandExecutor `json:"-" yaml:"-"`
|
||||
Extractor CommandExtractAttributes `json:"-" yaml:"-"`
|
||||
CommandExists CommandExists `json:"-" yaml:"-"`
|
||||
Input CommandInput `json:"-" yaml:"-"`
|
||||
stdin io.Reader `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
@ -45,7 +52,14 @@ func NewCommand() *Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *Command) ClearOutput() {
|
||||
c.Stdout = ""
|
||||
c.Stderr = ""
|
||||
c.ExitCode = 0
|
||||
}
|
||||
|
||||
func (c *Command) Defaults() {
|
||||
c.ClearOutput()
|
||||
c.Split = true
|
||||
c.FailOnError = true
|
||||
c.CommandExists = func() error {
|
||||
@ -55,10 +69,14 @@ func (c *Command) Defaults() {
|
||||
return nil
|
||||
}
|
||||
c.Executor = func(value any) ([]byte, error) {
|
||||
c.ClearOutput()
|
||||
args, err := c.Template(value)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if inputErr := c.SetInput(value); inputErr != nil {
|
||||
return nil, inputErr
|
||||
}
|
||||
cmd := exec.Command(c.Path, args...)
|
||||
c.SetCmdEnv(cmd)
|
||||
|
||||
@ -91,6 +109,10 @@ func (c *Command) Defaults() {
|
||||
}
|
||||
waitErr := cmd.Wait()
|
||||
|
||||
c.Stdout = string(stdOutOutput)
|
||||
c.Stderr = string(stdErrOutput)
|
||||
c.ExitCode = c.GetExitCodeFromError(waitErr)
|
||||
|
||||
if len(stdOutOutput) > 100 {
|
||||
slog.Info("execute()", "path", c.Path, "args", args, "output", string(stdOutOutput[:100]), "error", string(stdErrOutput))
|
||||
} else {
|
||||
@ -126,6 +148,15 @@ func (c *Command) Exists() bool {
|
||||
return c.CommandExists() == nil
|
||||
}
|
||||
|
||||
func (c *Command) GetExitCodeFromError(err error) (ec int) {
|
||||
if exitErr, ok := err.(*exec.ExitError); ok {
|
||||
if status, ok := exitErr.Sys().(syscall.WaitStatus); ok {
|
||||
return status.ExitStatus()
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Command) Template(value any) ([]string, error) {
|
||||
var args []string = make([]string, 0, len(c.Args) * 2)
|
||||
for i, arg := range c.Args {
|
||||
@ -154,6 +185,22 @@ func (c *Command) Execute(value any) ([]byte, error) {
|
||||
return c.Executor(value)
|
||||
}
|
||||
|
||||
func (c *Command) SetInput(value any) error {
|
||||
if len(c.Input) > 0 {
|
||||
if r, err := c.Input.Template(value); err != nil {
|
||||
return err
|
||||
} else {
|
||||
c.SetStdinReader(strings.NewReader(r.String()))
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *CommandInput) Template(value any) (result strings.Builder, err error) {
|
||||
err = template.Must(template.New("commandInput").Parse(string(*c))).Execute(&result, value)
|
||||
return
|
||||
}
|
||||
|
||||
func (c *CommandArg) UnmarshalValue(value string) error {
|
||||
*c = CommandArg(value)
|
||||
return nil
|
||||
|
@ -81,3 +81,22 @@ stdinavailable: true
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, expected, string(out))
|
||||
}
|
||||
|
||||
func TestCommandExitCode(t *testing.T) {
|
||||
c := NewCommand()
|
||||
assert.NotNil(t, c)
|
||||
decl := `
|
||||
path: ls
|
||||
args:
|
||||
- "amissingfile"
|
||||
`
|
||||
|
||||
assert.Nil(t, c.LoadDecl(decl))
|
||||
assert.Equal(t, "ls", c.Path)
|
||||
|
||||
out, err := c.Execute(nil)
|
||||
assert.NotNil(t, err)
|
||||
assert.Greater(t, c.ExitCode, 0)
|
||||
assert.Equal(t, string(out), c.Stdout)
|
||||
assert.Equal(t, string("ls: amissingfile: No such file or directory\n"), c.Stderr)
|
||||
}
|
||||
|
@ -37,7 +37,7 @@ func (c *Certificate) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Certificate) SetParsedURI(uri *url.URL) error {
|
||||
func (c *Certificate) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -39,7 +39,7 @@ func (x *Exec) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (x *Exec) SetParsedURI(uri *url.URL) error {
|
||||
func (x *Exec) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -35,7 +35,7 @@ func (g *Generic[Value]) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (g *Generic[Value]) SetParsedURI(uri *url.URL) error {
|
||||
func (g *Generic[Value]) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -10,6 +10,7 @@ import (
|
||||
"decl/internal/data"
|
||||
"decl/internal/folio"
|
||||
"runtime"
|
||||
"decl/internal/system"
|
||||
)
|
||||
|
||||
// Collects facts about the system
|
||||
@ -35,9 +36,24 @@ func NewSystem() *System {
|
||||
for k, v := range buildValues {
|
||||
s[k] = v
|
||||
}
|
||||
s.CurrentUser()
|
||||
s["importpath"] = []string {
|
||||
"/etc/jx/lib",
|
||||
}
|
||||
return &s
|
||||
}
|
||||
|
||||
func (s *System) CurrentUser() {
|
||||
processUser := system.ProcessUser()
|
||||
processGroup := system.ProcessGroup(processUser)
|
||||
(*s)["user"] = processUser.Username
|
||||
(*s)["gecos"] = processUser.Name
|
||||
(*s)["home"] = processUser.HomeDir
|
||||
(*s)["uid"] = processUser.Uid
|
||||
(*s)["group"] = processGroup.Name
|
||||
(*s)["gid"] = processUser.Gid
|
||||
}
|
||||
|
||||
func (s *System) URI() string {
|
||||
return fmt.Sprintf("%s://%s", s.Type(), "")
|
||||
}
|
||||
@ -46,7 +62,7 @@ func (s *System) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *System) SetParsedURI(uri *url.URL) error {
|
||||
func (s *System) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -8,7 +8,6 @@ import (
|
||||
"decl/internal/codec"
|
||||
"io"
|
||||
"decl/internal/mapper"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
var (
|
||||
@ -44,7 +43,7 @@ type Document interface {
|
||||
mapper.Mapper
|
||||
|
||||
NewResource(uri string) (Resource, error)
|
||||
NewResourceFromParsedURI(uri *url.URL) (Resource, error)
|
||||
NewResourceFromParsedURI(uri URIParser) (Resource, error)
|
||||
AddDeclaration(Declaration)
|
||||
AddResourceDeclaration(resourceType string, resourceDeclaration Resource)
|
||||
|
||||
|
@ -5,16 +5,34 @@ package data
|
||||
import (
|
||||
"errors"
|
||||
"net/url"
|
||||
"decl/internal/transport"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrInvalidURI error = errors.New("Invalid URI")
|
||||
)
|
||||
|
||||
type URIParser interface {
|
||||
URL() *url.URL
|
||||
NewResource(document Document) (newResource Resource, err error)
|
||||
ConstructResource(res Resource) (err error)
|
||||
Converter() (converter Converter, err error)
|
||||
Exists() bool
|
||||
|
||||
ContentReaderStream() (*transport.Reader, error)
|
||||
ContentWriterStream() (*transport.Writer, error)
|
||||
|
||||
String() string
|
||||
SetURL(url *url.URL)
|
||||
Extension() (string, string)
|
||||
|
||||
ContentType() string
|
||||
IsEmpty() bool
|
||||
}
|
||||
|
||||
type Identifier interface {
|
||||
URI() string
|
||||
SetURI(string) error
|
||||
SetParsedURI(*url.URL) error
|
||||
SetParsedURI(URIParser) error
|
||||
}
|
||||
|
||||
type DocumentElement interface {
|
||||
|
@ -22,6 +22,11 @@ type StateTransformer interface {
|
||||
Apply() error
|
||||
}
|
||||
|
||||
// Used by the resource factory to initialize new resources.
|
||||
type ResourceInitializer interface {
|
||||
Init(uri URIParser) error
|
||||
}
|
||||
|
||||
type Resource interface {
|
||||
Identifier
|
||||
Type() string
|
||||
|
@ -11,6 +11,7 @@ type Factory[Product comparable] func(*url.URL) Product
|
||||
type TypesRegistry[Product comparable] interface {
|
||||
New(uri string) (result Product, err error)
|
||||
NewFromParsedURI(uri *url.URL) (result Product, err error)
|
||||
NewFromType(typename string) (result Product, err error)
|
||||
Has(typename string) bool
|
||||
//Get(string) Factory[Product]
|
||||
}
|
||||
|
63
internal/ds/orderedset.go
Normal file
63
internal/ds/orderedset.go
Normal file
@ -0,0 +1,63 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package ds
|
||||
|
||||
import (
|
||||
"log/slog"
|
||||
)
|
||||
|
||||
type OrderedSet[Value comparable] struct {
|
||||
Values []*Value
|
||||
elements map[Value]int
|
||||
}
|
||||
|
||||
func NewOrderedSet[Value comparable]() *OrderedSet[Value] {
|
||||
return &OrderedSet[Value]{ elements: make(map[Value]int), Values: make([]*Value, 0, 10) }
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) Add(value Value) {
|
||||
slog.Info("OrderedSet.Add", "key", value, "s", s)
|
||||
s.Values = append(s.Values, &value)
|
||||
s.elements[value] = len(s.Values)
|
||||
slog.Info("OrderedSet.Add", "key", value, "s", s, "v", &s.Values)
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) Delete(key Value) {
|
||||
slog.Info("OrderedSet.Delete", "key", key, "s", s, "size", len(s.Values))
|
||||
if i, ok := s.elements[key]; ok {
|
||||
i--
|
||||
s.Values[i] = nil
|
||||
delete(s.elements, key)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) Contains(value Value) (result bool) {
|
||||
slog.Info("OrderedSet.Contains", "key", value, "s", s, "size", len(s.Values), "v", &s.Values)
|
||||
_, result = s.elements[value]
|
||||
return
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) Len() int {
|
||||
return len(s.elements)
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) AddItems(value []Value) {
|
||||
for _, v := range value {
|
||||
s.Add(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *OrderedSet[Value]) Items() []*Value {
|
||||
slog.Info("OrderedSet.Items - start", "s", s)
|
||||
result := make([]*Value, 0, len(s.elements) - 1)
|
||||
for _, v := range s.Values {
|
||||
slog.Info("OrderedSet.Items", "value", v)
|
||||
if v != nil {
|
||||
result = append(result, v)
|
||||
s.elements[*v] = len(result)
|
||||
}
|
||||
}
|
||||
slog.Info("OrderedSet.Items", "s", s, "result", result)
|
||||
s.Values = result
|
||||
return result
|
||||
}
|
51
internal/ds/orderedset_test.go
Normal file
51
internal/ds/orderedset_test.go
Normal file
@ -0,0 +1,51 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package ds
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
"log/slog"
|
||||
)
|
||||
|
||||
var (
|
||||
)
|
||||
|
||||
func TestNewOrderedSet(t *testing.T) {
|
||||
s := NewOrderedSet[string]()
|
||||
assert.NotNil(t, s)
|
||||
|
||||
testValues := []string{
|
||||
"foo",
|
||||
"bar",
|
||||
"baz",
|
||||
"quuz",
|
||||
}
|
||||
|
||||
for _,value := range testValues {
|
||||
|
||||
s.Add(value)
|
||||
|
||||
|
||||
slog.Info("TestNewOrderedSet - ADD", "item", value, "s", s)
|
||||
|
||||
assert.True(t, s.Contains(value))
|
||||
slog.Info("TestNewOrderedSet - CONTAINS", "s", s)
|
||||
|
||||
for x, item := range s.Items() {
|
||||
slog.Info("TestNewOrderedSet", "item", item, "s", s)
|
||||
assert.Equal(t, testValues[x], *item)
|
||||
}
|
||||
|
||||
}
|
||||
s.Delete("bar")
|
||||
|
||||
expectedValues := []string {
|
||||
"foo",
|
||||
"baz",
|
||||
"quuz",
|
||||
}
|
||||
for x, item := range s.Items() {
|
||||
assert.Equal(t, expectedValues[x], *item)
|
||||
}
|
||||
}
|
34
internal/ds/set.go
Normal file
34
internal/ds/set.go
Normal file
@ -0,0 +1,34 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package ds
|
||||
|
||||
import (
|
||||
)
|
||||
|
||||
type Set[Value comparable] map[Value]bool
|
||||
|
||||
func NewSet[Value comparable]() Set[Value] {
|
||||
return make(map[Value]bool)
|
||||
}
|
||||
|
||||
func (s Set[Value]) Add(value Value) {
|
||||
s[value] = true
|
||||
}
|
||||
|
||||
func (s Set[Value]) Delete(value Value) {
|
||||
delete(s, value)
|
||||
}
|
||||
|
||||
func (s Set[Value]) Contains(value Value) bool {
|
||||
return s[value]
|
||||
}
|
||||
|
||||
func (s Set[Value]) Len() int {
|
||||
return len(s)
|
||||
}
|
||||
|
||||
func (s Set[Value]) AddSlice(value []Value) {
|
||||
for _, v := range value {
|
||||
s.Add(v)
|
||||
}
|
||||
}
|
24
internal/ds/set_test.go
Normal file
24
internal/ds/set_test.go
Normal file
@ -0,0 +1,24 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package ds
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var (
|
||||
)
|
||||
|
||||
func TestNewSet(t *testing.T) {
|
||||
s := NewSet[string]()
|
||||
assert.NotNil(t, s)
|
||||
s["foo"] = true
|
||||
assert.True(t, s.Contains("foo"))
|
||||
|
||||
s.Add("bar")
|
||||
|
||||
assert.True(t, s.Contains("bar"))
|
||||
|
||||
|
||||
}
|
26
internal/ext/file.go
Normal file
26
internal/ext/file.go
Normal file
@ -0,0 +1,26 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package ext
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type FilePath string
|
||||
|
||||
func (f *FilePath) Exists() bool {
|
||||
_, err := os.Stat(string(*f))
|
||||
return !os.IsNotExist(err)
|
||||
}
|
||||
|
||||
func (f *FilePath) Add(relative string) {
|
||||
newPath := filepath.Join(string(*f), relative)
|
||||
*f = FilePath(newPath)
|
||||
}
|
||||
|
||||
|
||||
func (f FilePath) Abs() FilePath {
|
||||
result, _ := filepath.Abs(string(f))
|
||||
return FilePath(result)
|
||||
}
|
@ -105,12 +105,6 @@ func (d *Dir) Emit(document data.Document, filter data.ElementSelector) (resourc
|
||||
return nil, ErrEmptyDocument
|
||||
}
|
||||
|
||||
dirFileDeclaration := folio.NewDeclaration()
|
||||
dirFileDeclaration.Type = "file"
|
||||
if err = dirFileDeclaration.NewResource(nil); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
parentPaths := make(map[string]int)
|
||||
var containingDirectoryPath string
|
||||
for _,res := range document.Filter(func(d data.Declaration) bool {
|
||||
@ -130,8 +124,12 @@ func (d *Dir) Emit(document data.Document, filter data.ElementSelector) (resourc
|
||||
|
||||
containingDirectoryPath, _ = d.isParent(&parentPaths, parent, containingDirectoryPath)
|
||||
}
|
||||
|
||||
uri := fmt.Sprintf("file://%s", containingDirectoryPath)
|
||||
if err = dirFileDeclaration.SetURI(uri); err != nil {
|
||||
|
||||
dirFileDeclaration := folio.NewDeclaration()
|
||||
dirFileDeclaration.Type = "file"
|
||||
if err = dirFileDeclaration.NewResource(&uri); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
|
@ -120,8 +120,10 @@ func (j *JxFile) setdecoder(source data.ContentIdentifier) {
|
||||
for _,v := range strings.Split(source.ContentType(), ".") {
|
||||
_ = j.Format.Set(v)
|
||||
}
|
||||
slog.Info("JxFile.setdecoder()", "type", source.ContentType(), "format", j.Format)
|
||||
j.decoder = codec.NewDecoder(j.reader, j.Format)
|
||||
}
|
||||
slog.Info("JxFile.setdecoder()", "decoder", j.decoder)
|
||||
}
|
||||
|
||||
func (j *JxFile) Type() data.TypeName { return "jx" }
|
||||
@ -179,6 +181,7 @@ func (j *JxFile) ExtractMany(resourceSource data.Resource, filter data.ElementSe
|
||||
}
|
||||
break
|
||||
}
|
||||
slog.Info("JxFile.ExtractMany() loading", "document", j.index)
|
||||
}
|
||||
slog.Info("JxFile.ExtractMany()", "jxfile", j, "error", err)
|
||||
return
|
||||
@ -186,7 +189,7 @@ func (j *JxFile) ExtractMany(resourceSource data.Resource, filter data.ElementSe
|
||||
|
||||
func (j *JxFile) targetResource() (target data.Resource, err error) {
|
||||
if j.emitResource == nil {
|
||||
targetUrl := j.Uri.Parse()
|
||||
targetUrl := j.Uri.Parse().URL()
|
||||
targetUrl.Scheme = "file"
|
||||
q := targetUrl.Query()
|
||||
q.Set("format", string(j.Format))
|
||||
|
@ -15,6 +15,32 @@ import (
|
||||
"decl/internal/schema"
|
||||
)
|
||||
|
||||
type ConfigKey string
|
||||
|
||||
// Lookup a config value using block.key identifier (E.g. system.GOOS)
|
||||
func (c ConfigKey) GetValue() (value any, err error) {
|
||||
fields := strings.SplitN(string(c), ".", 2)
|
||||
if configBlock, ok := DocumentRegistry.ConfigNameMap.Get(fields[0]); ok && len(fields) > 1 {
|
||||
return configBlock.GetValue(fields[1])
|
||||
} else {
|
||||
return nil, fmt.Errorf("%w - %s", data.ErrUnknownConfigurationType, c)
|
||||
}
|
||||
}
|
||||
|
||||
func (c ConfigKey) Get() any {
|
||||
if v, err := c.GetValue(); err == nil {
|
||||
return v
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c ConfigKey) GetStringSlice() []string {
|
||||
if v, err := c.GetValue(); err == nil {
|
||||
return v.([]string)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type BlockType struct {
|
||||
Name string `json:"name" yaml:"name"`
|
||||
Type TypeName `json:"type" yaml:"type"`
|
||||
@ -135,9 +161,9 @@ func (b *Block) SetURI(uri string) (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
func (b *Block) SetParsedURI(uri *url.URL) (err error) {
|
||||
func (b *Block) SetParsedURI(uri data.URIParser) (err error) {
|
||||
if b.Values == nil {
|
||||
if err = b.NewConfigurationFromParsedURI(uri); err != nil {
|
||||
if err = b.NewConfigurationFromParsedURI(uri.URL()); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
@ -11,12 +11,10 @@ _ "errors"
|
||||
"gopkg.in/yaml.v3"
|
||||
"log/slog"
|
||||
_ "gitea.rosskeen.house/rosskeen.house/machine"
|
||||
//_ "gitea.rosskeen.house/pylon/luaruntime"
|
||||
"gitea.rosskeen.house/pylon/luaruntime"
|
||||
"decl/internal/codec"
|
||||
"decl/internal/data"
|
||||
"decl/internal/schema"
|
||||
"net/url"
|
||||
"runtime/debug"
|
||||
"errors"
|
||||
)
|
||||
|
||||
@ -33,6 +31,7 @@ type DeclarationType struct {
|
||||
OnError OnError `json:"onerror,omitempty" yaml:"onerror,omitempty"`
|
||||
Error string `json:"error,omitempty" yaml:"error,omitempty"`
|
||||
Requires Dependencies `json:"requires,omitempty" yaml:"requires,omitempty"`
|
||||
On *Events `json:"on,omitempty" yaml:"on,omitempty"`
|
||||
}
|
||||
|
||||
type Declaration struct {
|
||||
@ -43,14 +42,15 @@ type Declaration struct {
|
||||
OnError OnError `json:"onerror,omitempty" yaml:"onerror,omitempty"`
|
||||
Error string `json:"error,omitempty" yaml:"error,omitempty"`
|
||||
Requires Dependencies `json:"requires,omitempty" yaml:"requires,omitempty"`
|
||||
// runtime luaruntime.LuaRunner
|
||||
On *Events `json:"on,omitempty" yaml:"on,omitempty"`
|
||||
runtime luaruntime.LuaRunner
|
||||
document *Document
|
||||
configBlock data.Block
|
||||
ResourceTypes data.TypesRegistry[data.Resource] `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewDeclaration() *Declaration {
|
||||
return &Declaration{ ResourceTypes: DocumentRegistry.ResourceTypes }
|
||||
return &Declaration{ ResourceTypes: DocumentRegistry.ResourceTypes, runtime: luaruntime.New() }
|
||||
}
|
||||
|
||||
func NewDeclarationFromDocument(document *Document) *Declaration {
|
||||
@ -98,6 +98,7 @@ func (d *Declaration) Clone() data.Declaration {
|
||||
//runtime: luaruntime.New(),
|
||||
Config: d.Config,
|
||||
Requires: d.Requires,
|
||||
On: d.On,
|
||||
}
|
||||
}
|
||||
|
||||
@ -137,11 +138,12 @@ func (d *Declaration) Validate() (err error) {
|
||||
return err
|
||||
}
|
||||
|
||||
func (d *Declaration) NewResourceFromParsedURI(u *url.URL) (err error) {
|
||||
func (d *Declaration) NewResourceFromParsedURI(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
d.Attributes, err = d.ResourceTypes.NewFromParsedURI(&url.URL{ Scheme: string(d.Type) })
|
||||
d.Attributes, err = d.ResourceTypes.NewFromType(string(d.Type))
|
||||
} else {
|
||||
if d.Attributes, err = d.ResourceTypes.NewFromParsedURI(u); err == nil {
|
||||
parsed := u.URL()
|
||||
if d.Attributes, err = d.ResourceTypes.NewFromParsedURI(parsed); err == nil {
|
||||
err = d.Attributes.SetParsedURI(u)
|
||||
}
|
||||
}
|
||||
@ -149,15 +151,17 @@ func (d *Declaration) NewResourceFromParsedURI(u *url.URL) (err error) {
|
||||
}
|
||||
|
||||
func (d *Declaration) NewResource(uri *string) (err error) {
|
||||
slog.Info("Declaration.NewResource()")
|
||||
if d.ResourceTypes == nil {
|
||||
panic(fmt.Errorf("Undefined type registry: unable to create new resource %s", *uri))
|
||||
}
|
||||
|
||||
if uri == nil {
|
||||
d.Attributes, err = d.ResourceTypes.New(fmt.Sprintf("%s://", d.Type))
|
||||
d.Attributes, err = d.ResourceTypes.NewFromType(string(d.Type))
|
||||
} else {
|
||||
if d.Attributes, err = d.ResourceTypes.New(*uri); err == nil {
|
||||
err = d.Attributes.SetURI(*uri)
|
||||
}
|
||||
slog.Info("Declaration.NewResource()", "uri", *uri)
|
||||
parsedURI := URI(*uri).Parse()
|
||||
d.Attributes, err = d.ResourceTypes.NewFromParsedURI(parsedURI.URL())
|
||||
}
|
||||
return
|
||||
}
|
||||
@ -169,7 +173,6 @@ func (d *Declaration) Resource() data.Resource {
|
||||
func (d *Declaration) Apply(stateTransition string) (result error) {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
slog.Debug("Declaration.Apply()", "stacktrace", string(debug.Stack()))
|
||||
slog.Info("Declaration.Apply()", "error", r, "resourceerror", d.Error)
|
||||
if d.Error != "" {
|
||||
result = fmt.Errorf("%s - %s", r, d.Error)
|
||||
@ -245,7 +248,7 @@ func (d *Declaration) SetConfig(configDoc data.Document) {
|
||||
return
|
||||
}
|
||||
if d.Config != "" { // XXX
|
||||
panic("failed setting config")
|
||||
panic(fmt.Sprintf("failed setting config: %s", d.Config))
|
||||
}
|
||||
}
|
||||
|
||||
@ -254,7 +257,7 @@ func (d *Declaration) SetURI(uri string) (err error) {
|
||||
if d.Attributes == nil {
|
||||
err = d.NewResource(&uri)
|
||||
} else {
|
||||
err = d.Attributes.SetURI(uri)
|
||||
err = d.Attributes.SetParsedURI(URI(uri).Parse())
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
@ -268,7 +271,7 @@ func (d *Declaration) SetURI(uri string) (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
func (d *Declaration) SetParsedURI(uri *url.URL) (err error) {
|
||||
func (d *Declaration) SetParsedURI(uri data.URIParser) (err error) {
|
||||
slog.Info("Declaration.SetParsedURI()", "uri", uri, "declaration", d)
|
||||
if d.Attributes == nil {
|
||||
err = d.NewResourceFromParsedURI(uri)
|
||||
@ -298,7 +301,9 @@ func (d *Declaration) UnmarshalValue(value *DeclarationType) error {
|
||||
d.OnError = value.OnError
|
||||
d.Error = value.Error
|
||||
d.Requires = value.Requires
|
||||
newResource, resourceErr := d.ResourceTypes.New(fmt.Sprintf("%s://", value.Type))
|
||||
d.On = value.On
|
||||
newResource, resourceErr := d.ResourceTypes.NewFromType(string(value.Type))
|
||||
|
||||
slog.Info("Declaration.UnmarshalValue", "value", value, "error", resourceErr, "type", value.Type, "resource", newResource, "resourcetypes", d.ResourceTypes)
|
||||
if resourceErr != nil {
|
||||
slog.Info("Declaration.UnmarshalValue", "value", value, "error", resourceErr)
|
||||
@ -306,10 +311,30 @@ func (d *Declaration) UnmarshalValue(value *DeclarationType) error {
|
||||
}
|
||||
d.Attributes = newResource
|
||||
d.configBlock = d.Config.GetBlock()
|
||||
|
||||
if d.On != nil {
|
||||
if handler, ok := (*d.On)[EventTypeLoad]; ok {
|
||||
stackSize := d.runtime.Api().GetTop()
|
||||
if e := d.runtime.LoadScriptFromString(string(handler)); e != nil {
|
||||
d.Error = e.Error()
|
||||
}
|
||||
returnsCount := d.runtime.Api().GetTop() - stackSize
|
||||
if ! d.runtime.Api().IsNil(-1) {
|
||||
if returnsCount == 0 {
|
||||
// return nil
|
||||
} else {
|
||||
if lr,le := d.runtime.CopyReturnValuesFromCall(int(returnsCount)); le == nil {
|
||||
slog.Info("Event.Load", "result", lr, "error", le)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *Declaration) UnmarshalYAML(value *yaml.Node) error {
|
||||
func (d *Declaration) UnmarshalYAML(value *yaml.Node) (err error) {
|
||||
if d.ResourceTypes == nil {
|
||||
d.ResourceTypes = DocumentRegistry.ResourceTypes
|
||||
}
|
||||
@ -331,10 +356,17 @@ func (d *Declaration) UnmarshalYAML(value *yaml.Node) error {
|
||||
if unmarshalResourceErr := resourceAttrs.Attributes.Decode(d.Attributes); unmarshalResourceErr != nil {
|
||||
return unmarshalResourceErr
|
||||
}
|
||||
return nil
|
||||
|
||||
if i, ok := d.Attributes.(data.ResourceInitializer); ok {
|
||||
err = i.Init(nil)
|
||||
} else {
|
||||
err = fmt.Errorf("failed to execute init")
|
||||
}
|
||||
|
||||
func (d *Declaration) UnmarshalJSON(jsonData []byte) error {
|
||||
return
|
||||
}
|
||||
|
||||
func (d *Declaration) UnmarshalJSON(jsonData []byte) (err error) {
|
||||
if d.ResourceTypes == nil {
|
||||
d.ResourceTypes = DocumentRegistry.ResourceTypes
|
||||
}
|
||||
@ -354,24 +386,14 @@ func (d *Declaration) UnmarshalJSON(jsonData []byte) error {
|
||||
return unmarshalAttributesErr
|
||||
}
|
||||
|
||||
return nil
|
||||
if i, ok := d.Attributes.(data.ResourceInitializer); ok {
|
||||
err = i.Init(nil)
|
||||
} else {
|
||||
err = fmt.Errorf("failed to execute init")
|
||||
}
|
||||
|
||||
/*
|
||||
func (d *Declaration) MarshalJSON() ([]byte, error) {
|
||||
buf := new(bytes.Buffer)
|
||||
buf.WriteByte('"')
|
||||
buf.WriteString("value"))
|
||||
buf.WriteByte('"')
|
||||
return buf.Bytes(), nil
|
||||
return
|
||||
}
|
||||
*/
|
||||
|
||||
/*
|
||||
func (d *Declaration) MarshalYAML() (any, error) {
|
||||
return d, nil
|
||||
}
|
||||
*/
|
||||
|
||||
/*
|
||||
func (l *LuaWorker) Receive(m message.Envelope) {
|
||||
|
@ -57,6 +57,7 @@ func TestNewResourceDeclarationType(t *testing.T) {
|
||||
|
||||
e := resourceDeclaration.LoadString(decl, codec.FormatYaml)
|
||||
assert.Nil(t, e)
|
||||
|
||||
assert.Equal(t, TypeName("foo"), resourceDeclaration.Type)
|
||||
assert.NotNil(t, resourceDeclaration.Attributes)
|
||||
}
|
||||
|
@ -10,7 +10,6 @@ import (
|
||||
"io/fs"
|
||||
_ "os"
|
||||
"log/slog"
|
||||
"net/url"
|
||||
"github.com/sters/yaml-diff/yamldiff"
|
||||
"strings"
|
||||
"decl/internal/codec"
|
||||
@ -19,6 +18,7 @@ _ "decl/internal/types"
|
||||
"decl/internal/data"
|
||||
"decl/internal/schema"
|
||||
"context"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type DocumentType struct {
|
||||
@ -44,13 +44,23 @@ type Document struct {
|
||||
config data.Document
|
||||
Registry *Registry `json:"-" yaml:"-"`
|
||||
failedResources int `json:"-" yaml:"-"`
|
||||
importPaths *SearchPath `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewDocument(r *Registry) *Document {
|
||||
if r == nil {
|
||||
r = DocumentRegistry
|
||||
}
|
||||
return &Document{ Registry: r, Format: codec.FormatYaml, uris: mapper.New[string, data.Declaration](), configNames: mapper.New[string, data.Block]() }
|
||||
|
||||
var configImportPath ConfigKey = "system.importpath"
|
||||
|
||||
return &Document{
|
||||
Registry: r,
|
||||
Format: codec.FormatYaml,
|
||||
uris: mapper.New[string, data.Declaration](),
|
||||
configNames: mapper.New[string, data.Block](),
|
||||
importPaths: NewSearchPath(configImportPath.GetStringSlice()),
|
||||
}
|
||||
}
|
||||
|
||||
func (d *Document) GetURI() string {
|
||||
@ -59,6 +69,19 @@ func (d *Document) GetURI() string {
|
||||
|
||||
func (d *Document) SetURI(uri string) {
|
||||
d.URI = URI(uri)
|
||||
d.AddProjectPath()
|
||||
}
|
||||
|
||||
func (d *Document) AddProjectPath() {
|
||||
exists := d.URI.Exists()
|
||||
if exists {
|
||||
if u := d.URI.Parse().(*ParsedURI); u != nil {
|
||||
projectPath := filepath.Dir(filepath.Join(u.Hostname(), u.Path))
|
||||
if err := d.importPaths.AddPath(projectPath); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (d *Document) Types() data.TypesRegistry[data.Resource] {
|
||||
@ -99,6 +122,10 @@ func (d *Document) Set(key string, value any) {
|
||||
d.uris.Set(key, value.(data.Declaration))
|
||||
}
|
||||
|
||||
func (d *Document) Delete(key string) {
|
||||
d.uris.Delete(key)
|
||||
}
|
||||
|
||||
func (d *Document) GetResource(uri string) *Declaration {
|
||||
if decl, ok := d.uris[uri]; ok {
|
||||
return decl.(*Declaration)
|
||||
@ -142,7 +169,15 @@ func (d *Document) ImportedDocuments() (documents []data.Document) {
|
||||
func (d *Document) loadImports() (err error) {
|
||||
for _, uri := range d.Imports {
|
||||
if ! DocumentRegistry.HasDocument(uri) {
|
||||
if _, err = DocumentRegistry.Load(uri); err != nil {
|
||||
var load URI = uri
|
||||
if ! load.Exists() {
|
||||
foundURI := d.importPaths.FindURI(load)
|
||||
if foundURI != "" {
|
||||
load = foundURI
|
||||
}
|
||||
}
|
||||
slog.Info("Document.loadImports()", "load", load, "uri", uri, "importpaths", d.importPaths, "doc", d.URI)
|
||||
if _, err = DocumentRegistry.Load(load); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
@ -323,6 +358,10 @@ func (d *Document) MapResourceURI(uri string, declaration data.Declaration) {
|
||||
d.uris[uri] = declaration
|
||||
}
|
||||
|
||||
func (d *Document) UnMapResourceURI(uri string) {
|
||||
d.uris.Delete(uri)
|
||||
}
|
||||
|
||||
func (d *Document) AddDeclaration(declaration data.Declaration) {
|
||||
uri := declaration.URI()
|
||||
decl := declaration.(*Declaration)
|
||||
@ -382,7 +421,7 @@ func (d *Document) NewResourceFromURI(uri URI) (newResource data.Resource, err e
|
||||
return d.NewResourceFromParsedURI(uri.Parse())
|
||||
}
|
||||
|
||||
func (d *Document) NewResourceFromParsedURI(uri *url.URL) (newResource data.Resource, err error) {
|
||||
func (d *Document) NewResourceFromParsedURI(uri data.URIParser) (newResource data.Resource, err error) {
|
||||
if uri == nil {
|
||||
return nil, fmt.Errorf("%w: %s", ErrUnknownResourceType, uri)
|
||||
}
|
||||
|
@ -84,6 +84,9 @@ resources:
|
||||
|
||||
f.Name = "mytestresource"
|
||||
f.Size = 3
|
||||
|
||||
assert.Nil(t, f.Init(nil))
|
||||
|
||||
d.AddResourceDeclaration("foo", f)
|
||||
|
||||
ey := d.Generate(&documentYaml)
|
||||
|
31
internal/folio/events.go
Normal file
31
internal/folio/events.go
Normal file
@ -0,0 +1,31 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"errors"
|
||||
_ "gitea.rosskeen.house/pylon/luaruntime"
|
||||
_ "fmt"
|
||||
)
|
||||
|
||||
type EventHandler string
|
||||
|
||||
type Events map[EventType]EventHandler
|
||||
|
||||
var (
|
||||
ErrInvalidHandler error = errors.New("Invalid event handler")
|
||||
)
|
||||
|
||||
func NewEvents() *Events {
|
||||
e := make(Events)
|
||||
return &e
|
||||
}
|
||||
|
||||
func (e *Events) Set(t EventType, h EventHandler) (err error) {
|
||||
if err = t.Validate(); err == nil {
|
||||
(*e)[t] = h
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
|
22
internal/folio/events_test.go
Normal file
22
internal/folio/events_test.go
Normal file
@ -0,0 +1,22 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
|
||||
func TestNewEvent(t *testing.T) {
|
||||
var et EventType
|
||||
events := NewEvents()
|
||||
assert.NotNil(t, events)
|
||||
|
||||
assert.Nil(t, et.Set(EventTypeLoad))
|
||||
assert.Nil(t, events.Set(et, EventHandler(`
|
||||
print('hello world')
|
||||
`)))
|
||||
|
||||
}
|
39
internal/folio/eventtype.go
Normal file
39
internal/folio/eventtype.go
Normal file
@ -0,0 +1,39 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type EventType string
|
||||
|
||||
const (
|
||||
EventTypeLoad EventType = "load"
|
||||
EventTypeCreate EventType = "create"
|
||||
EventTypeRead EventType = "read"
|
||||
EventTypeUpdate EventType = "update"
|
||||
EventTypeDelete EventType = "delete"
|
||||
EventTypeError EventType = "error"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrUnknownEventType error = errors.New("Unknown EventType")
|
||||
)
|
||||
|
||||
func (e EventType) Validate() (err error) {
|
||||
switch e {
|
||||
case EventTypeLoad, EventTypeCreate, EventTypeRead, EventTypeUpdate, EventTypeDelete, EventTypeError:
|
||||
default:
|
||||
return fmt.Errorf("%w: %s", ErrUnknownEventType, e)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (e *EventType) Set(v EventType) (err error) {
|
||||
if err = v.Validate(); err == nil {
|
||||
(*e) = v
|
||||
}
|
||||
return
|
||||
}
|
29
internal/folio/eventtype_test.go
Normal file
29
internal/folio/eventtype_test.go
Normal file
@ -0,0 +1,29 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
|
||||
func TestEventType(t *testing.T) {
|
||||
for _, v := range []struct{ et EventType; expected error } {
|
||||
{ et: EventType("load"), expected: nil },
|
||||
{ et: EventType("create"), expected: nil },
|
||||
{ et: EventType("read"), expected: nil },
|
||||
{ et: EventType("update"), expected: nil },
|
||||
{ et: EventType("delete"), expected: nil },
|
||||
{ et: EventType("error"), expected: nil },
|
||||
{ et: EventType("foo"), expected: ErrUnknownEventType },
|
||||
} {
|
||||
assert.ErrorIs(t, v.et.Validate(), v.expected)
|
||||
}
|
||||
}
|
||||
|
||||
func TestEventTypeSet(t *testing.T) {
|
||||
var et EventType
|
||||
assert.Nil(t, et.Set(EventTypeLoad))
|
||||
}
|
@ -10,7 +10,6 @@ _ "fmt"
|
||||
"decl/internal/data"
|
||||
"decl/internal/codec"
|
||||
"io"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
type MockConfiguration struct {
|
||||
@ -38,7 +37,7 @@ func (m *MockConfiguration) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MockConfiguration) SetParsedURI(uri *url.URL) error {
|
||||
func (m *MockConfiguration) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -17,22 +17,30 @@ _ "gopkg.in/yaml.v3"
|
||||
func RegisterMocks() {
|
||||
TestResourceTypes.Register([]string{"foo"}, func(u *url.URL) data.Resource {
|
||||
f := NewFooResource()
|
||||
if u != nil {
|
||||
f.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
}
|
||||
return f
|
||||
})
|
||||
TestResourceTypes.Register([]string{"bar"}, func(u *url.URL) data.Resource {
|
||||
f := NewBarResource()
|
||||
if u != nil {
|
||||
f.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
}
|
||||
return f
|
||||
})
|
||||
TestResourceTypes.Register([]string{"testuser"}, func(u *url.URL) data.Resource {
|
||||
f := NewTestuserResource()
|
||||
if u != nil {
|
||||
f.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
}
|
||||
return f
|
||||
})
|
||||
TestResourceTypes.Register([]string{"file"}, func(u *url.URL) data.Resource {
|
||||
f := NewFileResource()
|
||||
if u != nil {
|
||||
f.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
}
|
||||
return f
|
||||
})
|
||||
}
|
||||
@ -40,23 +48,20 @@ func RegisterMocks() {
|
||||
|
||||
type MockFoo struct {
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
*MockResource `json:"-" yaml:"-"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
*MockResource `json:",inline" yaml:",inline"`
|
||||
Size int `json:"size" yaml:"size"`
|
||||
}
|
||||
|
||||
type MockBar struct {
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
*MockResource `json:"-" yaml:"-"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
*MockResource `json:",inline" yaml:",inline"`
|
||||
Size int `json:"size,omitempty" yaml:"size,omitempty"`
|
||||
Owner string `json:"owner,omitempty" yaml:"owner,omitempty"`
|
||||
}
|
||||
|
||||
type MockTestuser struct {
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
*MockResource `json:"-" yaml:"-"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
*MockResource `json:",inline" yaml:",inline"`
|
||||
Uid string `json:"uid" yaml:"uid"`
|
||||
Group string `json:"group" yaml:"group"`
|
||||
Home string `json:"home" yaml:"home"`
|
||||
@ -64,13 +69,12 @@ type MockTestuser struct {
|
||||
|
||||
type MockFile struct {
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
*MockResource `json:"-" yaml:"-"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
*MockResource `json:",inline" yaml:",inline"`
|
||||
Size int `json:"size" yaml:"size"`
|
||||
}
|
||||
|
||||
func NewMockResource(typename string, stater machine.Stater) *MockResource {
|
||||
return &MockResource {
|
||||
func NewMockResource(typename string, stater machine.Stater) (m *MockResource) {
|
||||
m = &MockResource {
|
||||
InjectType: func() string { return typename },
|
||||
InjectResolveId: func(ctx context.Context) string { return "bar" },
|
||||
InjectLoadDecl: func(string) error { return nil },
|
||||
@ -89,6 +93,13 @@ func NewMockResource(typename string, stater machine.Stater) *MockResource {
|
||||
InjectURI: func() string { return fmt.Sprintf("%s://bar", typename) },
|
||||
InjectNotify: func(*machine.EventMessage) {},
|
||||
}
|
||||
m.InjectInit = func(u data.URIParser) error {
|
||||
if u != nil {
|
||||
m.Name = filepath.Join(u.URL().Hostname(), u.URL().Path)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
func NewFooResource() *MockFoo {
|
||||
|
@ -11,10 +11,11 @@ _ "fmt"
|
||||
"decl/internal/data"
|
||||
"decl/internal/codec"
|
||||
"io"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
type MockResource struct {
|
||||
Name string `json:"name" yaml:"name"`
|
||||
InjectInit func(data.URIParser) error `json:"-" yaml:"-"`
|
||||
InjectURI func() string `json:"-" yaml:"-"`
|
||||
InjectType func() string `json:"-" yaml:"-"`
|
||||
InjectResolveId func(ctx context.Context) string `json:"-" yaml:"-"`
|
||||
@ -54,7 +55,7 @@ func (m *MockResource) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MockResource) SetParsedURI(uri *url.URL) error {
|
||||
func (m *MockResource) SetParsedURI(uri data.URIParser) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
@ -126,6 +127,10 @@ func (m *MockResource) Apply() error {
|
||||
return m.InjectApply()
|
||||
}
|
||||
|
||||
func (m *MockResource) Init(u data.URIParser) (error) {
|
||||
return m.InjectInit(u)
|
||||
}
|
||||
|
||||
func (m *MockResource) Type() string {
|
||||
return m.InjectType()
|
||||
}
|
||||
|
102
internal/folio/parseduri.go
Normal file
102
internal/folio/parseduri.go
Normal file
@ -0,0 +1,102 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"decl/internal/transport"
|
||||
"decl/internal/data"
|
||||
"decl/internal/identifier"
|
||||
"strings"
|
||||
)
|
||||
|
||||
|
||||
var (
|
||||
)
|
||||
|
||||
type ParsedURI url.URL
|
||||
|
||||
func NewParsedURI(uri URI) *ParsedURI {
|
||||
return uri.Parse().(*ParsedURI)
|
||||
}
|
||||
|
||||
func CastParsedURI(u *url.URL) *ParsedURI {
|
||||
return (*ParsedURI)(u)
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) URL() *url.URL {
|
||||
return (*url.URL)(uri)
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) Hostname() string {
|
||||
return (*url.URL)(uri).Hostname()
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) NewResource(document data.Document) (newResource data.Resource, err error) {
|
||||
if document == nil {
|
||||
declaration := NewDeclaration()
|
||||
if err = declaration.NewResourceFromParsedURI(uri); err == nil {
|
||||
return declaration.Attributes, err
|
||||
}
|
||||
} else {
|
||||
newResource, err = document.NewResourceFromParsedURI(uri)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) ConstructResource(res data.Resource) (err error) {
|
||||
if ri, ok := res.(data.ResourceInitializer); ok {
|
||||
err = ri.Init(uri)
|
||||
} else {
|
||||
err = res.SetParsedURI(uri)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) Converter() (converter data.Converter, err error) {
|
||||
return DocumentRegistry.ConverterTypes.NewFromParsedURI(uri.URL())
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) Exists() bool {
|
||||
return transport.Exists(uri.URL())
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) ContentReaderStream() (*transport.Reader, error) {
|
||||
return transport.NewReader(uri.URL())
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) ContentWriterStream() (*transport.Writer, error) {
|
||||
return transport.NewWriter(uri.URL())
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) String() string {
|
||||
return uri.URL().String()
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) SetURL(url *url.URL) {
|
||||
*uri = *(*ParsedURI)(url)
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) Extension() (string, string) {
|
||||
return (identifier.ID)(uri.Path).Extension()
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) ContentType() string {
|
||||
var ext strings.Builder
|
||||
exttype, fileext := uri.Extension()
|
||||
if fileext == "" {
|
||||
return exttype
|
||||
}
|
||||
ext.WriteString(exttype)
|
||||
ext.WriteRune('.')
|
||||
ext.WriteString(fileext)
|
||||
return ext.String()
|
||||
}
|
||||
|
||||
func (uri *ParsedURI) IsEmpty() bool {
|
||||
if uri == nil || (identifier.ID)(uri.String()).IsEmpty() {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
@ -69,7 +69,7 @@ func (r *Registry) SetDocument(key URI, value *Document) {
|
||||
|
||||
func (r *Registry) NewDocument(uri URI) (doc *Document) {
|
||||
doc = NewDocument(r)
|
||||
doc.URI = uri
|
||||
doc.SetURI(string(uri))
|
||||
r.Documents = append(r.Documents, doc)
|
||||
if uri != "" {
|
||||
r.UriMap[uri] = doc
|
||||
@ -88,17 +88,17 @@ func (r *Registry) AppendParsedURI(uri *url.URL, documents []data.Document) (add
|
||||
case data.ManyExtractor:
|
||||
var docs []data.Document
|
||||
docs, err = extractor.ExtractMany(sourceResource, nil)
|
||||
slog.Info("folio.Registry.Append() - ExtractMany", "uri", uri, "source", sourceResource, "docs", docs, "error", err)
|
||||
slog.Info("folio.Registry.AppendParsedURI() - ExtractMany", "uri", uri, "source", sourceResource, "docs", docs, "error", err)
|
||||
documents = append(documents, docs...)
|
||||
case data.Extractor:
|
||||
var singleDocument data.Document
|
||||
singleDocument, err = extractor.Extract(sourceResource, nil)
|
||||
slog.Info("folio.Registry.Append() - Extract", "uri", uri, "source", sourceResource, "doc", singleDocument, "error", err)
|
||||
slog.Info("folio.Registry.AppendParsedURI() - Extract", "uri", uri, "source", sourceResource, "doc", singleDocument, "error", err)
|
||||
documents = append(documents, singleDocument)
|
||||
}
|
||||
}
|
||||
}
|
||||
slog.Info("folio.Registry.Append()", "uri", uri, "converter", r.ConverterTypes, "error", err)
|
||||
slog.Info("folio.Registry.AppendParsedURI()", "uri", uri, "converter", r.ConverterTypes, "error", err)
|
||||
addedDocuments = documents
|
||||
return
|
||||
}
|
||||
@ -130,6 +130,8 @@ func (r *Registry) Append(uri URI, documents []data.Document) (addedDocuments []
|
||||
slog.Info("folio.Registry.Append() - Extract", "uri", uri, "source", sourceResource, "doc", singleDocument, "error", err)
|
||||
documents = append(documents, singleDocument)
|
||||
}
|
||||
} else {
|
||||
slog.Warn("folio.Registry.Append(): failed loading extractor as resource")
|
||||
}
|
||||
}
|
||||
slog.Info("folio.Registry.Append()", "uri", uri, "converter", r.ConverterTypes, "error", err)
|
||||
@ -146,3 +148,8 @@ func (r *Registry) LoadFromURL(uri *url.URL) (documents []data.Document, err err
|
||||
documents = make([]data.Document, 0, 10)
|
||||
return r.AppendParsedURI(uri, documents)
|
||||
}
|
||||
|
||||
func (r *Registry) LoadFromParsedURI(uri *url.URL) (documents []data.Document, err error) {
|
||||
documents = make([]data.Document, 0, 10)
|
||||
return r.AppendParsedURI(uri, documents)
|
||||
}
|
||||
|
@ -15,11 +15,11 @@ var (
|
||||
func NewResourceFromParsedURI(u *url.URL, document data.Document) (newResource data.Resource, err error) {
|
||||
if document == nil {
|
||||
declaration := NewDeclaration()
|
||||
if err = declaration.NewResourceFromParsedURI(u); err == nil {
|
||||
if err = declaration.NewResourceFromParsedURI((*ParsedURI)(u)); err == nil {
|
||||
return declaration.Attributes, err
|
||||
}
|
||||
} else {
|
||||
newResource, err = document.NewResourceFromParsedURI(u)
|
||||
newResource, err = document.NewResourceFromParsedURI((*ParsedURI)(u))
|
||||
return
|
||||
}
|
||||
return
|
||||
|
@ -56,7 +56,7 @@ func (r ResourceReference) Dereference(look data.ResourceMapper) data.Resource {
|
||||
}
|
||||
|
||||
func (r ResourceReference) Parse() *url.URL {
|
||||
return URI(r).Parse()
|
||||
return URI(r).Parse().URL()
|
||||
}
|
||||
|
||||
func (r ResourceReference) Exists() bool {
|
||||
|
65
internal/folio/searchpath.go
Normal file
65
internal/folio/searchpath.go
Normal file
@ -0,0 +1,65 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"decl/internal/ext"
|
||||
"decl/internal/ds"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"path/filepath"
|
||||
"decl/internal/data"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrSearchPathNotExist error = errors.New("Search path does not exist.")
|
||||
)
|
||||
|
||||
// A search path should find a file in list of paths
|
||||
|
||||
type SearchPath struct {
|
||||
paths *ds.OrderedSet[string]
|
||||
}
|
||||
|
||||
func NewSearchPath(paths []string) (s *SearchPath) {
|
||||
s = &SearchPath{ paths: ds.NewOrderedSet[string]() }
|
||||
s.paths.AddItems(paths)
|
||||
return
|
||||
}
|
||||
|
||||
func (s *SearchPath) AddPath(path string) error {
|
||||
if newPath := ext.FilePath(path).Abs(); newPath.Exists() {
|
||||
s.paths.Add(string(newPath))
|
||||
} else {
|
||||
return fmt.Errorf("%w: %s", ErrSearchPathNotExist, path)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *SearchPath) Find(relative string) string {
|
||||
pathValues := s.paths.Items()
|
||||
for i := len(pathValues) - 1; i >= 0; i-- {
|
||||
p := *pathValues[i]
|
||||
searchPath := ext.FilePath(p)
|
||||
searchPath.Add(relative)
|
||||
slog.Info("SearchPath.Find()", "searchpath", p, "file", relative, "target", searchPath)
|
||||
if searchPath.Exists() {
|
||||
return string(searchPath)
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (s *SearchPath) FindParsedURI(uri data.URIParser) (result URI) {
|
||||
u := uri.URL()
|
||||
filePath := filepath.Join(u.Hostname(), u.Path)
|
||||
if absPath := s.Find(filePath); absPath != "" {
|
||||
result = URI(fmt.Sprintf("%s://%s", u.Scheme, string(absPath)))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *SearchPath) FindURI(uri URI) (result URI) {
|
||||
return s.FindParsedURI(uri.Parse())
|
||||
}
|
24
internal/folio/searchpath_test.go
Normal file
24
internal/folio/searchpath_test.go
Normal file
@ -0,0 +1,24 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package folio
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
"os"
|
||||
)
|
||||
|
||||
func TestSearchPath(t *testing.T) {
|
||||
|
||||
assert.Nil(t, TempDir.CreateFile("test.jx.yaml", ""))
|
||||
|
||||
sp := NewSearchPath(nil)
|
||||
assert.NotNil(t, sp)
|
||||
|
||||
assert.Nil(t, sp.AddPath(string(TempDir)))
|
||||
|
||||
absPath := sp.Find("test.jx.yaml")
|
||||
_, err := os.Stat(absPath)
|
||||
assert.True(t, !os.IsNotExist(err))
|
||||
|
||||
}
|
@ -35,14 +35,24 @@ func (u URI) NewResource(document data.Document) (newResource data.Resource, err
|
||||
return
|
||||
}
|
||||
|
||||
func (u URI) ConstructResource(res data.Resource) (err error) {
|
||||
parsedURI := u.Parse()
|
||||
if ri, ok := res.(data.ResourceInitializer); ok {
|
||||
err = ri.Init(parsedURI)
|
||||
} else {
|
||||
err = res.SetParsedURI(parsedURI)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u URI) Converter() (converter data.Converter, err error) {
|
||||
return DocumentRegistry.ConverterTypes.New(string(u))
|
||||
}
|
||||
|
||||
func (u URI) Parse() *url.URL {
|
||||
func (u URI) Parse() data.URIParser {
|
||||
url, e := url.Parse(string(u))
|
||||
if e == nil {
|
||||
return url
|
||||
return (*ParsedURI)(url)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -10,17 +10,17 @@ import (
|
||||
|
||||
func TestURI(t *testing.T) {
|
||||
var file URI = URI(fmt.Sprintf("file://%s", TempDir))
|
||||
u := file.Parse()
|
||||
u := file.Parse().URL()
|
||||
assert.Equal(t, "file", u.Scheme)
|
||||
assert.True(t, file.Exists())
|
||||
file = URI(fmt.Sprintf("0file:_/%s", TempDir))
|
||||
u = file.Parse()
|
||||
assert.Nil(t, u)
|
||||
x := file.Parse()
|
||||
assert.Nil(t, x)
|
||||
}
|
||||
|
||||
func TestURISetURL(t *testing.T) {
|
||||
var file URI = URI(fmt.Sprintf("file://%s", TempDir))
|
||||
u := file.Parse()
|
||||
u := file.Parse().URL()
|
||||
var fileFromURL URI
|
||||
fileFromURL.SetURL(u)
|
||||
assert.Equal(t, fileFromURL, file)
|
||||
|
@ -18,10 +18,15 @@ func (m Store[Key, Value]) Set(key Key, value Value) {
|
||||
m[key] = value
|
||||
}
|
||||
|
||||
func (m Store[Key, Value]) Delete(key Key) {
|
||||
delete(m, key)
|
||||
}
|
||||
|
||||
type Mapper interface {
|
||||
Get(key string) (any, bool)
|
||||
Has(key string) (bool)
|
||||
Set(key string, value any)
|
||||
Delete(key string)
|
||||
}
|
||||
|
||||
type Getter[Key comparable, Value comparable] interface {
|
||||
@ -33,6 +38,7 @@ type Map[Key comparable, Value comparable] interface {
|
||||
Get(key Key) (Value, bool)
|
||||
Has(key Key) (bool)
|
||||
Set(key Key, value Value)
|
||||
Delete(key Key)
|
||||
}
|
||||
|
||||
func New[Key comparable, Value comparable]() Store[Key, Value] {
|
||||
|
@ -1,129 +0,0 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package resource
|
||||
|
||||
import (
|
||||
_ "context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"gopkg.in/yaml.v3"
|
||||
"io"
|
||||
"log/slog"
|
||||
_ "net/url"
|
||||
_ "os"
|
||||
"os/exec"
|
||||
"strings"
|
||||
"text/template"
|
||||
"decl/internal/codec"
|
||||
)
|
||||
|
||||
type CommandExecutor func(value any) ([]byte, error)
|
||||
type CommandExtractAttributes func(output []byte, target any) error
|
||||
|
||||
type CommandArg string
|
||||
|
||||
type Command struct {
|
||||
Path string `json:"path" yaml:"path"`
|
||||
Args []CommandArg `json:"args" yaml:"args"`
|
||||
Split bool `json:"split" yaml:"split"`
|
||||
Executor CommandExecutor `json:"-" yaml:"-"`
|
||||
Extractor CommandExtractAttributes `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewCommand() *Command {
|
||||
c := &Command{ Split: true }
|
||||
c.Executor = func(value any) ([]byte, error) {
|
||||
args, err := c.Template(value)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
cmd := exec.Command(c.Path, args...)
|
||||
|
||||
slog.Info("execute() - cmd", "path", c.Path, "args", args)
|
||||
output, stdoutPipeErr := cmd.StdoutPipe()
|
||||
if stdoutPipeErr != nil {
|
||||
return nil, stdoutPipeErr
|
||||
}
|
||||
|
||||
stderr, pipeErr := cmd.StderrPipe()
|
||||
if pipeErr != nil {
|
||||
return nil, pipeErr
|
||||
}
|
||||
|
||||
if startErr := cmd.Start(); startErr != nil {
|
||||
return nil, startErr
|
||||
}
|
||||
|
||||
slog.Info("execute() - start", "cmd", cmd)
|
||||
stdOutOutput, _ := io.ReadAll(output)
|
||||
stdErrOutput, _ := io.ReadAll(stderr)
|
||||
|
||||
slog.Info("execute() - io", "stdout", string(stdOutOutput), "stderr", string(stdErrOutput))
|
||||
waitErr := cmd.Wait()
|
||||
|
||||
slog.Info("execute()", "path", c.Path, "args", args, "output", string(stdOutOutput), "error", string(stdErrOutput))
|
||||
|
||||
if len(stdErrOutput) > 0 {
|
||||
return stdOutOutput, fmt.Errorf(string(stdErrOutput))
|
||||
}
|
||||
return stdOutOutput, waitErr
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *Command) Load(r io.Reader) error {
|
||||
return codec.NewYAMLDecoder(r).Decode(c)
|
||||
}
|
||||
|
||||
func (c *Command) LoadDecl(yamlResourceDeclaration string) error {
|
||||
return codec.NewYAMLStringDecoder(yamlResourceDeclaration).Decode(c)
|
||||
}
|
||||
|
||||
func (c *Command) Template(value any) ([]string, error) {
|
||||
var args []string = make([]string, 0, len(c.Args) * 2)
|
||||
for i, arg := range c.Args {
|
||||
var commandLineArg strings.Builder
|
||||
err := template.Must(template.New(fmt.Sprintf("arg%d", i)).Parse(string(arg))).Execute(&commandLineArg, value)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if commandLineArg.Len() > 0 {
|
||||
var splitArg []string
|
||||
if c.Split {
|
||||
splitArg = strings.Split(commandLineArg.String(), " ")
|
||||
} else {
|
||||
splitArg = []string{commandLineArg.String()}
|
||||
}
|
||||
slog.Info("Template()", "split", splitArg, "len", len(splitArg))
|
||||
args = append(args, splitArg...)
|
||||
}
|
||||
}
|
||||
|
||||
slog.Info("Template()", "Args", c.Args, "lencargs", len(c.Args), "args", args, "lenargs", len(args), "value", value)
|
||||
return args, nil
|
||||
}
|
||||
|
||||
func (c *Command) Execute(value any) ([]byte, error) {
|
||||
return c.Executor(value)
|
||||
}
|
||||
|
||||
func (c *CommandArg) UnmarshalValue(value string) error {
|
||||
*c = CommandArg(value)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *CommandArg) UnmarshalJSON(data []byte) error {
|
||||
var s string
|
||||
if unmarshalRouteTypeErr := json.Unmarshal(data, &s); unmarshalRouteTypeErr != nil {
|
||||
return unmarshalRouteTypeErr
|
||||
}
|
||||
return c.UnmarshalValue(s)
|
||||
}
|
||||
|
||||
func (c *CommandArg) UnmarshalYAML(value *yaml.Node) error {
|
||||
var s string
|
||||
if err := value.Decode(&s); err != nil {
|
||||
return err
|
||||
}
|
||||
return c.UnmarshalValue(s)
|
||||
}
|
@ -1,58 +0,0 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
|
||||
package resource
|
||||
|
||||
import (
|
||||
_ "fmt"
|
||||
"github.com/stretchr/testify/assert"
|
||||
_ "os"
|
||||
_ "strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestNewCommand(t *testing.T) {
|
||||
c := NewCommand()
|
||||
assert.NotNil(t, c)
|
||||
}
|
||||
|
||||
func TestCommandLoad(t *testing.T) {
|
||||
c := NewCommand()
|
||||
assert.NotNil(t, c)
|
||||
|
||||
decl := `
|
||||
path: find
|
||||
args:
|
||||
- "{{ .Path }}"
|
||||
`
|
||||
|
||||
assert.Nil(t, c.LoadDecl(decl))
|
||||
assert.Equal(t, "find", c.Path)
|
||||
}
|
||||
|
||||
func TestCommandTemplate(t *testing.T) {
|
||||
c := NewCommand()
|
||||
assert.NotNil(t, c)
|
||||
|
||||
decl := `
|
||||
path: find
|
||||
args:
|
||||
- "{{ .Path }}"
|
||||
`
|
||||
|
||||
assert.Nil(t, c.LoadDecl(decl))
|
||||
assert.Equal(t, "find", c.Path)
|
||||
assert.Equal(t, 1, len(c.Args))
|
||||
|
||||
f := NewFile()
|
||||
f.Path = "./"
|
||||
args, templateErr := c.Template(f)
|
||||
assert.Nil(t, templateErr)
|
||||
assert.Equal(t, 1, len(args))
|
||||
|
||||
assert.Equal(t, "./", string(args[0]))
|
||||
|
||||
out, err := c.Execute(f)
|
||||
assert.Nil(t, err)
|
||||
assert.Greater(t, len(out), 0)
|
||||
}
|
@ -13,14 +13,16 @@ import (
|
||||
)
|
||||
|
||||
type UriSchemeValidator func(scheme string) bool
|
||||
type UriNormalize func() error
|
||||
|
||||
type Common struct {
|
||||
SchemeCheck UriSchemeValidator `json:"-" yaml:"-"`
|
||||
NormalizePath UriNormalize `json:"-" yaml:"-"`
|
||||
includeQueryParamsInURI bool `json:"-" yaml:"-"`
|
||||
resourceType TypeName `json:"-" yaml:"-"`
|
||||
Uri folio.URI `json:"uri,omitempty" yaml:"uri,omitempty"`
|
||||
parsedURI *url.URL `json:"-" yaml:"-"`
|
||||
Path string `json:"path,omitempty" yaml:"path,omitempty"`
|
||||
absPath string `json:"-" yaml:"-"`
|
||||
|
||||
exttype string `json:"-" yaml:"-"`
|
||||
fileext string `json:"-" yaml:"-"`
|
||||
@ -34,6 +36,7 @@ type Common struct {
|
||||
func NewCommon(resourceType TypeName, includeQueryParams bool) *Common {
|
||||
c := &Common{ includeQueryParamsInURI: includeQueryParams, resourceType: resourceType }
|
||||
c.SchemeCheck = c.IsValidResourceScheme
|
||||
c.NormalizePath = c.NormalizeFilePath
|
||||
return c
|
||||
}
|
||||
|
||||
@ -54,9 +57,13 @@ func (c *Common) SetResourceMapper(resources data.ResourceMapper) {
|
||||
|
||||
func (c *Common) Clone() *Common {
|
||||
return &Common {
|
||||
Uri: c.Uri,
|
||||
SchemeCheck: c.SchemeCheck,
|
||||
NormalizePath: c.NormalizePath,
|
||||
includeQueryParamsInURI: c.includeQueryParamsInURI,
|
||||
resourceType: c.resourceType,
|
||||
parsedURI: c.parsedURI,
|
||||
Path: c.Path,
|
||||
absPath: c.absPath,
|
||||
exttype: c.exttype,
|
||||
fileext: c.fileext,
|
||||
normalizePath: c.normalizePath,
|
||||
@ -74,41 +81,35 @@ func (c *Common) URIPath() string {
|
||||
return c.Path
|
||||
}
|
||||
|
||||
func (c *Common) URI() string {
|
||||
return string(c.Uri)
|
||||
func (c *Common) URI() folio.URI {
|
||||
slog.Info("Common.URI", "parsed", c.parsedURI)
|
||||
return folio.URI(c.parsedURI.String())
|
||||
}
|
||||
|
||||
func (c *Common) SetURI(uri string) (err error) {
|
||||
c.SetURIFromString(uri)
|
||||
err = c.SetParsedURI(c.Uri.Parse())
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Common) SetURIFromString(uri string) {
|
||||
c.Uri = folio.URI(uri)
|
||||
c.exttype, c.fileext = c.Uri.Extension()
|
||||
}
|
||||
|
||||
func (c *Common) SetParsedURI(u *url.URL) (err error) {
|
||||
func (c *Common) SetParsedURI(u data.URIParser) (err error) {
|
||||
if u != nil {
|
||||
slog.Info("Common.SetParsedURI()", "parsed", u, "uri", c.Uri)
|
||||
if c.Uri.IsEmpty() {
|
||||
c.SetURIFromString(u.String())
|
||||
}
|
||||
c.parsedURI = u
|
||||
|
||||
slog.Info("Common.SetParsedURI()", "parsed", u, "uri", c)
|
||||
|
||||
c.parsedURI = u.URL()
|
||||
|
||||
c.exttype, c.fileext = u.Extension()
|
||||
if c.SchemeCheck(c.parsedURI.Scheme) {
|
||||
if c.includeQueryParamsInURI {
|
||||
c.Path = filepath.Join(c.parsedURI.Hostname(), c.parsedURI.RequestURI())
|
||||
} else {
|
||||
c.Path = filepath.Join(c.parsedURI.Hostname(), c.parsedURI.Path)
|
||||
}
|
||||
if c.absPath, err = filepath.Abs(c.Path); err != nil {
|
||||
return
|
||||
}
|
||||
if err = c.NormalizePath(); err != nil {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
err = fmt.Errorf("%w: %s is not a %s resource, parsed: %t", ErrInvalidResourceURI, c.Uri, c.Type(), (u != nil))
|
||||
err = fmt.Errorf("%w: %s is not a %s resource, parsed: %t", ErrInvalidResourceURI, c.URI(), c.Type(), (u != nil))
|
||||
return
|
||||
}
|
||||
|
||||
@ -117,26 +118,27 @@ func (c *Common) UseConfig(config data.ConfigurationValueGetter) {
|
||||
}
|
||||
|
||||
func (c *Common) ResolveId(ctx context.Context) string {
|
||||
if e := c.NormalizePath(); e != nil {
|
||||
panic(e)
|
||||
var err error
|
||||
if c.absPath, err = filepath.Abs(c.Path); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if err = c.NormalizePath(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return c.Path
|
||||
}
|
||||
|
||||
func (c *Common) NormalizePath() error {
|
||||
// Common path normalization for a file resource.
|
||||
func (c *Common) NormalizeFilePath() (err error) {
|
||||
if c.config != nil {
|
||||
if prefixPath, configErr := c.config.GetValue("prefix"); configErr == nil {
|
||||
c.Path = filepath.Join(prefixPath.(string), c.Path)
|
||||
}
|
||||
}
|
||||
if c.normalizePath {
|
||||
filePath, fileAbsErr := filepath.Abs(c.Path)
|
||||
if fileAbsErr == nil {
|
||||
c.Path = filePath
|
||||
c.Path = c.absPath
|
||||
}
|
||||
return fileAbsErr
|
||||
}
|
||||
return nil
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Common) Type() string { return string(c.resourceType) }
|
||||
|
@ -6,6 +6,7 @@ import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
"net/url"
|
||||
"decl/internal/folio"
|
||||
)
|
||||
|
||||
func TestNewCommon(t *testing.T) {
|
||||
@ -16,16 +17,15 @@ func TestNewCommon(t *testing.T) {
|
||||
func TestCommon(t *testing.T) {
|
||||
expectedCommon := NewCommon(FileTypeName, false)
|
||||
expectedCommon.resourceType = "file"
|
||||
expectedCommon.Uri = "file:///tmp/foo"
|
||||
expectedCommon.parsedURI = &url.URL{ Scheme: "file", Path: "/tmp/foo"}
|
||||
expectedCommon.Path = "/tmp/foo"
|
||||
for _, v := range []struct{ uri string; expected *Common }{
|
||||
for _, v := range []struct{ uri folio.URI; expected *Common }{
|
||||
{ uri: "file:///tmp/foo", expected: expectedCommon },
|
||||
}{
|
||||
|
||||
c := NewCommon(FileTypeName, false)
|
||||
c.resourceType = "file"
|
||||
assert.Nil(t, c.SetURI(v.uri))
|
||||
assert.Nil(t, c.SetParsedURI(v.uri.Parse()))
|
||||
assert.Equal(t, v.expected.resourceType , c.resourceType)
|
||||
assert.Equal(t, v.expected.Path, c.Path)
|
||||
assert.Equal(t, v.expected.parsedURI.Scheme, c.parsedURI.Scheme)
|
||||
|
@ -28,6 +28,7 @@ _ "os/exec"
|
||||
"gitea.rosskeen.house/rosskeen.house/machine"
|
||||
"decl/internal/codec"
|
||||
"decl/internal/data"
|
||||
"decl/internal/folio"
|
||||
)
|
||||
|
||||
const (
|
||||
@ -50,7 +51,6 @@ type Container struct {
|
||||
stater machine.Stater `yaml:"-" json:"-"`
|
||||
Id string `json:"ID,omitempty" yaml:"ID,omitempty"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
// Path string `json:"path" yaml:"path"`
|
||||
Cmd []string `json:"cmd,omitempty" yaml:"cmd,omitempty"`
|
||||
Entrypoint strslice.StrSlice `json:"entrypoint,omitempty" yaml:"entrypoint,omitempty"`
|
||||
Args []string `json:"args,omitempty" yaml:"args,omitempty"`
|
||||
@ -81,21 +81,19 @@ type Container struct {
|
||||
NetworkSettings *NetworkSettings
|
||||
*/
|
||||
|
||||
// State string `yaml:"state,omitempty" json:"state,omitempty"`
|
||||
|
||||
// config ConfigurationValueGetter
|
||||
apiClient ContainerClient
|
||||
// Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
ResourceTypes.Register([]string{"container"}, func(u *url.URL) data.Resource {
|
||||
c := NewContainer(nil)
|
||||
c.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
if err := c.Common.SetParsedURI(u); err == nil {
|
||||
return c
|
||||
ResourceTypes.Register([]string{"container"}, func(u *url.URL) (c data.Resource) {
|
||||
c = NewContainer(nil)
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(c); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
@ -114,6 +112,22 @@ func NewContainer(containerClientApi ContainerClient) *Container {
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Container) Init(u data.URIParser) error {
|
||||
if u == nil {
|
||||
u = folio.URI(c.URI()).Parse()
|
||||
}
|
||||
uri := u.URL()
|
||||
c.Name = filepath.Join(uri.Hostname(), uri.Path)
|
||||
return c.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (c *Container) SetParsedURI(u data.URIParser) (err error) {
|
||||
if err = c.Common.SetParsedURI(u); err == nil {
|
||||
c.Name = filepath.Join(c.Common.parsedURI.Hostname(), c.Common.parsedURI.Path)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Container) SetResourceMapper(resources data.ResourceMapper) {
|
||||
c.Resources = resources
|
||||
}
|
||||
@ -122,7 +136,7 @@ func (c *Container) Clone() data.Resource {
|
||||
return &Container {
|
||||
Id: c.Id,
|
||||
Name: c.Name,
|
||||
Common: c.Common,
|
||||
Common: c.Common.Clone(),
|
||||
Cmd: c.Cmd,
|
||||
Entrypoint: c.Entrypoint,
|
||||
Args: c.Args,
|
||||
@ -146,7 +160,6 @@ func (c *Container) Clone() data.Resource {
|
||||
SizeRw: c.SizeRw,
|
||||
SizeRootFs: c.SizeRootFs,
|
||||
Networks: c.Networks,
|
||||
// State: c.State,
|
||||
apiClient: c.apiClient,
|
||||
}
|
||||
}
|
||||
@ -158,10 +171,6 @@ func (c *Container) StateMachine() machine.Stater {
|
||||
return c.stater
|
||||
}
|
||||
|
||||
func (c *Container) UseConfig(config data.ConfigurationValueGetter) {
|
||||
c.config = config
|
||||
}
|
||||
|
||||
func (c *Container) JSON() ([]byte, error) {
|
||||
return json.Marshal(c)
|
||||
}
|
||||
@ -429,7 +438,7 @@ func (c *Container) Type() string { return "container" }
|
||||
func (c *Container) ResolveId(ctx context.Context) string {
|
||||
var err error
|
||||
|
||||
if err = c.Common.SetURI(c.URI()); err != nil {
|
||||
if err = c.Common.SetParsedURI(folio.URI(c.URI()).Parse()); err != nil {
|
||||
triggerErr := c.StateMachine().Trigger("notexists")
|
||||
panic(fmt.Errorf("%w: %s %s, %w", err, c.Type(), c.Name, triggerErr))
|
||||
}
|
||||
|
@ -101,11 +101,14 @@ type ContainerImage struct {
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"container-image"}, func(u *url.URL) data.Resource {
|
||||
c := NewContainerImage(nil)
|
||||
c.Name = ContainerImageNameFromURI(u)
|
||||
slog.Info("NewContainerImage", "container", c)
|
||||
return c
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"container-image"}, func(u *url.URL) (c data.Resource) {
|
||||
c = NewContainerImage(nil)
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(c); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
@ -118,13 +121,24 @@ func NewContainerImage(containerClientApi ContainerImageClient) *ContainerImage
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return &ContainerImage{
|
||||
Common: &Common{ includeQueryParamsInURI: true, resourceType: ContainerImageTypeName },
|
||||
c := &ContainerImage{
|
||||
Common: NewCommon(ContainerImageTypeName, true),
|
||||
apiClient: apiClient,
|
||||
InjectJX: true,
|
||||
PushImage: false,
|
||||
ConverterTypes: folio.DocumentRegistry.ConverterTypes,
|
||||
}
|
||||
c.Common.NormalizePath = c.NormalizePath
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *ContainerImage) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(c.URI()).Parse()
|
||||
}
|
||||
err = c.SetParsedURI(u)
|
||||
c.Name = ContainerImageNameFromURI(u.URL())
|
||||
return
|
||||
}
|
||||
|
||||
func (c *ContainerImage) RegistryAuthConfig() (authConfig registry.AuthConfig, err error) {
|
||||
@ -175,6 +189,10 @@ func (c *ContainerImage) RegistryAuth() (string, error) {
|
||||
}
|
||||
}
|
||||
|
||||
func (c *ContainerImage) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *ContainerImage) SetResourceMapper(resources data.ResourceMapper) {
|
||||
c.Resources = resources
|
||||
}
|
||||
@ -249,24 +267,6 @@ func (c *ContainerImage) URI() string {
|
||||
return URIFromContainerImageName(c.Name)
|
||||
}
|
||||
|
||||
/*
|
||||
func (c *ContainerImage) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == c.Type() {
|
||||
c.Name = strings.Join([]string{resourceUri.Hostname(), resourceUri.RequestURI()}, ":")
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a %s", ErrInvalidResourceURI, uri, c.Type())
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (c *ContainerImage) UseConfig(config data.ConfigurationValueGetter) {
|
||||
c.config = config
|
||||
}
|
||||
*/
|
||||
|
||||
func (c *ContainerImage) JSON() ([]byte, error) {
|
||||
return json.Marshal(c)
|
||||
}
|
||||
@ -789,12 +789,6 @@ func (c *ContainerImage) Delete(ctx context.Context) error {
|
||||
|
||||
_, err := c.apiClient.ImageRemove(ctx, c.Id, options)
|
||||
return err
|
||||
/*
|
||||
for _, img := range deletedImages {
|
||||
fmt.Printf("Deleted image: %s\n", img.Deleted)
|
||||
fmt.Printf("Untagged image: %s\n", img.Untagged)
|
||||
}
|
||||
*/
|
||||
}
|
||||
|
||||
func (c *ContainerImage) Type() string { return "container-image" }
|
||||
@ -821,5 +815,3 @@ func (c *ContainerImage) ResolveId(ctx context.Context) string {
|
||||
}
|
||||
return c.Id
|
||||
}
|
||||
|
||||
|
||||
|
@ -15,7 +15,6 @@ _ "log/slog"
|
||||
"net/url"
|
||||
_ "os"
|
||||
_ "os/exec"
|
||||
"path/filepath"
|
||||
_ "strings"
|
||||
"encoding/json"
|
||||
"io"
|
||||
@ -48,21 +47,24 @@ type ContainerNetwork struct {
|
||||
Internal bool `json:"internal,omitempty" yaml:"internal,omitempty"`
|
||||
Labels map[string]string `json:"labels,omitempty" yaml:"labels,omitempty"`
|
||||
Created time.Time `json:"created" yaml:"created"`
|
||||
//State string `yaml:"state"`
|
||||
|
||||
apiClient ContainerNetworkClient
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"container-network"}, func(u *url.URL) data.Resource {
|
||||
n := NewContainerNetwork(nil)
|
||||
n.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
return n
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"container-network"}, func(u *url.URL) (n data.Resource) {
|
||||
n = NewContainerNetwork(nil)
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(n); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
func NewContainerNetwork(containerClientApi ContainerNetworkClient) *ContainerNetwork {
|
||||
func NewContainerNetwork(containerClientApi ContainerNetworkClient) (cn *ContainerNetwork) {
|
||||
var apiClient ContainerNetworkClient = containerClientApi
|
||||
if apiClient == nil {
|
||||
var err error
|
||||
@ -71,10 +73,23 @@ func NewContainerNetwork(containerClientApi ContainerNetworkClient) *ContainerNe
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return &ContainerNetwork{
|
||||
Common: &Common{ includeQueryParamsInURI: true, resourceType: ContainerNetworkTypeName },
|
||||
cn = &ContainerNetwork{
|
||||
apiClient: apiClient,
|
||||
}
|
||||
cn.Common = NewCommon(ContainerNetworkTypeName, true)
|
||||
cn.Common.NormalizePath = cn.NormalizePath
|
||||
return cn
|
||||
}
|
||||
|
||||
func (n *ContainerNetwork) Init(u data.URIParser) error {
|
||||
if u == nil {
|
||||
u = folio.URI(n.URI()).Parse()
|
||||
}
|
||||
return n.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (n *ContainerNetwork) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (n *ContainerNetwork) SetResourceMapper(resources data.ResourceMapper) {
|
||||
@ -86,7 +101,6 @@ func (n *ContainerNetwork) Clone() data.Resource {
|
||||
Common: n.Common,
|
||||
Id: n.Id,
|
||||
Name: n.Name,
|
||||
//State: n.State,
|
||||
apiClient: n.apiClient,
|
||||
}
|
||||
}
|
||||
@ -148,24 +162,6 @@ func (n *ContainerNetwork) URI() string {
|
||||
return fmt.Sprintf("container-network://%s", n.Name)
|
||||
}
|
||||
|
||||
/*
|
||||
func (n *ContainerNetwork) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == n.Type() {
|
||||
n.Name, e = filepath.Abs(filepath.Join(resourceUri.Hostname(), resourceUri.RequestURI()))
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a %s", ErrInvalidResourceURI, uri, n.Type())
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (n *ContainerNetwork) UseConfig(config data.ConfigurationValueGetter) {
|
||||
n.config = config
|
||||
}
|
||||
*/
|
||||
|
||||
func (n *ContainerNetwork) JSON() ([]byte, error) {
|
||||
return json.Marshal(n)
|
||||
}
|
||||
|
@ -39,9 +39,22 @@ type Exec struct {
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"exec"}, func(u *url.URL) data.Resource {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"exec"}, func(u *url.URL) (res data.Resource) {
|
||||
x := NewExec()
|
||||
return x
|
||||
res = x
|
||||
if u != nil {
|
||||
uri := folio.CastParsedURI(u)
|
||||
if ri, ok := res.(data.ResourceInitializer); ok {
|
||||
if err := ri.Init(uri); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
} else {
|
||||
if err := x.SetParsedURI(uri); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
@ -56,7 +69,7 @@ func (x *Exec) SetResourceMapper(resources data.ResourceMapper) {
|
||||
|
||||
func (x *Exec) Clone() data.Resource {
|
||||
return &Exec {
|
||||
Common: x.Common,
|
||||
Common: x.Common.Clone(),
|
||||
Id: x.Id,
|
||||
CreateTemplate: x.CreateTemplate,
|
||||
ReadTemplate: x.ReadTemplate,
|
||||
@ -76,13 +89,16 @@ func (x *Exec) URI() string {
|
||||
return fmt.Sprintf("exec://%s", x.Id)
|
||||
}
|
||||
|
||||
func (x *Exec) SetURI(uri string) (err error) {
|
||||
err = x.Common.SetURI(uri)
|
||||
func (x *Exec) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(x.URI()).Parse()
|
||||
}
|
||||
err = x.SetParsedURI(u)
|
||||
x.Id = x.Common.Path
|
||||
return
|
||||
}
|
||||
|
||||
func (x *Exec) SetParsedURI(uri *url.URL) (err error) {
|
||||
func (x *Exec) SetParsedURI(uri data.URIParser) (err error) {
|
||||
err = x.Common.SetParsedURI(uri)
|
||||
x.Id = x.Common.Path
|
||||
return
|
||||
|
@ -18,6 +18,7 @@ import (
|
||||
_ "strings"
|
||||
"testing"
|
||||
"decl/internal/command"
|
||||
"decl/internal/folio"
|
||||
)
|
||||
|
||||
func TestNewExecResource(t *testing.T) {
|
||||
@ -55,9 +56,10 @@ func TestCreateExec(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestExecSetURI(t *testing.T) {
|
||||
var uri folio.URI = "exec://12345_key"
|
||||
x := NewExec()
|
||||
assert.NotNil(t, x)
|
||||
e := x.SetURI("exec://" + "12345_key")
|
||||
e := x.SetParsedURI(uri.Parse())
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, "exec", x.Type())
|
||||
assert.Equal(t, "12345_key", x.Path)
|
||||
|
@ -14,7 +14,6 @@ import (
|
||||
"net/url"
|
||||
"os"
|
||||
"os/user"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"syscall"
|
||||
"time"
|
||||
@ -56,19 +55,14 @@ var ErrInvalidFileGroup error = errors.New("Unknown Group")
|
||||
type FileMode string
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"file"}, func(u *url.URL) data.Resource {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"file"}, func(u *url.URL) (res data.Resource) {
|
||||
f := NewFile()
|
||||
f.parsedURI = u
|
||||
//f.Uri.SetURL(u)
|
||||
f.Path = filepath.Join(u.Hostname(), u.Path)
|
||||
f.exttype, f.fileext = f.Uri.Extension()
|
||||
|
||||
slog.Info("folio.DocumentRegistry.ResourceTypes.Register()()", "url", u, "file", f)
|
||||
/*
|
||||
if absPath, err := filepath.Abs(f.Path); err == nil {
|
||||
f.Filesystem = os.DirFS(filepath.Dir(absPath))
|
||||
slog.Info("FileFactory", "uri", u)
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(f); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
*/
|
||||
return f
|
||||
})
|
||||
}
|
||||
@ -83,18 +77,12 @@ The `SerializeContent` the flag allows forcing the content to be serialized in t
|
||||
|
||||
*/
|
||||
type File struct {
|
||||
Uri folio.URI `json:"uri,omitempty" yaml:"uri,omitempty"`
|
||||
parsedURI *url.URL `json:"-" yaml:"-"`
|
||||
*Common `json:",inline" yaml:",inline"`
|
||||
Filesystem fs.FS `json:"-" yaml:"-"`
|
||||
|
||||
exttype string `json:"-" yaml:"-"`
|
||||
fileext string `json:"-" yaml:"-"`
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
normalizePath bool `json:"-" yaml:"-"`
|
||||
absPath string `json:"-" yaml:"-"`
|
||||
basePath int `json:"-" yaml:"-"`
|
||||
|
||||
Path string `json:"path" yaml:"path"`
|
||||
Owner string `json:"owner" yaml:"owner"`
|
||||
Group string `json:"group" yaml:"group"`
|
||||
Mode FileMode `json:"mode" yaml:"mode"`
|
||||
@ -109,10 +97,8 @@ type File struct {
|
||||
Size int64 `json:"size,omitempty" yaml:"size,omitempty"`
|
||||
Target string `json:"target,omitempty" yaml:"target,omitempty"`
|
||||
FileType FileType `json:"filetype" yaml:"filetype"`
|
||||
State string `json:"state,omitempty" yaml:"state,omitempty"`
|
||||
SerializeContent bool `json:"serializecontent,omitempty" yaml:"serializecontent,omitempty"`
|
||||
GzipContent bool `json:"gzipcontent,omitempty" yaml:"gzipcontent,omitempty"`
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
@ -123,23 +109,49 @@ type ResourceFileInfo struct {
|
||||
func NewFile() *File {
|
||||
currentUser, _ := user.Current()
|
||||
group, _ := user.LookupGroupId(currentUser.Gid)
|
||||
f := &File{ normalizePath: false, Owner: currentUser.Username, Group: group.Name, Mode: "0644", FileType: RegularFile, SerializeContent: false }
|
||||
f := &File{
|
||||
Common: NewCommon(FileTypeName, true),
|
||||
Owner: currentUser.Username,
|
||||
Group: group.Name,
|
||||
Mode: "0644",
|
||||
FileType: RegularFile,
|
||||
SerializeContent: false,
|
||||
}
|
||||
f.PathNormalization(false)
|
||||
slog.Info("NewFile()", "file", f)
|
||||
return f
|
||||
}
|
||||
|
||||
func NewNormalizedFile() *File {
|
||||
f := NewFile()
|
||||
f.normalizePath = true
|
||||
f.PathNormalization(true)
|
||||
return f
|
||||
}
|
||||
|
||||
func (f *File) Init(u data.URIParser) error {
|
||||
if u == nil {
|
||||
u = folio.URI(f.URI()).Parse()
|
||||
}
|
||||
return f.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (f *File) NormalizePath() error {
|
||||
return f.Common.NormalizePath()
|
||||
}
|
||||
|
||||
func (f *File) ContentType() string {
|
||||
var ext strings.Builder
|
||||
if f.parsedURI.Scheme != "file" {
|
||||
return f.parsedURI.Scheme
|
||||
}
|
||||
if f.fileext == "" {
|
||||
return f.exttype
|
||||
}
|
||||
ext.WriteString(f.exttype)
|
||||
ext.WriteRune('.')
|
||||
ext.WriteString(f.fileext)
|
||||
return ext.String()
|
||||
}
|
||||
|
||||
func (f *File) SetResourceMapper(resources data.ResourceMapper) {
|
||||
f.Resources = resources
|
||||
@ -147,13 +159,7 @@ func (f *File) SetResourceMapper(resources data.ResourceMapper) {
|
||||
|
||||
func (f *File) Clone() data.Resource {
|
||||
return &File {
|
||||
Uri: f.Uri,
|
||||
parsedURI: f.parsedURI,
|
||||
exttype: f.exttype,
|
||||
fileext: f.fileext,
|
||||
normalizePath: f.normalizePath,
|
||||
absPath: f.absPath,
|
||||
Path: f.Path,
|
||||
Common: f.Common.Clone(),
|
||||
Owner: f.Owner,
|
||||
Group: f.Group,
|
||||
Mode: f.Mode,
|
||||
@ -165,7 +171,6 @@ func (f *File) Clone() data.Resource {
|
||||
Size: f.Size,
|
||||
Target: f.Target,
|
||||
FileType: f.FileType,
|
||||
State: f.State,
|
||||
}
|
||||
}
|
||||
|
||||
@ -197,11 +202,11 @@ func (f *File) Notify(m *machine.EventMessage) {
|
||||
if triggerErr := f.StateMachine().Trigger("state_read"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
f.State = "absent"
|
||||
f.Common.State = "absent"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
f.State = "absent"
|
||||
f.Common.State = "absent"
|
||||
if ! errors.Is(readErr, ErrResourceStateAbsent) {
|
||||
panic(readErr)
|
||||
}
|
||||
@ -212,25 +217,36 @@ func (f *File) Notify(m *machine.EventMessage) {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
f.State = "absent"
|
||||
f.Common.State = "absent"
|
||||
panic(e)
|
||||
}
|
||||
case "start_update":
|
||||
if updateErr := f.Update(ctx); updateErr == nil {
|
||||
if triggerErr := f.stater.Trigger("updated"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
f.Common.State = "absent"
|
||||
}
|
||||
} else {
|
||||
f.Common.State = "absent"
|
||||
panic(updateErr)
|
||||
}
|
||||
case "start_delete":
|
||||
if deleteErr := f.Delete(ctx); deleteErr == nil {
|
||||
if triggerErr := f.StateMachine().Trigger("deleted"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
f.State = "present"
|
||||
f.Common.State = "present"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
f.State = "present"
|
||||
f.Common.State = "present"
|
||||
panic(deleteErr)
|
||||
}
|
||||
case "absent":
|
||||
f.State = "absent"
|
||||
f.Common.State = "absent"
|
||||
case "present", "created", "read":
|
||||
f.State = "present"
|
||||
f.Common.State = "present"
|
||||
}
|
||||
case machine.EXITSTATEEVENT:
|
||||
}
|
||||
@ -240,12 +256,8 @@ func (f *File) SetGzipContent(flag bool) {
|
||||
f.GzipContent = flag
|
||||
}
|
||||
|
||||
func (f *File) PathNormalization(flag bool) {
|
||||
f.normalizePath = flag
|
||||
}
|
||||
|
||||
func (f *File) FilePath() string {
|
||||
return f.Path
|
||||
return f.Common.Path
|
||||
}
|
||||
|
||||
func (f *File) SetFS(fsys fs.FS) {
|
||||
@ -253,63 +265,23 @@ func (f *File) SetFS(fsys fs.FS) {
|
||||
}
|
||||
|
||||
func (f *File) URI() string {
|
||||
return fmt.Sprintf("file://%s", f.Path)
|
||||
return fmt.Sprintf("file://%s", f.Common.Path)
|
||||
}
|
||||
|
||||
func (f *File) RelativePath() string {
|
||||
return f.Path[f.basePath:]
|
||||
return f.Common.Path[f.basePath:]
|
||||
}
|
||||
|
||||
func (f *File) SetBasePath(index int) {
|
||||
if index < len(f.Path) {
|
||||
if index < len(f.Common.Path) {
|
||||
f.basePath = index
|
||||
}
|
||||
}
|
||||
|
||||
func (f *File) SetURI(uri string) (err error) {
|
||||
slog.Info("File.SetURI()", "uri", uri, "file", f, "parsed", f.parsedURI)
|
||||
f.SetURIFromString(uri)
|
||||
err = f.SetParsedURI(f.Uri.Parse())
|
||||
return
|
||||
}
|
||||
|
||||
func (f *File) DetectGzip() bool {
|
||||
return (f.parsedURI.Query().Get("gzip") == "true" || f.fileext == "gz" || f.exttype == "tgz" || f.exttype == "gz" || f.fileext == "tgz" )
|
||||
}
|
||||
|
||||
func (f *File) SetURIFromString(uri string) {
|
||||
f.Uri = folio.URI(uri)
|
||||
f.exttype, f.fileext = f.Uri.Extension()
|
||||
}
|
||||
|
||||
func (f *File) SetParsedURI(u *url.URL) (err error) {
|
||||
if u != nil {
|
||||
if u.Scheme == "" {
|
||||
u.Scheme = "file"
|
||||
f.Uri = ""
|
||||
}
|
||||
if f.Uri.IsEmpty() {
|
||||
f.SetURIFromString(u.String())
|
||||
}
|
||||
slog.Info("File.SetParsedURI()", "parsed", u, "path", f.Path)
|
||||
f.parsedURI = u
|
||||
if f.parsedURI.Scheme == "file" {
|
||||
f.Path = filepath.Join(f.parsedURI.Hostname(), f.parsedURI.Path)
|
||||
slog.Info("File.SetParsedURI()", "path", f.Path)
|
||||
if err = f.NormalizePath(); err != nil {
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
err = fmt.Errorf("%w: %s is not a file", ErrInvalidResourceURI, f.Uri)
|
||||
return
|
||||
}
|
||||
|
||||
func (f *File) UseConfig(config data.ConfigurationValueGetter) {
|
||||
f.config = config
|
||||
}
|
||||
|
||||
func (f *File) JSON() ([]byte, error) {
|
||||
return json.Marshal(f)
|
||||
}
|
||||
@ -325,7 +297,7 @@ func (f *File) Validate() (err error) {
|
||||
|
||||
func (f *File) Apply() error {
|
||||
ctx := context.Background()
|
||||
switch f.State {
|
||||
switch f.Common.State {
|
||||
case "absent":
|
||||
return f.Delete(ctx)
|
||||
case "present":
|
||||
@ -367,9 +339,10 @@ func (f *File) ResolveId(ctx context.Context) string {
|
||||
if e := f.NormalizePath(); e != nil {
|
||||
panic(e)
|
||||
}
|
||||
return f.Path
|
||||
return f.Common.Path
|
||||
}
|
||||
|
||||
/*
|
||||
func (f *File) NormalizePath() (err error) {
|
||||
if f.config != nil {
|
||||
if prefixPath, configErr := f.config.GetValue("prefix"); configErr == nil {
|
||||
@ -381,6 +354,7 @@ func (f *File) NormalizePath() (err error) {
|
||||
}
|
||||
return
|
||||
}
|
||||
*/
|
||||
|
||||
func (f *File) GetContentSourceRef() string {
|
||||
return string(f.ContentSourceRef)
|
||||
@ -393,7 +367,7 @@ func (f *File) SetContentSourceRef(uri string) {
|
||||
|
||||
func (f *File) Stat() (info fs.FileInfo, err error) {
|
||||
if _, ok := f.Filesystem.(embed.FS); ok {
|
||||
info, err = fs.Stat(f.Filesystem, f.Path)
|
||||
info, err = fs.Stat(f.Filesystem, f.Common.Path)
|
||||
} else {
|
||||
info, err = os.Lstat(f.absPath)
|
||||
}
|
||||
@ -453,12 +427,12 @@ func (f *File) Create(ctx context.Context) error {
|
||||
|
||||
switch f.FileType {
|
||||
case SymbolicLinkFile:
|
||||
linkErr := os.Symlink(f.Target, f.Path)
|
||||
linkErr := os.Symlink(f.Target, f.Common.Path)
|
||||
if linkErr != nil {
|
||||
return linkErr
|
||||
}
|
||||
case DirectoryFile:
|
||||
if mkdirErr := os.MkdirAll(f.Path, mode); mkdirErr != nil {
|
||||
if mkdirErr := os.MkdirAll(f.Common.Path, mode); mkdirErr != nil {
|
||||
return mkdirErr
|
||||
}
|
||||
default:
|
||||
@ -486,7 +460,8 @@ func (f *File) Create(ctx context.Context) error {
|
||||
})
|
||||
|
||||
var createdFileWriter io.WriteCloser
|
||||
createdFile, fileErr := os.Create(f.Path)
|
||||
createdFile, fileErr := os.Create(f.Common.Path)
|
||||
slog.Info("File.Create(): os.Create()", "path", f.Common.Path, "error", fileErr)
|
||||
if fileErr != nil {
|
||||
return fileErr
|
||||
}
|
||||
@ -500,6 +475,7 @@ func (f *File) Create(ctx context.Context) error {
|
||||
|
||||
defer createdFile.Close()
|
||||
|
||||
slog.Info("File.Create(): Chmod()", "path", f.Common.Path, "mode", mode)
|
||||
if chmodErr := createdFile.Chmod(mode); chmodErr != nil {
|
||||
return chmodErr
|
||||
}
|
||||
@ -511,16 +487,20 @@ func (f *File) Create(ctx context.Context) error {
|
||||
|
||||
f.Sha256 = fmt.Sprintf("%x", hash.Sum(nil))
|
||||
if !f.Mtime.IsZero() && !f.Atime.IsZero() {
|
||||
if chtimesErr := os.Chtimes(f.Path, f.Atime, f.Mtime); chtimesErr != nil {
|
||||
slog.Info("File.Create(): Chtimes()", "path", f.Common.Path, "atime", f.Atime, "mtime", f.Mtime)
|
||||
if chtimesErr := os.Chtimes(f.Common.Path, f.Atime, f.Mtime); chtimesErr != nil {
|
||||
return chtimesErr
|
||||
}
|
||||
} else {
|
||||
slog.Info("File.Create(): Chtimes() SKIPPED", "path", f.Common.Path, "atime", f.Atime, "mtime", f.Mtime)
|
||||
}
|
||||
}
|
||||
slog.Info("File.Create(): Chown()", "path", f.Common.Path, "uid", uid, "gid", gid)
|
||||
|
||||
if chownErr := os.Chown(f.Path, uid, gid); chownErr != nil {
|
||||
if chownErr := os.Chown(f.Common.Path, uid, gid); chownErr != nil {
|
||||
return chownErr
|
||||
}
|
||||
f.State = "present"
|
||||
f.Common.State = "present"
|
||||
return nil
|
||||
}
|
||||
|
||||
@ -529,7 +509,7 @@ func (f *File) Update(ctx context.Context) error {
|
||||
}
|
||||
|
||||
func (f *File) Delete(ctx context.Context) error {
|
||||
return os.Remove(f.Path)
|
||||
return os.Remove(f.Common.Path)
|
||||
}
|
||||
|
||||
func (f *File) UpdateContentAttributes() {
|
||||
@ -585,11 +565,11 @@ func (f *File) ContentSourceRefStat() (info fs.FileInfo) {
|
||||
|
||||
func (f *File) ReadStat() (err error) {
|
||||
var info fs.FileInfo
|
||||
slog.Info("ReadStat()", "filesystem", f.Filesystem, "path", f.Path)
|
||||
slog.Info("ReadStat()", "filesystem", f.Filesystem, "path", f.Common.Path)
|
||||
|
||||
info, err = f.Stat()
|
||||
|
||||
slog.Info("ReadStat()", "filesystem", f.Filesystem, "path", f.Path, "info", info, "error", err)
|
||||
slog.Info("ReadStat()", "filesystem", f.Filesystem, "path", f.Common.Path, "info", info, "error", err)
|
||||
|
||||
if err == nil {
|
||||
_ = f.SetFileInfo(info)
|
||||
@ -603,9 +583,9 @@ func (f *File) ReadStat() (err error) {
|
||||
}
|
||||
|
||||
|
||||
slog.Info("ReadStat()", "stat", info, "path", f.Path)
|
||||
slog.Info("ReadStat()", "stat", info, "path", f.Common.Path)
|
||||
if err != nil {
|
||||
f.State = "absent"
|
||||
f.Common.State = "absent"
|
||||
return
|
||||
}
|
||||
|
||||
@ -613,27 +593,29 @@ func (f *File) ReadStat() (err error) {
|
||||
}
|
||||
|
||||
func (f *File) open() (file io.ReadCloser, err error) {
|
||||
slog.Info("open()", "file", f.Path, "fs", f.Filesystem)
|
||||
slog.Info("open()", "file", f.Common.Path, "fs", f.Filesystem)
|
||||
if _, ok := f.Filesystem.(embed.FS); ok {
|
||||
file, err = f.Filesystem.Open(f.Path)
|
||||
file, err = f.Filesystem.Open(f.Common.Path)
|
||||
} else {
|
||||
file, err = os.Open(f.Path)
|
||||
file, err = os.Open(f.Common.Path)
|
||||
}
|
||||
if f.GzipContent && f.DetectGzip() {
|
||||
file, err = gzip.NewReader(file)
|
||||
}
|
||||
slog.Info("open()", "file", f.Path, "error", err)
|
||||
slog.Info("open()", "file", f.Common.Path, "error", err)
|
||||
return
|
||||
}
|
||||
|
||||
func (f *File) Read(ctx context.Context) ([]byte, error) {
|
||||
/*
|
||||
if normalizePathErr := f.NormalizePath(); normalizePathErr != nil {
|
||||
return nil, normalizePathErr
|
||||
}
|
||||
*/
|
||||
|
||||
statErr := f.ReadStat()
|
||||
if statErr != nil {
|
||||
return nil, fmt.Errorf("%w - %w", ErrResourceStateAbsent, statErr)
|
||||
return nil, fmt.Errorf("%w - %w: %s", ErrResourceStateAbsent, statErr, f.Path)
|
||||
}
|
||||
|
||||
switch f.FileType {
|
||||
@ -653,13 +635,13 @@ func (f *File) Read(ctx context.Context) ([]byte, error) {
|
||||
f.Sha256 = fmt.Sprintf("%x", sha256.Sum256(fileContent))
|
||||
}
|
||||
case SymbolicLinkFile:
|
||||
linkTarget, pathErr := os.Readlink(f.Path)
|
||||
linkTarget, pathErr := os.Readlink(f.Common.Path)
|
||||
if pathErr != nil {
|
||||
return nil, pathErr
|
||||
}
|
||||
f.Target = linkTarget
|
||||
}
|
||||
f.State = "present"
|
||||
f.Common.State = "present"
|
||||
return yaml.Marshal(f)
|
||||
}
|
||||
|
||||
|
@ -23,6 +23,7 @@ import (
|
||||
"io/fs"
|
||||
"decl/internal/codec"
|
||||
"decl/internal/data"
|
||||
"decl/internal/folio"
|
||||
"log/slog"
|
||||
)
|
||||
|
||||
@ -37,7 +38,8 @@ func TestNewFileNormalized(t *testing.T) {
|
||||
|
||||
f := NewNormalizedFile()
|
||||
assert.NotNil(t, f)
|
||||
assert.Nil(t, f.SetURI("file://" + indirectFile))
|
||||
f.Path = indirectFile
|
||||
assert.Nil(t, f.Init(nil))
|
||||
|
||||
assert.NotEqual(t, indirectFile, f.Path)
|
||||
assert.Equal(t, absFilePath, f.Path)
|
||||
@ -84,13 +86,21 @@ func TestReadFile(t *testing.T) {
|
||||
testFile := NewFile()
|
||||
e := testFile.LoadDecl(decl)
|
||||
assert.Nil(t, e)
|
||||
|
||||
assert.Equal(t, "present", testFile.Common.State)
|
||||
assert.Equal(t, file, testFile.Common.Path)
|
||||
|
||||
applyErr := testFile.Apply()
|
||||
assert.Nil(t, applyErr)
|
||||
|
||||
assert.FileExists(t, file)
|
||||
|
||||
f := NewFile()
|
||||
assert.NotEqual(t, nil, f)
|
||||
assert.NotNil(t, f)
|
||||
|
||||
f.Path = file
|
||||
assert.Nil(t, f.Init(nil))
|
||||
|
||||
r, e := f.Read(ctx)
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, ProcessTestUserName, f.Owner)
|
||||
@ -118,7 +128,8 @@ func TestUseConfig(t *testing.T) {
|
||||
return nil, data.ErrUnknownConfigurationKey
|
||||
}))
|
||||
|
||||
assert.Nil(t, f.SetURI(fmt.Sprintf("file://%s", file)))
|
||||
uri := folio.URI(fmt.Sprintf("file://%s", file))
|
||||
assert.Nil(t, f.Init(uri.Parse()))
|
||||
assert.Equal(t, filepath.Join("/tmp", file), f.FilePath())
|
||||
}
|
||||
|
||||
@ -263,8 +274,8 @@ func TestFileSetURI(t *testing.T) {
|
||||
file, _ := filepath.Abs(TempDir.FilePath("testuri.txt"))
|
||||
f := NewFile()
|
||||
assert.NotNil(t, f)
|
||||
e := f.SetURI("file://" + file)
|
||||
assert.Nil(t, e)
|
||||
uri := folio.URI("file://" + file).Parse()
|
||||
assert.Nil(t, f.Init(uri))
|
||||
assert.Equal(t, "file", f.Type())
|
||||
assert.Equal(t, file, f.Path)
|
||||
}
|
||||
@ -302,6 +313,7 @@ func TestFileUpdateAttributesFromFileInfo(t *testing.T) {
|
||||
|
||||
func TestFileReadStat(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
link := TempDir.FilePath("link.txt")
|
||||
linkTargetFile := TempDir.FilePath("testuri.txt")
|
||||
|
||||
@ -309,8 +321,8 @@ func TestFileReadStat(t *testing.T) {
|
||||
assert.NotNil(t, f)
|
||||
|
||||
f.Path = linkTargetFile
|
||||
e := f.NormalizePath()
|
||||
assert.Nil(t, e)
|
||||
f.PathNormalization(true)
|
||||
assert.Nil(t, f.Init(nil))
|
||||
|
||||
statErr := f.ReadStat()
|
||||
assert.Error(t, statErr)
|
||||
@ -324,8 +336,9 @@ func TestFileReadStat(t *testing.T) {
|
||||
|
||||
l := NewFile()
|
||||
assert.NotNil(t, l)
|
||||
l.PathNormalization(true)
|
||||
assert.Nil(t, l.Init(nil))
|
||||
|
||||
assert.Nil(t, l.NormalizePath())
|
||||
l.FileType = SymbolicLinkFile
|
||||
l.Path = link
|
||||
l.Target = linkTargetFile
|
||||
@ -340,6 +353,9 @@ func TestFileReadStat(t *testing.T) {
|
||||
|
||||
testRead := NewFile()
|
||||
testRead.Path = link
|
||||
|
||||
assert.Nil(t, testRead.Init(nil))
|
||||
|
||||
_,testReadErr := testRead.Read(ctx)
|
||||
assert.Nil(t, testReadErr)
|
||||
assert.Equal(t, linkTargetFile, testRead.Target)
|
||||
@ -355,6 +371,8 @@ func TestFileResourceFileInfo(t *testing.T) {
|
||||
f.Mode = "0600"
|
||||
f.Content = "some test data"
|
||||
f.State = "present"
|
||||
assert.Nil(t, f.Init(nil))
|
||||
|
||||
assert.Nil(t, f.Apply())
|
||||
|
||||
_, readErr := f.Read(context.Background())
|
||||
@ -378,20 +396,45 @@ func TestFileClone(t *testing.T) {
|
||||
assert.NotNil(t, f)
|
||||
|
||||
f.Path = testFile
|
||||
assert.Nil(t, f.Init(nil))
|
||||
f.Mode = "0600"
|
||||
f.State = "present"
|
||||
assert.Nil(t, f.Apply())
|
||||
|
||||
origin := time.Now()
|
||||
|
||||
_,readErr := f.Read(ctx)
|
||||
assert.Nil(t, readErr)
|
||||
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
assert.Greater(t, origin, f.Mtime)
|
||||
|
||||
clone := f.Clone().(*File)
|
||||
assert.Equal(t, f, clone)
|
||||
clone.Mtime = time.Time{}
|
||||
assert.Equal(t, f.Common.Path, clone.Common.Path)
|
||||
assert.Equal(t, f.Common.absPath, clone.Common.absPath)
|
||||
assert.Equal(t, f.Common.parsedURI, clone.Common.parsedURI)
|
||||
assert.Equal(t, f.Common.exttype, clone.Common.exttype)
|
||||
assert.Equal(t, f.Common.fileext, clone.Common.fileext)
|
||||
assert.Equal(t, f.Common.State, clone.Common.State)
|
||||
assert.Equal(t, f.Size, clone.Size)
|
||||
assert.Equal(t, f.Owner, clone.Owner)
|
||||
assert.Equal(t, f.Group, clone.Group)
|
||||
assert.Equal(t, f.Mode, clone.Mode)
|
||||
assert.Equal(t, f.Atime, clone.Atime)
|
||||
assert.Equal(t, f.Mtime, clone.Mtime)
|
||||
assert.Equal(t, f.Ctime, clone.Ctime)
|
||||
assert.Equal(t, f.Content, clone.Content)
|
||||
assert.Equal(t, f.Sha256, clone.Sha256)
|
||||
|
||||
clone.Mtime = time.Now()
|
||||
clone.Path = testCloneFile
|
||||
assert.Nil(t, clone.Init(nil))
|
||||
|
||||
assert.NotEqual(t, f.absPath, clone.absPath)
|
||||
|
||||
slog.Info("TestFileClone", "clone", clone)
|
||||
assert.Nil(t, clone.Apply())
|
||||
slog.Info("TestFileClone - applied mtime change", "clone", clone)
|
||||
|
||||
_,updateReadErr := f.Read(ctx)
|
||||
assert.Nil(t, updateReadErr)
|
||||
@ -399,7 +442,8 @@ func TestFileClone(t *testing.T) {
|
||||
_, cloneReadErr := clone.Read(ctx)
|
||||
assert.Nil(t, cloneReadErr)
|
||||
|
||||
fmt.Printf("file %#v\nclone %#v\n", f, clone)
|
||||
slog.Info("TestFileClone - read mtime change", "orig", f.Mtime, "clone", clone.Mtime)
|
||||
fmt.Printf("file %#v\n %#v\nclone %#v\n %#v\n", f, f.Common, clone, clone.Common)
|
||||
assert.NotEqual(t, f.Mtime, clone.Mtime)
|
||||
}
|
||||
|
||||
@ -412,12 +456,16 @@ func TestFileErrors(t *testing.T) {
|
||||
stater := f.StateMachine()
|
||||
|
||||
f.Path = testFile
|
||||
assert.Nil(t, f.Init(nil))
|
||||
f.Mode = "631"
|
||||
assert.Nil(t, stater.Trigger("create"))
|
||||
|
||||
assert.FileExists(t, f.Path)
|
||||
|
||||
read := NewFile()
|
||||
readStater := read.StateMachine()
|
||||
read.Path = testFile
|
||||
assert.Nil(t, read.Init(nil))
|
||||
assert.Nil(t, readStater.Trigger("read"))
|
||||
assert.Equal(t, FileMode("0631"), read.Mode)
|
||||
|
||||
@ -534,7 +582,8 @@ func TestFilePathURI(t *testing.T) {
|
||||
e := f.LoadDecl(decl)
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, "", f.FilePath())
|
||||
assert.ErrorContains(t, f.Validate(), "path: String length must be greater than or equal to 1")
|
||||
// assert.ErrorContains(t, f.Validate(), "path: String length must be greater than or equal to 1")
|
||||
assert.ErrorContains(t, f.Validate(), "path is required")
|
||||
}
|
||||
|
||||
func TestFileAbsent(t *testing.T) {
|
||||
@ -593,8 +642,8 @@ func TestFileSetURIError(t *testing.T) {
|
||||
file := TempDir.FilePath("fooread.txt")
|
||||
f := NewFile()
|
||||
assert.NotNil(t, f)
|
||||
e := f.SetURI("foo://" + file)
|
||||
assert.NotNil(t, e)
|
||||
uri := folio.URI("foo://" + file).Parse()
|
||||
e := f.Init(uri)
|
||||
assert.ErrorIs(t, e, ErrInvalidResourceURI)
|
||||
}
|
||||
|
||||
@ -602,8 +651,8 @@ func TestFileContentType(t *testing.T) {
|
||||
file := TempDir.FilePath("fooread.txt")
|
||||
f := NewFile()
|
||||
assert.NotNil(t, f)
|
||||
e := f.SetURI("file://" + file)
|
||||
assert.Nil(t, e)
|
||||
uri := folio.URI("file://" + file).Parse()
|
||||
assert.Nil(t, f.Init(uri))
|
||||
assert.Equal(t, "txt", f.ContentType())
|
||||
}
|
||||
|
||||
|
@ -27,8 +27,9 @@ type decodeGroup Group
|
||||
type GroupType string
|
||||
|
||||
const (
|
||||
GroupTypeAddGroup = "addgroup"
|
||||
GroupTypeGroupAdd = "groupadd"
|
||||
GroupTypeName TypeName = "group"
|
||||
GroupTypeAddGroup GroupType = "addgroup"
|
||||
GroupTypeGroupAdd GroupType = "groupadd"
|
||||
)
|
||||
|
||||
var ErrUnsupportedGroupType error = errors.New("The GroupType is not supported on this system")
|
||||
@ -47,28 +48,26 @@ type Group struct {
|
||||
ReadCommand *command.Command `json:"-" yaml:"-"`
|
||||
UpdateCommand *command.Command `json:"-" yaml:"-"`
|
||||
DeleteCommand *command.Command `json:"-" yaml:"-"`
|
||||
State string `json:"state,omitempty" yaml:"state,omitempty"`
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
groupStatus *user.Group `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewGroup() *Group {
|
||||
return &Group{}
|
||||
func NewGroup() (g *Group) {
|
||||
g = &Group{}
|
||||
g.Common = NewCommon(GroupTypeName, true)
|
||||
g.Common.NormalizePath = g.NormalizePath
|
||||
return
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"group"}, func(u *url.URL) data.Resource {
|
||||
group := NewGroup()
|
||||
group.Name = u.Hostname()
|
||||
group.GID = LookupGIDString(u.Hostname())
|
||||
if _, addGroupPathErr := exec.LookPath("addgroup"); addGroupPathErr == nil {
|
||||
group.GroupType = GroupTypeAddGroup
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"group"}, func(u *url.URL) (group data.Resource) {
|
||||
group = NewGroup()
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(group); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if _, pathErr := exec.LookPath("groupadd"); pathErr == nil {
|
||||
group.GroupType = GroupTypeGroupAdd
|
||||
}
|
||||
group.CreateCommand, group.ReadCommand, group.UpdateCommand, group.DeleteCommand = group.GroupType.NewCRUD()
|
||||
return group
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
@ -82,15 +81,36 @@ func FindSystemGroupType() GroupType {
|
||||
return GroupTypeAddGroup
|
||||
}
|
||||
|
||||
func (g *Group) Init(u data.URIParser) error {
|
||||
if u == nil {
|
||||
u = folio.URI(g.URI()).Parse()
|
||||
}
|
||||
uri := u.URL()
|
||||
g.Name = uri.Hostname()
|
||||
g.GID = LookupGIDString(uri.Hostname())
|
||||
if _, addGroupPathErr := exec.LookPath("addgroup"); addGroupPathErr == nil {
|
||||
g.GroupType = GroupTypeAddGroup
|
||||
}
|
||||
if _, pathErr := exec.LookPath("groupadd"); pathErr == nil {
|
||||
g.GroupType = GroupTypeGroupAdd
|
||||
}
|
||||
g.CreateCommand, g.ReadCommand, g.UpdateCommand, g.DeleteCommand = g.GroupType.NewCRUD()
|
||||
return g.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (g *Group) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (g *Group) SetResourceMapper(resources data.ResourceMapper) {
|
||||
g.Resources = resources
|
||||
}
|
||||
|
||||
func (g *Group) Clone() data.Resource {
|
||||
newg := &Group {
|
||||
Common: g.Common,
|
||||
Name: g.Name,
|
||||
GID: g.GID,
|
||||
State: g.State,
|
||||
GroupType: g.GroupType,
|
||||
}
|
||||
newg.CreateCommand, newg.ReadCommand, newg.UpdateCommand, newg.DeleteCommand = g.GroupType.NewCRUD()
|
||||
@ -109,40 +129,62 @@ func (g *Group) Notify(m *machine.EventMessage) {
|
||||
switch m.On {
|
||||
case machine.ENTERSTATEEVENT:
|
||||
switch m.Dest {
|
||||
case "start_stat":
|
||||
if statErr := g.ReadStat(); statErr == nil {
|
||||
if triggerErr := g.StateMachine().Trigger("exists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if triggerErr := g.StateMachine().Trigger("notexists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
case "start_read":
|
||||
if _,readErr := g.Read(ctx); readErr == nil {
|
||||
if triggerErr := g.StateMachine().Trigger("state_read"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
g.Common.State = "absent"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
g.Common.State = "absent"
|
||||
panic(readErr)
|
||||
}
|
||||
case "start_create":
|
||||
if e := g.Create(ctx); e == nil {
|
||||
if triggerErr := g.stater.Trigger("created"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
g.State = "absent"
|
||||
case "present":
|
||||
g.State = "present"
|
||||
g.Common.State = "absent"
|
||||
|
||||
|
||||
case "start_update":
|
||||
if updateErr := g.Update(ctx); updateErr == nil {
|
||||
if triggerErr := g.stater.Trigger("updated"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
g.Common.State = "absent"
|
||||
}
|
||||
} else {
|
||||
g.Common.State = "absent"
|
||||
panic(updateErr)
|
||||
}
|
||||
|
||||
case "absent":
|
||||
g.Common.State = "absent"
|
||||
case "present", "created", "read":
|
||||
g.Common.State = "present"
|
||||
}
|
||||
case machine.EXITSTATEEVENT:
|
||||
}
|
||||
}
|
||||
|
||||
func (g *Group) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == "group" {
|
||||
g.Name = resourceUri.Hostname()
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a group", ErrInvalidResourceURI, uri)
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (g *Group) URI() string {
|
||||
return fmt.Sprintf("group://%s", g.Name)
|
||||
}
|
||||
|
||||
func (g *Group) UseConfig(config data.ConfigurationValueGetter) {
|
||||
g.config = config
|
||||
}
|
||||
|
||||
func (g *Group) ResolveId(ctx context.Context) string {
|
||||
return LookupUIDString(g.Name)
|
||||
}
|
||||
@ -153,7 +195,7 @@ func (g *Group) Validate() error {
|
||||
|
||||
func (g *Group) Apply() error {
|
||||
ctx := context.Background()
|
||||
switch g.State {
|
||||
switch g.Common.State {
|
||||
case "present":
|
||||
_, NoGroupExists := LookupGID(g.Name)
|
||||
if NoGroupExists != nil {
|
||||
@ -198,10 +240,26 @@ func (g *Group) Create(ctx context.Context) (error) {
|
||||
return e
|
||||
}
|
||||
|
||||
func (g *Group) ReadStat() (err error) {
|
||||
if g.groupStatus == nil {
|
||||
if g.groupStatus, err = user.LookupGroup(g.Name); err != nil {
|
||||
g.Common.State = "absent"
|
||||
return err
|
||||
}
|
||||
}
|
||||
if len(g.groupStatus.Gid) < 1 {
|
||||
g.Common.State = "absent"
|
||||
return ErrResourceStateAbsent
|
||||
}
|
||||
g.GID = g.groupStatus.Gid
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
func (g *Group) Read(ctx context.Context) ([]byte, error) {
|
||||
exErr := g.ReadCommand.Extractor(nil, g)
|
||||
if exErr != nil {
|
||||
g.State = "absent"
|
||||
g.Common.State = "absent"
|
||||
}
|
||||
if yaml, yamlErr := yaml.Marshal(g); yamlErr != nil {
|
||||
return yaml, yamlErr
|
||||
@ -210,8 +268,9 @@ func (g *Group) Read(ctx context.Context) ([]byte, error) {
|
||||
}
|
||||
}
|
||||
|
||||
func (g *Group) Update(ctx context.Context) (error) {
|
||||
return g.Create(ctx)
|
||||
func (g *Group) Update(ctx context.Context) (err error) {
|
||||
_, err = g.UpdateCommand.Execute(g)
|
||||
return
|
||||
}
|
||||
|
||||
func (g *Group) Delete(ctx context.Context) (error) {
|
||||
@ -345,7 +404,7 @@ func NewGroupReadCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
g := target.(*Group)
|
||||
g.State = "absent"
|
||||
g.Common.State = "absent"
|
||||
var readGroup *user.Group
|
||||
var e error
|
||||
if g.Name != "" {
|
||||
@ -360,7 +419,7 @@ func NewGroupReadCommand() *command.Command {
|
||||
g.Name = readGroup.Name
|
||||
g.GID = readGroup.Gid
|
||||
if g.GID != "" {
|
||||
g.State = "present"
|
||||
g.Common.State = "present"
|
||||
}
|
||||
}
|
||||
return e
|
||||
@ -369,8 +428,18 @@ func NewGroupReadCommand() *command.Command {
|
||||
}
|
||||
|
||||
func NewGroupUpdateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "addgroup"
|
||||
c.FailOnError = false
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("{{ if .GID }}-g {{ .GID }}{{ end }}"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
return nil
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewGroupDelDeleteCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
@ -416,7 +485,7 @@ func NewReadGroupsCommand() *command.Command {
|
||||
g := (*Groups)[lineIndex]
|
||||
g.Name = groupRecord[0]
|
||||
g.GID = groupRecord[2]
|
||||
g.State = "present"
|
||||
g.Common.State = "present"
|
||||
g.GroupType = SystemGroupType
|
||||
lineIndex++
|
||||
}
|
||||
|
@ -60,16 +60,11 @@ func init() {
|
||||
}
|
||||
|
||||
func HTTPFactory(u *url.URL) data.Resource {
|
||||
var err error
|
||||
h := NewHTTP()
|
||||
|
||||
slog.Info("HTTP.Factory", "http", h, "url", u)
|
||||
if err = h.SetParsedURI(u); err != nil {
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(h); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if err = h.Open(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return h
|
||||
}
|
||||
@ -102,15 +97,34 @@ type HTTP struct {
|
||||
}
|
||||
|
||||
func NewHTTP() *HTTP {
|
||||
h := &HTTP{ client: &http.Client{}, Common: &Common{ includeQueryParamsInURI: true, resourceType: HTTPTypeName, SchemeCheck: func(scheme string) bool {
|
||||
h := &HTTP{ client: &http.Client{} }
|
||||
h.Common = NewCommon(HTTPTypeName, true)
|
||||
h.Common.SchemeCheck = h.SchemeCheck
|
||||
h.Common.NormalizePath = h.NormalizePath
|
||||
slog.Info("NewHTTP()", "http", h)
|
||||
return h
|
||||
}
|
||||
|
||||
func (h *HTTP) SchemeCheck(scheme string) bool {
|
||||
switch scheme {
|
||||
case "http", "https":
|
||||
return true
|
||||
}
|
||||
return false
|
||||
} } }
|
||||
slog.Info("NewHTTP()", "http", h)
|
||||
return h
|
||||
}
|
||||
|
||||
func (h *HTTP) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(h.URI()).Parse()
|
||||
}
|
||||
if err = h.SetParsedURI(u); err == nil {
|
||||
err = h.Open()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (h *HTTP) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (h *HTTP) SetResourceMapper(resources data.ResourceMapper) {
|
||||
@ -219,16 +233,9 @@ func (h *HTTP) URI() string {
|
||||
return h.Endpoint.String()
|
||||
}
|
||||
|
||||
func (h *HTTP) SetURI(uri string) (err error) {
|
||||
if err = h.Common.SetURI(uri); err == nil {
|
||||
h.Endpoint = h.Common.Uri
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (h *HTTP) SetParsedURI(u *url.URL) (err error) {
|
||||
func (h *HTTP) SetParsedURI(u data.URIParser) (err error) {
|
||||
if err = h.Common.SetParsedURI(u); err == nil {
|
||||
h.Endpoint = h.Common.Uri
|
||||
h.Endpoint = h.Common.URI()
|
||||
}
|
||||
return
|
||||
}
|
||||
@ -266,11 +273,13 @@ func (h *HTTP) ContentSourceRefStat() (info fs.FileInfo) {
|
||||
}
|
||||
|
||||
func (h *HTTP) ReadStat() (err error) {
|
||||
|
||||
if h.reader == nil {
|
||||
if err = h.OpenGetter(); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
var info fs.FileInfo
|
||||
info, err = h.reader.Stat()
|
||||
|
||||
@ -332,14 +341,14 @@ func (h *HTTP) Apply() error {
|
||||
|
||||
func (h *HTTP) Load(docData []byte, f codec.Format) (err error) {
|
||||
if err = f.StringDecoder(string(docData)).Decode(h); err == nil {
|
||||
err = h.Common.SetURI(string(h.Endpoint))
|
||||
err = h.Common.SetParsedURI(folio.URI(h.Endpoint).Parse())
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (h *HTTP) LoadReader(r io.ReadCloser, f codec.Format) (err error) {
|
||||
if err = f.Decoder(r).Decode(h); err == nil {
|
||||
err = h.Common.SetURI(string(h.Endpoint))
|
||||
err = h.Common.SetParsedURI(folio.URI(h.Endpoint).Parse())
|
||||
//err = h.setParsedURI(h.Endpoint)
|
||||
}
|
||||
return
|
||||
@ -347,7 +356,7 @@ func (h *HTTP) LoadReader(r io.ReadCloser, f codec.Format) (err error) {
|
||||
|
||||
func (h *HTTP) LoadString(docData string, f codec.Format) (err error) {
|
||||
if err = f.StringDecoder(docData).Decode(h); err == nil {
|
||||
err = h.Common.SetURI(string(h.Endpoint))
|
||||
err = h.Common.SetParsedURI(folio.URI(h.Endpoint).Parse())
|
||||
//err = h.setParsedURI(h.Endpoint)
|
||||
}
|
||||
return
|
||||
@ -358,7 +367,7 @@ func (h *HTTP) LoadDecl(yamlResourceDeclaration string) error {
|
||||
}
|
||||
|
||||
func (h *HTTP) ResolveId(ctx context.Context) string {
|
||||
_ = h.Common.SetURI(h.Endpoint.String())
|
||||
_ = h.Common.SetParsedURI(folio.URI(h.Endpoint).Parse())
|
||||
slog.Info("HTTP.ResolveId()", "uri", h.Endpoint.String())
|
||||
return h.Endpoint.String()
|
||||
}
|
||||
|
@ -30,20 +30,11 @@ const (
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"iptable"}, func(u *url.URL) data.Resource {
|
||||
i := NewIptable()
|
||||
i.Table = IptableName(u.Hostname())
|
||||
if len(u.Path) > 0 {
|
||||
fields := strings.FieldsFunc(u.Path, func(c rune) bool { return c == '/' })
|
||||
slog.Info("iptables factory", "iptable", i, "uri", u, "fields", fields, "number_fields", len(fields))
|
||||
if len(fields) > 0 {
|
||||
i.Chain = IptableChain(fields[0])
|
||||
if len(fields) < 3 {
|
||||
i.ResourceType = IptableTypeChain
|
||||
} else {
|
||||
i.ResourceType = IptableTypeRule
|
||||
id, _ := strconv.ParseUint(fields[1], 10, 32)
|
||||
i.Id = uint(id)
|
||||
}
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(i); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
i.CreateCommand, i.ReadCommand, i.UpdateCommand, i.DeleteCommand = i.ResourceType.NewCRUD()
|
||||
}
|
||||
return i
|
||||
@ -101,6 +92,8 @@ type ExtensionFlag struct {
|
||||
Value string `json:"value" yaml:"value"`
|
||||
}
|
||||
|
||||
type TargetFlag ExtensionFlag
|
||||
|
||||
type IptablePort uint16
|
||||
|
||||
type IptableType string
|
||||
@ -110,6 +103,8 @@ const (
|
||||
IptableTypeChain = "chain"
|
||||
)
|
||||
|
||||
type IptableRule string
|
||||
|
||||
var (
|
||||
ErrInvalidIptableName error = errors.New("The IptableName is not a valid table")
|
||||
)
|
||||
@ -119,7 +114,6 @@ var (
|
||||
type Iptable struct {
|
||||
*Common `json:",inline" yaml:",inline"`
|
||||
stater machine.Stater `json:"-" yaml:"-"`
|
||||
parsedURI *url.URL `json:"-" yaml:"-"`
|
||||
Id uint `json:"id,omitempty" yaml:"id,omitempty"`
|
||||
Table IptableName `json:"table" yaml:"table"`
|
||||
Chain IptableChain `json:"chain" yaml:"chain"`
|
||||
@ -133,15 +127,16 @@ type Iptable struct {
|
||||
Flags []ExtensionFlag `json:"extension_flags,omitempty" yaml:"extension_flags,omitempty"`
|
||||
Proto IptableProto `json:"proto,omitempty" yaml:"proto,omitempty"`
|
||||
Jump string `json:"jump,omitempty" yaml:"jump,omitempty"`
|
||||
TargetFlags []TargetFlag `json:"target_flags,omitempty" yaml:"target_flags,omitempty"`
|
||||
ChainLength uint `json:"-" yaml:"-"`
|
||||
|
||||
ResourceType IptableType `json:"resourcetype,omitempty" yaml:"resourcetype,omitempty"`
|
||||
CreateCommand *command.Command `yaml:"-" json:"-"`
|
||||
ReadCommand *command.Command `yaml:"-" json:"-"`
|
||||
ReadChainCommand *command.Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *command.Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *command.Command `yaml:"-" json:"-"`
|
||||
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `yaml:"-" json:"-"`
|
||||
}
|
||||
|
||||
@ -155,10 +150,39 @@ func (n IptableName) Validate() error {
|
||||
}
|
||||
}
|
||||
|
||||
func NewIptable() *Iptable {
|
||||
i := &Iptable{ ResourceType: IptableTypeRule, Common: &Common{ resourceType: IptableTypeName } }
|
||||
func NewIptable() (i *Iptable) {
|
||||
i = &Iptable{ ResourceType: IptableTypeRule }
|
||||
i.Common = NewCommon(IptableTypeName, false)
|
||||
i.Common.NormalizePath = i.NormalizePath
|
||||
i.CreateCommand, i.ReadCommand, i.UpdateCommand, i.DeleteCommand = i.ResourceType.NewCRUD()
|
||||
return i
|
||||
i.ReadChainCommand = NewIptableChainReadCommand()
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(i.URI()).Parse()
|
||||
}
|
||||
uri := u.URL()
|
||||
err = i.SetParsedURI(u)
|
||||
|
||||
i.Table = IptableName(uri.Hostname())
|
||||
if len(uri.Path) > 0 {
|
||||
fields := strings.FieldsFunc(uri.Path, func(c rune) bool { return c == '/' })
|
||||
slog.Info("iptables factory", "iptable", i, "uri", uri, "fields", fields, "number_fields", len(fields))
|
||||
if len(fields) > 0 {
|
||||
i.Chain = IptableChain(fields[0])
|
||||
if len(fields) < 3 {
|
||||
i.ResourceType = IptableTypeChain
|
||||
} else {
|
||||
i.ResourceType = IptableTypeRule
|
||||
id, _ := strconv.ParseUint(fields[1], 10, 32)
|
||||
i.SetId(uint(id))
|
||||
}
|
||||
}
|
||||
i.CreateCommand, i.ReadCommand, i.UpdateCommand, i.DeleteCommand = i.ResourceType.NewCRUD()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) SetResourceMapper(resources data.ResourceMapper) {
|
||||
@ -195,36 +219,109 @@ func (i *Iptable) Notify(m *machine.EventMessage) {
|
||||
switch m.On {
|
||||
case machine.ENTERSTATEEVENT:
|
||||
switch m.Dest {
|
||||
case "start_create":
|
||||
if e := i.Create(ctx); e == nil {
|
||||
if triggerErr := i.stater.Trigger("created"); triggerErr == nil {
|
||||
case "start_stat":
|
||||
if statErr := i.ReadStat(ctx); statErr == nil {
|
||||
if triggerErr := i.StateMachine().Trigger("exists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if triggerErr := i.StateMachine().Trigger("notexists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
case "start_read":
|
||||
if _,readErr := i.Read(ctx); readErr == nil {
|
||||
if triggerErr := i.stater.Trigger("state_read"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
i.Common.State = "absent"
|
||||
case "present":
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
i.Common.State = "absent"
|
||||
panic(readErr)
|
||||
}
|
||||
case "start_create":
|
||||
if createErr := i.Create(ctx); createErr == nil {
|
||||
if triggerErr := i.stater.Trigger("created"); triggerErr == nil {
|
||||
slog.Info("ContainerImage.Notify()", "created", i, "error", triggerErr)
|
||||
return
|
||||
} else {
|
||||
slog.Info("ContainerImage.Notify()", "created", i, "error", triggerErr)
|
||||
i.Common.State = "absent"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
i.Common.State = "absent"
|
||||
panic(createErr)
|
||||
}
|
||||
case "start_update":
|
||||
if createErr := i.Update(ctx); createErr == nil {
|
||||
if triggerErr := i.stater.Trigger("updated"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
i.Common.State = "absent"
|
||||
}
|
||||
} else {
|
||||
i.Common.State = "absent"
|
||||
panic(createErr)
|
||||
}
|
||||
case "start_delete":
|
||||
if deleteErr := i.Delete(ctx); deleteErr == nil {
|
||||
if triggerErr := i.stater.Trigger("deleted"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
panic(deleteErr)
|
||||
}
|
||||
case "present", "created", "read":
|
||||
i.Common.State = "present"
|
||||
case "absent":
|
||||
i.Common.State = "absent"
|
||||
}
|
||||
case machine.EXITSTATEEVENT:
|
||||
}
|
||||
}
|
||||
|
||||
// Set the chain ID and update the mapped URI
|
||||
func (i *Iptable) SetId(id uint) {
|
||||
if i.Id != id {
|
||||
uri := i.URI()
|
||||
i.Id = id
|
||||
decl, ok := i.Resources.Get(uri)
|
||||
if ok {
|
||||
i.Resources.Delete(uri)
|
||||
}
|
||||
i.Resources.Set(i.URI(), decl)
|
||||
}
|
||||
}
|
||||
|
||||
func (i *Iptable) URI() string {
|
||||
return fmt.Sprintf("iptable://%s/%s/%d", i.Table, i.Chain, i.Id)
|
||||
}
|
||||
|
||||
func (i *Iptable) SetURI(uri string) (err error) {
|
||||
i.parsedURI, err = url.Parse(uri)
|
||||
if err == nil {
|
||||
fields := strings.FieldsFunc(i.parsedURI.Path, func(c rune) bool { return c == '/' })
|
||||
|
||||
func (i *Iptable) SetParsedURI(uri data.URIParser) (err error) {
|
||||
if err = i.Common.SetParsedURI(uri); err == nil {
|
||||
err = i.setFieldsFromPath()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (i *Iptable) setFieldsFromPath() (err error) {
|
||||
fields := strings.FieldsFunc(i.Common.Path, func(c rune) bool { return c == '/' })
|
||||
fieldsLen := len(fields)
|
||||
if i.parsedURI.Scheme == "iptable" && fieldsLen > 0 {
|
||||
i.Table = IptableName(i.parsedURI.Hostname())
|
||||
if fieldsLen > 0 {
|
||||
i.Table = IptableName(fields[0])
|
||||
if err = i.Table.Validate(); err != nil {
|
||||
return err
|
||||
}
|
||||
i.Chain = IptableChain(fields[0])
|
||||
if fieldsLen < 2 {
|
||||
i.Chain = IptableChain(fields[1])
|
||||
if fieldsLen < 3 {
|
||||
i.ResourceType = IptableTypeChain
|
||||
} else {
|
||||
i.ResourceType = IptableTypeRule
|
||||
@ -232,16 +329,11 @@ func (i *Iptable) SetURI(uri string) (err error) {
|
||||
i.Id = uint(id)
|
||||
}
|
||||
} else {
|
||||
err = fmt.Errorf("%w: %s is not an iptable rule", ErrInvalidResourceURI, uri)
|
||||
}
|
||||
err = fmt.Errorf("%w: %s is not an iptable rule", ErrInvalidResourceURI, i.Common.URI())
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) UseConfig(config data.ConfigurationValueGetter) {
|
||||
i.config = config
|
||||
}
|
||||
|
||||
func (i *Iptable) Validate() error {
|
||||
s := NewSchema(i.Type())
|
||||
jsonDoc, jsonErr := i.JSON()
|
||||
@ -316,6 +408,46 @@ func (i *Iptable) ResolveId(ctx context.Context) string {
|
||||
return fmt.Sprintf("%d", i.Id)
|
||||
}
|
||||
|
||||
func (f *ExtensionFlag) Match(name string, value string) bool {
|
||||
start := 0
|
||||
if name[1] == '-' {
|
||||
start = 2
|
||||
} else if name[0] == '-' {
|
||||
start = 1
|
||||
}
|
||||
return f.Name == name[start:] && f.Value == value
|
||||
}
|
||||
|
||||
func (i *Iptable) HasExtensionFlag(name string) bool {
|
||||
optName := strings.Trim(name, "-")
|
||||
for _, ext := range i.Flags {
|
||||
if ext.Name == optName {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (i *Iptable) HasTargetFlag(name string) bool {
|
||||
optName := strings.Trim(name, "-")
|
||||
for _, ext := range i.TargetFlags {
|
||||
if ext.Name == optName {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (f *TargetFlag) Match(name string, value string) bool {
|
||||
start := 0
|
||||
if name[1] == '-' {
|
||||
start = 2
|
||||
} else if name[0] == '-' {
|
||||
start = 1
|
||||
}
|
||||
return f.Name == name[start:] && f.Value == value
|
||||
}
|
||||
|
||||
func (i *Iptable) SetFlagValue(opt, value string) bool {
|
||||
switch opt {
|
||||
case "-i":
|
||||
@ -354,7 +486,11 @@ func (i *Iptable) SetFlagValue(opt, value string) bool {
|
||||
return true
|
||||
default:
|
||||
if opt[0] == '-' {
|
||||
if len(i.Jump) > 0 {
|
||||
i.TargetFlags = append(i.TargetFlags, TargetFlag{ Name: strings.Trim(opt, "-"), Value: strings.TrimSpace(value)})
|
||||
} else {
|
||||
i.Flags = append(i.Flags, ExtensionFlag{ Name: strings.Trim(opt, "-"), Value: strings.TrimSpace(value)})
|
||||
}
|
||||
return true
|
||||
}
|
||||
}
|
||||
@ -383,8 +519,14 @@ func (i *Iptable) GetFlagValue(opt string) any {
|
||||
return strconv.Itoa(int(i.Sport))
|
||||
default:
|
||||
if opt[0] == '-' {
|
||||
slog.Info("Iptable.GetFlagValue()", "opt", opt, "iptable", i)
|
||||
if i.HasExtensionFlag(opt) {
|
||||
return i.Flags
|
||||
}
|
||||
if i.HasTargetFlag(opt) {
|
||||
return i.TargetFlags
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@ -407,9 +549,11 @@ func (i *Iptable) SetRule(flags []string) (assigned bool) {
|
||||
|
||||
func (i *Iptable) MatchRule(flags []string) (match bool) {
|
||||
match = true
|
||||
next:
|
||||
for index, flag := range flags {
|
||||
if flag[0] == '-' {
|
||||
value := flags[index + 1]
|
||||
slog.Info("Iptable.MatchRule()", "flag", flag, "value", value)
|
||||
switch v := i.GetFlagValue(flag).(type) {
|
||||
case []string:
|
||||
for _,element := range v {
|
||||
@ -417,48 +561,57 @@ func (i *Iptable) MatchRule(flags []string) (match bool) {
|
||||
continue
|
||||
}
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
case []ExtensionFlag:
|
||||
for _,element := range v {
|
||||
if element.Name == flag && element.Value == value {
|
||||
continue
|
||||
if element.Match(flag, value) {
|
||||
continue next
|
||||
}
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
case []TargetFlag:
|
||||
for _,element := range v {
|
||||
if element.Match(flag, value) {
|
||||
continue next
|
||||
}
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
case IptableCIDR:
|
||||
if v == IptableCIDR(value) {
|
||||
continue
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
case IptableName:
|
||||
if v == IptableName(value) {
|
||||
continue
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
case IptableChain:
|
||||
if v == IptableChain(value) {
|
||||
continue
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
default:
|
||||
if v.(string) == value {
|
||||
continue
|
||||
}
|
||||
slog.Info("Iptable.MatchRule() - FAILED", "flag", flag, "value", v)
|
||||
match = false
|
||||
}
|
||||
}
|
||||
}
|
||||
slog.Info("Iptable.MatchRule()", "flags", flags, "match", match)
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) ReadChainLength() error {
|
||||
c := command.NewCommand()
|
||||
c.Path = "iptables"
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("-S"),
|
||||
command.CommandArg("{{ .Chain }}"),
|
||||
}
|
||||
output,err := c.Execute(i)
|
||||
output,err := i.ReadChainCommand.Execute(i)
|
||||
if err == nil {
|
||||
linesCount := strings.Count(string(output), "\n")
|
||||
if linesCount > 0 {
|
||||
@ -470,13 +623,16 @@ func (i *Iptable) ReadChainLength() error {
|
||||
return err
|
||||
}
|
||||
|
||||
func (i *Iptable) Create(ctx context.Context) error {
|
||||
func (i *Iptable) Create(ctx context.Context) (err error) {
|
||||
slog.Info("Iptable.Create()", "iptable", i)
|
||||
if i.Id > 0 {
|
||||
if lenErr := i.ReadChainLength(); lenErr != nil {
|
||||
return lenErr
|
||||
}
|
||||
}
|
||||
_, err := i.CreateCommand.Execute(i)
|
||||
|
||||
_, err = i.CreateCommand.Execute(i)
|
||||
slog.Info("Iptable.Create()", "err", err, "iptable", i, "createcommand", i.CreateCommand)
|
||||
//slog.Info("IptableChain Create()", "err", err, "errstr", err.Error(), "iptable", i, "createcommand", i.CreateCommand)
|
||||
// TODO add Command status/error handler rather than using the read extractor
|
||||
if i.CreateCommand.Extractor != nil {
|
||||
@ -484,7 +640,20 @@ func (i *Iptable) Create(ctx context.Context) error {
|
||||
return i.CreateCommand.Extractor([]byte(err.Error()), i)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
return err
|
||||
}
|
||||
|
||||
func (i *Iptable) ReadStat(ctx context.Context) (err error) {
|
||||
if i.ReadCommand.Exists() {
|
||||
var out []byte
|
||||
if out, err = i.ReadCommand.Execute(i); err == nil {
|
||||
err = i.ReadCommand.Extractor(out, i)
|
||||
}
|
||||
}
|
||||
if i.Id == 0 {
|
||||
return ErrResourceStateAbsent
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (i *Iptable) Read(ctx context.Context) ([]byte, error) {
|
||||
@ -503,8 +672,15 @@ func (i *Iptable) Update(ctx context.Context) error {
|
||||
return i.Create(ctx)
|
||||
}
|
||||
|
||||
func (i *Iptable) Delete(ctx context.Context) error {
|
||||
return nil
|
||||
func (i *Iptable) Delete(ctx context.Context) (err error) {
|
||||
if i.Id < 1 {
|
||||
return fmt.Errorf("Failed to find rule to delete")
|
||||
}
|
||||
var out []byte
|
||||
if out, err = i.DeleteCommand.Execute(i); err == nil {
|
||||
err = i.DeleteCommand.Extractor(out, i)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (i *Iptable) Type() string { return "iptable" }
|
||||
@ -552,9 +728,7 @@ func NewIptableCreateCommand() *command.Command {
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("-t"),
|
||||
command.CommandArg("{{ .Table }}"),
|
||||
command.CommandArg("{{ if le .Id .ChainLength }}-R{{ else }}-A{{ end }}"),
|
||||
command.CommandArg("{{ .Chain }}"),
|
||||
command.CommandArg("{{ if le .Id .ChainLength }}{{ .Id }}{{ end }}"),
|
||||
command.CommandArg("{{ if and (le .Id .ChainLength) (gt .Id 0) }}-R {{ .Chain }} {{ .Id }}{{ else }}-A {{ .Chain }}{{ end }}"),
|
||||
command.CommandArg("{{ if .In }}-i {{ .In }}{{ else if .Out }}-o {{ .Out }}{{ end }}"),
|
||||
command.CommandArg("{{ range .Match }}-m {{ . }} {{- end }}"),
|
||||
command.CommandArg("{{ if .Source }}-s {{ .Source }}{{ end }}"),
|
||||
@ -562,8 +736,9 @@ func NewIptableCreateCommand() *command.Command {
|
||||
command.CommandArg("{{ if .Destination }}-d {{ .Destination }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Dport }}--dport {{ .Dport }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Proto }}-p {{ .Proto }}{{ end }}"),
|
||||
command.CommandArg("{{ range .Flags }} --{{ .Name }} {{ .Value }} {{ end }}"),
|
||||
command.CommandArg("{{ range .Flags -}} --{{ .Name }} {{ .Value }} {{- end }}"),
|
||||
command.CommandArg("{{ if .Jump }}-j {{ .Jump }}{{ end }}"),
|
||||
command.CommandArg("{{ range .TargetFlags -}} --{{ .Name }} {{ .Value }}{{- end }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
@ -584,7 +759,7 @@ func IptableExtractRule(lineNumber uint, ruleLine string, target *Iptable) (stat
|
||||
}
|
||||
} else {
|
||||
if target.MatchRule(flags) {
|
||||
target.Id = lineNumber
|
||||
target.SetId(lineNumber)
|
||||
state = "present"
|
||||
err = nil
|
||||
}
|
||||
@ -661,7 +836,7 @@ func NewIptableReadChainCommand() *command.Command {
|
||||
}
|
||||
for lineIndex, line := range lines[1:] {
|
||||
i := (*IptableChainRules)[lineIndex]
|
||||
i.Id = uint(lineIndex + 1)
|
||||
i.SetId(uint(lineIndex + 1))
|
||||
ruleFields := strings.Split(strings.TrimSpace(line), " ")
|
||||
if ruleFields[0] == "-A" {
|
||||
flags := ruleFields[2:]
|
||||
@ -682,7 +857,16 @@ func NewIptableUpdateCommand() *command.Command {
|
||||
}
|
||||
|
||||
func NewIptableDeleteCommand() *command.Command {
|
||||
return nil
|
||||
c := command.NewCommand()
|
||||
c.Path = "iptables"
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("-t"),
|
||||
command.CommandArg("{{ .Table }}"),
|
||||
command.CommandArg("-D"),
|
||||
command.CommandArg("{{ .Chain }}"),
|
||||
command.CommandArg("{{ .Id }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewIptableChainCreateCommand() *command.Command {
|
||||
@ -766,7 +950,7 @@ func RuleExtractorMatchFlags(out []byte, target any) (err error) {
|
||||
slog.Info("RuleExtractorMatchFlags()", "flags", flags, "ipt", ipt)
|
||||
err = nil
|
||||
ipt.Common.State = "present"
|
||||
ipt.Id = uint(linesIndex)
|
||||
ipt.SetId(uint(linesIndex))
|
||||
return
|
||||
}
|
||||
}
|
||||
@ -838,6 +1022,14 @@ func NewIptableChainUpdateCommand() *command.Command {
|
||||
}
|
||||
|
||||
func NewIptableChainDeleteCommand() *command.Command {
|
||||
return nil
|
||||
c := command.NewCommand()
|
||||
c.Path = "iptables"
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("-t"),
|
||||
command.CommandArg("{{ .Table }}"),
|
||||
command.CommandArg("-X"),
|
||||
command.CommandArg("{{ .Chain }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
|
@ -20,6 +20,7 @@ _ "syscall"
|
||||
"testing"
|
||||
_ "time"
|
||||
"decl/internal/command"
|
||||
"decl/internal/data"
|
||||
)
|
||||
|
||||
func TestNewIptableResource(t *testing.T) {
|
||||
@ -77,10 +78,59 @@ func TestReadIptable(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestCreateIptable(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
testRule := NewIptable()
|
||||
assert.NotNil(t, testRule)
|
||||
|
||||
declarationAttributes := `
|
||||
table: "filter"
|
||||
id: 5
|
||||
chain: "INPUT"
|
||||
source: "192.168.0.0/24"
|
||||
destination: "192.168.0.1"
|
||||
jump: "ACCEPT"
|
||||
state: present
|
||||
`
|
||||
|
||||
m := &MockCommand{
|
||||
Executor: func(value any) ([]byte, error) {
|
||||
return nil, nil
|
||||
},
|
||||
Extractor: func(output []byte, target any) error {
|
||||
testRule.Table = "filter"
|
||||
testRule.Id = 3
|
||||
testRule.Chain = "INPUT"
|
||||
testRule.In = "eth0"
|
||||
testRule.Source = "192.168.0.0/24"
|
||||
testRule.State = "present"
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
mockReadChain := &MockCommand{
|
||||
Executor: func(value any) ([]byte, error) {
|
||||
return []byte(`
|
||||
-P INPUT ACCEPT
|
||||
-A INPUT -j LIBVIRT_INP
|
||||
`), nil
|
||||
},
|
||||
}
|
||||
|
||||
e := testRule.LoadDecl(declarationAttributes)
|
||||
assert.Nil(t, e)
|
||||
|
||||
testRule.ReadChainCommand = (*command.Command)(mockReadChain)
|
||||
testRule.ReadCommand = (*command.Command)(m)
|
||||
testRule.CreateCommand = (*command.Command)(m)
|
||||
|
||||
assert.Nil(t, testRule.Create(ctx))
|
||||
|
||||
assert.Equal(t, uint(2), testRule.ChainLength)
|
||||
_, err := testRule.Read(ctx)
|
||||
assert.Nil(t, err)
|
||||
|
||||
//assert.Equal(t, uint(3), testRule.ChainLength)
|
||||
assert.Equal(t, uint(3), testRule.Id)
|
||||
}
|
||||
|
||||
func TestIptableSetFlagValue(t *testing.T) {
|
||||
@ -117,6 +167,7 @@ func TestIptableRuleExtractorById(t *testing.T) {
|
||||
|
||||
func TestIptableRuleExtractorByFlags(t *testing.T) {
|
||||
ipt := NewIptable()
|
||||
ipt.Resources = data.NewResourceMapper()
|
||||
assert.NotNil(t, ipt)
|
||||
ipt.Table = IptableName("filter")
|
||||
ipt.Chain = IptableChain("FOO")
|
||||
|
@ -18,6 +18,7 @@ _ "strconv"
|
||||
"decl/internal/codec"
|
||||
"decl/internal/data"
|
||||
"decl/internal/folio"
|
||||
"decl/internal/command"
|
||||
)
|
||||
|
||||
const (
|
||||
@ -25,52 +26,17 @@ const (
|
||||
)
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"route"}, func(u *url.URL) data.Resource {
|
||||
n := NewNetworkRoute()
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"route"}, func(u *url.URL) (n data.Resource) {
|
||||
n = NewNetworkRoute()
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(n); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return n
|
||||
})
|
||||
}
|
||||
|
||||
/*
|
||||
|
||||
ROUTE := NODE_SPEC [ INFO_SPEC ]
|
||||
NODE_SPEC := [ TYPE ] PREFIX [ tos TOS ]
|
||||
[ table TABLE_ID ] [ proto RTPROTO ]
|
||||
[ scope SCOPE ] [ metric METRIC ]
|
||||
[ ttl-propagate { enabled | disabled } ]
|
||||
INFO_SPEC := { NH | nhid ID } OPTIONS FLAGS [ nexthop NH ]...
|
||||
|
||||
NH := [ encap ENCAPTYPE ENCAPHDR ] [ via [ FAMILY ] ADDRESS ]
|
||||
[ dev STRING ] [ weight NUMBER ] NHFLAGS
|
||||
FAMILY := [ inet | inet6 | mpls | bridge | link ]
|
||||
OPTIONS := FLAGS [ mtu NUMBER ] [ advmss NUMBER ] [ as [ to ] ADDRESS ]
|
||||
[ rtt TIME ] [ rttvar TIME ] [ reordering NUMBER ]
|
||||
[ window NUMBER ] [ cwnd NUMBER ] [ initcwnd NUMBER ]
|
||||
[ ssthresh NUMBER ] [ realms REALM ] [ src ADDRESS ]
|
||||
[ rto_min TIME ] [ hoplimit NUMBER ] [ initrwnd NUMBER ]
|
||||
[ features FEATURES ] [ quickack BOOL ] [ congctl NAME ]
|
||||
[ pref PREF ] [ expires TIME ] [ fastopen_no_cookie BOOL ]
|
||||
NHFLAGS := [ onlink | pervasive ]
|
||||
PREF := [ low | medium | high ]
|
||||
TIME := NUMBER[s|ms]
|
||||
BOOL := [1|0]
|
||||
FEATURES := ecn
|
||||
ENCAPTYPE := [ mpls | ip | ip6 | seg6 | seg6local | rpl | ioam6 ]
|
||||
ENCAPHDR := [ MPLSLABEL | SEG6HDR | SEG6LOCAL | IOAM6HDR ]
|
||||
SEG6HDR := [ mode SEGMODE ] segs ADDR1,ADDRi,ADDRn [hmac HMACKEYID] [cleanup]
|
||||
SEGMODE := [ encap | inline ]
|
||||
SEG6LOCAL := action ACTION [ OPTIONS ] [ count ]
|
||||
ACTION := { End | End.X | End.T | End.DX2 | End.DX6 | End.DX4 |
|
||||
End.DT6 | End.DT4 | End.DT46 | End.B6 | End.B6.Encaps |
|
||||
End.BM | End.S | End.AS | End.AM | End.BPF }
|
||||
OPTIONS := OPTION [ OPTIONS ]
|
||||
OPTION := { srh SEG6HDR | nh4 ADDR | nh6 ADDR | iif DEV | oif DEV |
|
||||
table TABLEID | vrftable TABLEID | endpoint PROGNAME }
|
||||
IOAM6HDR := trace prealloc type IOAM6_TRACE_TYPE ns IOAM6_NAMESPACE size IOAM6_TRACE_SIZE
|
||||
ROUTE_GET_FLAGS := [ fibmatch ]
|
||||
|
||||
*/
|
||||
|
||||
type NetworkRouteType string
|
||||
|
||||
const (
|
||||
@ -126,21 +92,33 @@ type NetworkRoute struct {
|
||||
Scope NetworkRouteScope `json:"scope" yaml:"scope"`
|
||||
Proto NetworkRouteProto `json:"proto" yaml:"proto"`
|
||||
|
||||
CreateCommand *Command `yaml:"-" json:"-"`
|
||||
ReadCommand *Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *Command `yaml:"-" json:"-"`
|
||||
CreateCommand *command.Command `yaml:"-" json:"-"`
|
||||
ReadCommand *command.Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *command.Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *command.Command `yaml:"-" json:"-"`
|
||||
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewNetworkRoute() *NetworkRoute {
|
||||
n := &NetworkRoute{Rtid: NetworkRouteTableMain, Common: &Common{ resourceType: NetworkRouteTypeName } }
|
||||
n := &NetworkRoute{Rtid: NetworkRouteTableMain}
|
||||
n.Common = NewCommon(NetworkRouteTypeName, false)
|
||||
n.Common.NormalizePath = n.NormalizePath
|
||||
n.CreateCommand, n.ReadCommand, n.UpdateCommand, n.DeleteCommand = n.NewCRUD()
|
||||
return n
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(n.URI()).Parse()
|
||||
}
|
||||
return n.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) SetResourceMapper(resources data.ResourceMapper) {
|
||||
n.Resources = resources
|
||||
}
|
||||
@ -210,14 +188,6 @@ func (n *NetworkRoute) URI() string {
|
||||
return fmt.Sprintf("route://%s", n.Id)
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) SetURI(uri string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) UseConfig(config data.ConfigurationValueGetter) {
|
||||
n.config = config
|
||||
}
|
||||
|
||||
func (n *NetworkRoute) Validate() error {
|
||||
return fmt.Errorf("failed")
|
||||
}
|
||||
@ -472,41 +442,41 @@ func (n *NetworkRoute) UnmarshalJSON(data []byte) error {
|
||||
}
|
||||
|
||||
|
||||
func (n *NetworkRoute) NewCRUD() (create *Command, read *Command, update *Command, del *Command) {
|
||||
func (n *NetworkRoute) NewCRUD() (create *command.Command, read *command.Command, update *command.Command, del *command.Command) {
|
||||
return NewNetworkRouteCreateCommand(), NewNetworkRouteReadCommand(), NewNetworkRouteUpdateCommand(), NewNetworkRouteDeleteCommand()
|
||||
}
|
||||
|
||||
func NewNetworkRouteCreateCommand() *Command {
|
||||
c := NewCommand()
|
||||
func NewNetworkRouteCreateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "ip"
|
||||
c.Args = []CommandArg{
|
||||
CommandArg("route"),
|
||||
CommandArg("add"),
|
||||
CommandArg("{{ if .To }}to {{ .To }}{{ end }}"),
|
||||
CommandArg("{{ if .Rtid }}table {{ .Rtid }}{{ end }}"),
|
||||
CommandArg("{{ if .Gateway }}via {{ .Gateway }}{{ end }}"),
|
||||
CommandArg("{{ if .Proto }}protocol {{ .Proto }}{{ end }}"),
|
||||
CommandArg("{{ if .Scope }}scope {{ .Scope }}{{ end }}"),
|
||||
CommandArg("{{ if .RouteType }}type {{ .RouteType }}{{ end }}"),
|
||||
CommandArg("{{ if .Interface }}dev {{ .Interface }}{{ end }}"),
|
||||
CommandArg("{{ if .Metric }}metric {{ .Metric }}{{ end }}"),
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("route"),
|
||||
command.CommandArg("add"),
|
||||
command.CommandArg("{{ if .To }}to {{ .To }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Rtid }}table {{ .Rtid }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Gateway }}via {{ .Gateway }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Proto }}protocol {{ .Proto }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Scope }}scope {{ .Scope }}{{ end }}"),
|
||||
command.CommandArg("{{ if .RouteType }}type {{ .RouteType }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Interface }}dev {{ .Interface }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Metric }}metric {{ .Metric }}{{ end }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewNetworkRouteReadCommand() *Command {
|
||||
c := NewCommand()
|
||||
func NewNetworkRouteReadCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "ip"
|
||||
c.Args = []CommandArg{
|
||||
CommandArg("route"),
|
||||
CommandArg("show"),
|
||||
CommandArg("{{ if .To }}to {{ .To }}{{ end }}"),
|
||||
CommandArg("{{ if .Rtid }}table {{ .Rtid }}{{ end }}"),
|
||||
CommandArg("{{ if .Gateway }}via {{ .Gateway }}{{ end }}"),
|
||||
CommandArg("{{ if .Proto }}protocol {{ .Proto }}{{ end }}"),
|
||||
CommandArg("{{ if .Scope }}scope {{ .Scope }}{{ end }}"),
|
||||
CommandArg("{{ if .RouteType }}type {{ .RouteType }}{{ end }}"),
|
||||
CommandArg("{{ if .Interface }}dev {{ .Interface }}{{ end }}"),
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("route"),
|
||||
command.CommandArg("show"),
|
||||
command.CommandArg("{{ if .To }}to {{ .To }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Rtid }}table {{ .Rtid }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Gateway }}via {{ .Gateway }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Proto }}protocol {{ .Proto }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Scope }}scope {{ .Scope }}{{ end }}"),
|
||||
command.CommandArg("{{ if .RouteType }}type {{ .RouteType }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Interface }}dev {{ .Interface }}{{ end }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
n := target.(*NetworkRoute)
|
||||
@ -529,16 +499,16 @@ func NewNetworkRouteReadCommand() *Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func NewNetworkRouteUpdateCommand() *Command {
|
||||
c := NewCommand()
|
||||
func NewNetworkRouteUpdateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "ip"
|
||||
c.Args = []CommandArg{
|
||||
CommandArg("del"),
|
||||
CommandArg("{{ .Name }}"),
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("del"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewNetworkRouteDeleteCommand() *Command {
|
||||
func NewNetworkRouteDeleteCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
@ -23,6 +23,9 @@ import (
|
||||
"decl/internal/tempdir"
|
||||
)
|
||||
|
||||
const (
|
||||
PackageTypeName TypeName = "package"
|
||||
)
|
||||
var (
|
||||
PackageTempDir tempdir.Path = "jx_package_resource"
|
||||
)
|
||||
@ -50,6 +53,7 @@ var SupportedPackageTypes []PackageType = []PackageType{PackageTypeApk, PackageT
|
||||
var SystemPackageType PackageType = FindSystemPackageType()
|
||||
|
||||
type Package struct {
|
||||
*Common `yaml:",inline" json:",inline"`
|
||||
stater machine.Stater `yaml:"-" json:"-"`
|
||||
Source string `yaml:"source,omitempty" json:"source,omitempty"`
|
||||
Name string `yaml:"name" json:"name"`
|
||||
@ -62,18 +66,12 @@ type Package struct {
|
||||
ReadCommand *command.Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *command.Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *command.Command `yaml:"-" json:"-"`
|
||||
// state attributes
|
||||
State string `yaml:"state,omitempty" json:"state,omitempty"`
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `yaml:"-" json:"-"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"package", string(PackageTypeApk), string(PackageTypeApt), string(PackageTypeDeb), string(PackageTypeDnf), string(PackageTypeRpm), string(PackageTypePip), string(PackageTypeYum)}, func(u *url.URL) data.Resource {
|
||||
p := NewPackage()
|
||||
e := p.SetParsedURI(u)
|
||||
slog.Info("PackageFactory SetParsedURI()", "error", e)
|
||||
return p
|
||||
return ConstructNewPackage(u)
|
||||
})
|
||||
}
|
||||
|
||||
@ -88,7 +86,41 @@ func FindSystemPackageType() PackageType {
|
||||
}
|
||||
|
||||
func NewPackage() *Package {
|
||||
return &Package{ PackageType: SystemPackageType }
|
||||
return &Package{
|
||||
Common: NewCommon(PackageTypeName, true),
|
||||
PackageType: SystemPackageType,
|
||||
}
|
||||
}
|
||||
|
||||
func ConstructNewPackage(uri *url.URL) (p *Package) {
|
||||
p = NewPackage()
|
||||
if uri != nil {
|
||||
if err := folio.CastParsedURI(uri).ConstructResource(p); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (p *Package) Init(u data.URIParser) error {
|
||||
if u == nil {
|
||||
u = folio.URI(p.URI()).Parse()
|
||||
}
|
||||
uri := u.URL()
|
||||
|
||||
p.Name = filepath.Join(uri.Hostname(), uri.Path)
|
||||
p.Version = uri.Query().Get("version")
|
||||
if p.Version == "" {
|
||||
p.Version = "latest"
|
||||
}
|
||||
indicatedPackageType := PackageType(uri.Query().Get("type"))
|
||||
if indicatedPackageType.Validate() != nil {
|
||||
p.PackageType = SystemPackageType
|
||||
}
|
||||
|
||||
p.CreateCommand, p.ReadCommand, p.UpdateCommand, p.DeleteCommand = p.PackageType.NewCRUD()
|
||||
|
||||
return p.SetParsedURI(u)
|
||||
}
|
||||
|
||||
func (p *Package) SetResourceMapper(resources data.ResourceMapper) {
|
||||
@ -97,11 +129,11 @@ func (p *Package) SetResourceMapper(resources data.ResourceMapper) {
|
||||
|
||||
func (p *Package) Clone() data.Resource {
|
||||
newp := &Package {
|
||||
Common: p.Common.Clone(),
|
||||
Name: p.Name,
|
||||
Required: p.Required,
|
||||
Version: p.Version,
|
||||
PackageType: p.PackageType,
|
||||
State: p.State,
|
||||
}
|
||||
newp.CreateCommand, newp.ReadCommand, newp.UpdateCommand, newp.DeleteCommand = newp.PackageType.NewCRUD()
|
||||
return newp
|
||||
@ -135,11 +167,11 @@ func (p *Package) Notify(m *machine.EventMessage) {
|
||||
if triggerErr := p.StateMachine().Trigger("state_read"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
if ! errors.Is(readErr, ErrResourceStateAbsent) {
|
||||
panic(readErr)
|
||||
}
|
||||
@ -164,17 +196,17 @@ func (p *Package) Notify(m *machine.EventMessage) {
|
||||
if triggerErr := p.StateMachine().Trigger("deleted"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
panic(deleteErr)
|
||||
}
|
||||
case "absent":
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
case "present", "created", "updated", "read":
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
}
|
||||
case machine.EXITSTATEEVENT:
|
||||
}
|
||||
@ -192,36 +224,25 @@ func (p *Package) URI() string {
|
||||
}
|
||||
|
||||
|
||||
func (p *Package) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
e = p.SetParsedURI(resourceUri)
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (p *Package) SetParsedURI(uri *url.URL) (err error) {
|
||||
if uri.Scheme == "package" {
|
||||
p.Name = filepath.Join(uri.Hostname(), uri.Path)
|
||||
p.Version = uri.Query().Get("version")
|
||||
func (p *Package) SetParsedURI(uri data.URIParser) (err error) {
|
||||
u := uri.URL()
|
||||
if u.Scheme == "package" {
|
||||
p.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
p.Version = u.Query().Get("version")
|
||||
if p.Version == "" {
|
||||
p.Version = "latest"
|
||||
}
|
||||
indicatedPackageType := PackageType(uri.Query().Get("type"))
|
||||
indicatedPackageType := PackageType(u.Query().Get("type"))
|
||||
if indicatedPackageType.Validate() != nil {
|
||||
p.PackageType = SystemPackageType
|
||||
}
|
||||
} else {
|
||||
err = fmt.Errorf("%w: %s is not a package resource ", ErrInvalidResourceURI, uri.String())
|
||||
err = fmt.Errorf("%w: %s is not a package resource ", ErrInvalidResourceURI, u.String())
|
||||
}
|
||||
p.CreateCommand, p.ReadCommand, p.UpdateCommand, p.DeleteCommand = p.PackageType.NewCRUD()
|
||||
return
|
||||
}
|
||||
|
||||
func (p *Package) UseConfig(config data.ConfigurationValueGetter) {
|
||||
p.config = config
|
||||
}
|
||||
|
||||
func (p *Package) JSON() ([]byte, error) {
|
||||
return json.Marshal(p)
|
||||
}
|
||||
@ -331,6 +352,9 @@ func (p *Package) LoadDecl(yamlResourceDeclaration string) error {
|
||||
func (p *Package) Type() string { return "package" }
|
||||
|
||||
func (p *Package) Read(ctx context.Context) (resourceYaml []byte, err error) {
|
||||
if p.Version == "latest" {
|
||||
p.Version = ""
|
||||
}
|
||||
if p.ReadCommand.Exists() {
|
||||
var out []byte
|
||||
out, err = p.ReadCommand.Execute(p)
|
||||
@ -495,10 +519,10 @@ func NewApkReadCommand() *command.Command {
|
||||
if packageName == p.Name {
|
||||
p.Name = packageName
|
||||
p.Version = packageVersion
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
} else {
|
||||
slog.Info("NewApkReadCommand().Extrctor() mismatch", "name", p.Name, "parsed", packageName)
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@ -554,7 +578,7 @@ func NewApkReadPackagesCommand() *command.Command {
|
||||
packageName := strings.Join(packageFields[0:numberOfFields - 2], "-")
|
||||
packageVersion := strings.Join(packageFields[numberOfFields - 2:numberOfFields - 1], "-")
|
||||
p.Name = packageName
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.Version = packageVersion
|
||||
}
|
||||
}
|
||||
@ -595,16 +619,16 @@ func NewAptReadCommand() *command.Command {
|
||||
switch key {
|
||||
case "Package":
|
||||
if value != p.Name {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
return nil
|
||||
}
|
||||
case "Status":
|
||||
statusFields := strings.SplitN(value, " ", 3)
|
||||
if len(statusFields) > 1 {
|
||||
if statusFields[2] == "installed" {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
} else {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
}
|
||||
}
|
||||
case "Version":
|
||||
@ -671,7 +695,7 @@ func NewAptReadPackagesCommand() *command.Command {
|
||||
packageName := packageFields[0]
|
||||
packageVersion := installedPackage[1]
|
||||
p.Name = packageName
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.Version = packageVersion
|
||||
p.PackageType = PackageTypeApt
|
||||
}
|
||||
@ -711,16 +735,16 @@ func NewDebReadCommand() *command.Command {
|
||||
switch key {
|
||||
case "Package":
|
||||
if value != p.Name {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
return nil
|
||||
}
|
||||
case "Status":
|
||||
statusFields := strings.SplitN(value, " ", 3)
|
||||
if len(statusFields) > 1 {
|
||||
if statusFields[2] == "installed" {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
} else {
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
}
|
||||
}
|
||||
case "Version":
|
||||
@ -787,7 +811,7 @@ func NewDebReadPackagesCommand() *command.Command {
|
||||
p.Version = packageVersionFields[0]
|
||||
}
|
||||
p.Name = packageName
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.PackageType = PackageTypeDeb
|
||||
lineIndex++
|
||||
}
|
||||
@ -833,7 +857,7 @@ func NewDnfReadCommand() *command.Command {
|
||||
packageName := strings.TrimSpace(strings.Join(packageNameField[0:lenName - 1], "."))
|
||||
|
||||
if packageName == p.Name {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
packageVersionField := strings.Split(fields[1], ":")
|
||||
if len(packageVersionField) > 1 {
|
||||
//packageEpoch := strings.TrimSpace(packageVersionField[0])
|
||||
@ -846,7 +870,7 @@ func NewDnfReadCommand() *command.Command {
|
||||
slog.Info("DnfReadCommaond.Extract()", "package", packageName, "package", p)
|
||||
}
|
||||
}
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
slog.Info("Extract()", "package", p)
|
||||
return nil
|
||||
}
|
||||
@ -907,7 +931,7 @@ func NewDnfReadPackagesCommand() *command.Command {
|
||||
p.Version = packageVersionFields[0]
|
||||
}
|
||||
p.Name = packageName
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.PackageType = PackageTypeDnf
|
||||
lineIndex++
|
||||
}
|
||||
@ -949,13 +973,13 @@ func NewRpmReadCommand() *command.Command {
|
||||
packageVersion := strings.Join(packageFields[numberOfFields - 2:numberOfFields - 1], "-")
|
||||
slog.Info("Package[RPM].Extract()", "name", packageName, "version", packageVersion, "package", p)
|
||||
if packageName == p.Name {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.Version = packageVersion
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
slog.Info("Extract()", "package", p)
|
||||
return nil
|
||||
}
|
||||
@ -1013,13 +1037,13 @@ func NewPipReadCommand() *command.Command {
|
||||
packageName := packageFields[0]
|
||||
packageVersion := packageFields[1]
|
||||
if packageName == p.Name {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
p.Version = packageVersion
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
slog.Info("Extract()", "package", p)
|
||||
return nil
|
||||
}
|
||||
@ -1083,7 +1107,7 @@ func NewYumReadCommand() *command.Command {
|
||||
//packageArch := strings.TrimSpace(packageNameField[1])
|
||||
|
||||
if packageName == p.Name {
|
||||
p.State = "present"
|
||||
p.Common.State = "present"
|
||||
packageVersionField := strings.Split(fields[1], ":")
|
||||
//packageEpoch := strings.TrimSpace(packageVersionField[0])
|
||||
packageVersion := strings.TrimSpace(packageVersionField[1])
|
||||
@ -1091,7 +1115,7 @@ func NewYumReadCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
p.State = "absent"
|
||||
p.Common.State = "absent"
|
||||
slog.Info("Extract()", "package", p)
|
||||
return nil
|
||||
}
|
||||
|
@ -4,19 +4,12 @@ package resource
|
||||
|
||||
import (
|
||||
"context"
|
||||
_ "encoding/json"
|
||||
"fmt"
|
||||
"github.com/stretchr/testify/assert"
|
||||
_ "gopkg.in/yaml.v3"
|
||||
_ "io"
|
||||
"log/slog"
|
||||
_ "net/http"
|
||||
_ "net/http/httptest"
|
||||
_ "net/url"
|
||||
_ "os"
|
||||
_ "strings"
|
||||
"testing"
|
||||
"decl/internal/command"
|
||||
"decl/internal/folio"
|
||||
)
|
||||
|
||||
func TestNewPackageResource(t *testing.T) {
|
||||
@ -354,6 +347,8 @@ type: %s
|
||||
assert.NotNil(t, p)
|
||||
loadErr := p.LoadDecl(decl)
|
||||
assert.Nil(t, loadErr)
|
||||
assert.Nil(t, p.Init(nil))
|
||||
|
||||
p.ReadCommand = SystemPackageType.NewReadCommand()
|
||||
/*
|
||||
p.ReadCommand.Executor = func(value any) ([]byte, error) {
|
||||
@ -363,6 +358,7 @@ type: %s
|
||||
p.ResolveId(ctx)
|
||||
yaml, readErr := p.Read(ctx)
|
||||
assert.ErrorIs(t, readErr, ErrResourceStateAbsent)
|
||||
slog.Info("TestReadPackageError", "package", p, "common", p.Common, "yaml", yaml)
|
||||
assert.YAMLEq(t, expected, string(yaml))
|
||||
slog.Info("read()", "yaml", yaml)
|
||||
assert.Equal(t, "", p.Version)
|
||||
@ -375,7 +371,9 @@ func TestCreatePackage(t *testing.T) {
|
||||
func TestPackageSetURI(t *testing.T) {
|
||||
p := NewPackage()
|
||||
assert.NotNil(t, p)
|
||||
e := p.SetURI("package://" + "12345_key?type=apk")
|
||||
|
||||
uri := folio.URI("package://" + "12345_key?type=apk").Parse()
|
||||
e := p.Init(uri)
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, "package", p.Type())
|
||||
assert.Equal(t, "12345_key", p.Name)
|
||||
|
@ -44,12 +44,17 @@ var ErrPKIInvalidEncodingType error = errors.New("Invalid EncodingType")
|
||||
var ErrPKIFailedDecodingPemBlock error = errors.New("Failed decoding pem block")
|
||||
|
||||
func init() {
|
||||
ResourceTypes.Register([]string{"pki"}, func(u *url.URL) data.Resource {
|
||||
ResourceTypes.Register([]string{"pki"}, func(u *url.URL) (data.Resource) {
|
||||
k := NewPKI()
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(k); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
ref := folio.ResourceReference(filepath.Join(u.Hostname(), u.Path))
|
||||
if len(ref) > 0 {
|
||||
k.PrivateKeyRef = ref
|
||||
}
|
||||
}
|
||||
return k
|
||||
})
|
||||
}
|
||||
@ -83,8 +88,6 @@ type PKI struct {
|
||||
|
||||
Bits int `json:"bits" yaml:"bits"`
|
||||
EncodingType EncodingType `json:"type" yaml:"type"`
|
||||
//State string `json:"state,omitempty" yaml:"state,omitempty"`
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
@ -102,7 +105,6 @@ func (k *PKI) Clone() data.Resource {
|
||||
return &PKI {
|
||||
Common: k.Common.Clone(),
|
||||
EncodingType: k.EncodingType,
|
||||
//State: k.State,
|
||||
}
|
||||
}
|
||||
|
||||
@ -119,6 +121,16 @@ func (k *PKI) Notify(m *machine.EventMessage) {
|
||||
switch m.On {
|
||||
case machine.ENTERSTATEEVENT:
|
||||
switch m.Dest {
|
||||
case "start_stat":
|
||||
if statErr := k.ReadStat(); statErr == nil {
|
||||
if triggerErr := k.StateMachine().Trigger("exists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if triggerErr := k.StateMachine().Trigger("notexists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
case "start_read":
|
||||
if _,readErr := k.Read(ctx); readErr == nil {
|
||||
if triggerErr := k.StateMachine().Trigger("state_read"); triggerErr == nil {
|
||||
@ -175,22 +187,6 @@ func (k *PKI) URI() string {
|
||||
return fmt.Sprintf("pki://%s", filepath.Join(u.Hostname(), u.RequestURI()))
|
||||
}
|
||||
|
||||
func (k *PKI) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == "pki" {
|
||||
k.PrivateKeyRef = folio.ResourceReference(fmt.Sprintf("pki://%s", filepath.Join(resourceUri.Hostname(), resourceUri.Path)))
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a cert", ErrInvalidResourceURI, uri)
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (k *PKI) UseConfig(config data.ConfigurationValueGetter) {
|
||||
k.config = config
|
||||
}
|
||||
|
||||
func (k *PKI) Validate() error {
|
||||
return fmt.Errorf("failed")
|
||||
}
|
||||
@ -228,6 +224,27 @@ func (k *PKI) LoadDecl(yamlResourceDeclaration string) error {
|
||||
return k.LoadString(yamlResourceDeclaration, codec.FormatYaml)
|
||||
}
|
||||
|
||||
func (k *PKI) ReadStat() (err error) {
|
||||
var resourcesErr []string
|
||||
if ! k.PrivateKeyRef.Exists() {
|
||||
resourcesErr = append(resourcesErr, string(k.PrivateKeyRef))
|
||||
}
|
||||
|
||||
if ! k.PublicKeyRef.Exists() {
|
||||
resourcesErr = append(resourcesErr, string(k.PublicKeyRef))
|
||||
}
|
||||
|
||||
if ! k.CertificateRef.Exists() {
|
||||
resourcesErr = append(resourcesErr, string(k.CertificateRef))
|
||||
}
|
||||
|
||||
if len(resourcesErr) > 0 {
|
||||
err = fmt.Errorf("PKI resources missing: %s", strings.Join(resourcesErr, ","))
|
||||
// k.State = "absent"
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (k *PKI) ResolveId(ctx context.Context) string {
|
||||
return string(k.PrivateKeyRef)
|
||||
}
|
||||
@ -463,62 +480,6 @@ func (k *PKI) Update(ctx context.Context) (err error) {
|
||||
|
||||
func (k *PKI) Type() string { return "pki" }
|
||||
|
||||
/*
|
||||
func (k *PKI) UnmarshalValue(value *PKIContent) error {
|
||||
d.Type = value.Type
|
||||
d.Transition = value.Transition
|
||||
d.Config = value.Config
|
||||
newResource, resourceErr := ResourceTypes.New(fmt.Sprintf("%s://", value.Type))
|
||||
if resourceErr != nil {
|
||||
slog.Info("Declaration.UnmarshalValue", "value", value, "error", resourceErr)
|
||||
return resourceErr
|
||||
}
|
||||
d.Attributes = newResource
|
||||
return nil
|
||||
}
|
||||
|
||||
func (k *PKI) UnmarshalYAML(value *yaml.Node) error {
|
||||
t := &DeclarationType{}
|
||||
if unmarshalResourceTypeErr := value.Decode(t); unmarshalResourceTypeErr != nil {
|
||||
return unmarshalResourceTypeErr
|
||||
}
|
||||
|
||||
if err := d.UnmarshalValue(t); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
|
||||
if err := d.UnmarshalValue(t); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
resourceAttrs := struct {
|
||||
Attributes Resource `json:"attributes"`
|
||||
}{Attributes: d.Attributes}
|
||||
if unmarshalAttributesErr := json.Unmarshal(data, &resourceAttrs); unmarshalAttributesErr != nil {
|
||||
return unmarshalAttributesErr
|
||||
}
|
||||
return nil
|
||||
}
|
||||
*/
|
||||
|
||||
func (t *EncodingType) UnmarshalValue(value string) error {
|
||||
switch value {
|
||||
case string(EncodingTypePem):
|
||||
|
@ -19,7 +19,6 @@ _ "os"
|
||||
"strings"
|
||||
"testing"
|
||||
"path/filepath"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
type TestResourceMapper func(key string) (data.Declaration, bool)
|
||||
@ -36,6 +35,9 @@ func (rm TestResourceMapper) Has(key string) (ok bool) {
|
||||
func (rm TestResourceMapper) Set(key string, value data.Declaration) {
|
||||
}
|
||||
|
||||
func (rm TestResourceMapper) Delete(key string) {
|
||||
}
|
||||
|
||||
type StringContentReadWriter func() (any, error)
|
||||
|
||||
func (s StringContentReadWriter) ContentWriterStream() (*transport.Writer, error) {
|
||||
@ -56,7 +58,7 @@ func (s StringContentReadWriter) LoadString(docData string, f codec.Format) (err
|
||||
func (s StringContentReadWriter) LoadDecl(yamlResourceDeclaration string) error { return nil }
|
||||
func (s StringContentReadWriter) ResolveId(ctx context.Context) (string) { return "" }
|
||||
func (s StringContentReadWriter) SetURI(uri string) (error) { return nil }
|
||||
func (s StringContentReadWriter) SetParsedURI(uri *url.URL) (error) { return nil }
|
||||
func (s StringContentReadWriter) SetParsedURI(uri data.URIParser) (error) { return nil }
|
||||
func (s StringContentReadWriter) URI() (string) { return "" }
|
||||
func (s StringContentReadWriter) Validate() (error) { return nil }
|
||||
func (s StringContentReadWriter) ResourceType() data.TypeName { return "" }
|
||||
|
@ -20,67 +20,6 @@ var (
|
||||
)
|
||||
|
||||
type ResourceReference string
|
||||
/*
|
||||
type ResourceSelector func(r *Declaration) bool
|
||||
|
||||
type Resource interface {
|
||||
Type() string
|
||||
StateMachine() machine.Stater
|
||||
URI() string
|
||||
SetURI(string) error
|
||||
UseConfig(config ConfigurationValueGetter)
|
||||
ResolveId(context.Context) string
|
||||
ResourceLoader
|
||||
StateTransformer
|
||||
ResourceReader
|
||||
ResourceValidator
|
||||
Clone() Resource
|
||||
SetResourceMapper(resources ResourceMapper)
|
||||
}
|
||||
|
||||
type ContentReader interface {
|
||||
ContentReaderStream() (*transport.Reader, error)
|
||||
}
|
||||
|
||||
type ContentWriter interface {
|
||||
ContentWriterStream() (*transport.Writer, error)
|
||||
}
|
||||
|
||||
type ContentReadWriter interface {
|
||||
ContentReader
|
||||
ContentWriter
|
||||
}
|
||||
|
||||
type ResourceValidator interface {
|
||||
Validate() error
|
||||
}
|
||||
|
||||
type ResourceCreator interface {
|
||||
Create(context.Context) error
|
||||
}
|
||||
|
||||
type ResourceReader interface {
|
||||
Read(context.Context) ([]byte, error)
|
||||
}
|
||||
|
||||
type ResourceUpdater interface {
|
||||
Update(context.Context) error
|
||||
}
|
||||
|
||||
type ResourceDeleter interface {
|
||||
Delete(context.Context) error
|
||||
}
|
||||
|
||||
type ResourceDecoder struct {
|
||||
}
|
||||
|
||||
type ResourceCrudder struct {
|
||||
ResourceCreator
|
||||
ResourceReader
|
||||
ResourceUpdater
|
||||
ResourceDeleter
|
||||
}
|
||||
*/
|
||||
|
||||
func NewResource(uri string) data.Resource {
|
||||
r, e := ResourceTypes.New(uri)
|
||||
@ -90,50 +29,17 @@ func NewResource(uri string) data.Resource {
|
||||
return nil
|
||||
}
|
||||
|
||||
/*
|
||||
|
||||
// Return a Content ReadWriter for the resource referred to.
|
||||
func (r ResourceReference) Lookup(look data.ResourceMapper) data.ContentReadWriter {
|
||||
slog.Info("ResourceReference.Lookup()", "resourcereference", r, "resourcemapper", look)
|
||||
if look != nil {
|
||||
if v,ok := look.Get(string(r)); ok {
|
||||
return v.(data.ContentReadWriter)
|
||||
func ResourceConstructor(res data.Resource, uri data.URIParser) (err error) {
|
||||
if uri != nil {
|
||||
return uri.ConstructResource(res)
|
||||
} else {
|
||||
if ri, ok := res.(data.ResourceInitializer); ok {
|
||||
return ri.Init(uri)
|
||||
}
|
||||
}
|
||||
return r
|
||||
return
|
||||
}
|
||||
|
||||
func (r ResourceReference) Dereference(look data.ResourceMapper) data.Resource {
|
||||
slog.Info("ResourceReference.Dereference()", "resourcereference", r, "resourcemapper", look)
|
||||
if look != nil {
|
||||
if v,ok := look.Get(string(r)); ok {
|
||||
return v.(*Declaration).Attributes
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r ResourceReference) Parse() *url.URL {
|
||||
u, e := url.Parse(string(r))
|
||||
if e == nil {
|
||||
return u
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r ResourceReference) Exists() bool {
|
||||
return transport.ExistsURI(string(r))
|
||||
}
|
||||
|
||||
func (r ResourceReference) ContentReaderStream() (*transport.Reader, error) {
|
||||
return transport.NewReaderURI(string(r))
|
||||
}
|
||||
|
||||
func (r ResourceReference) ContentWriterStream() (*transport.Writer, error) {
|
||||
return transport.NewWriterURI(string(r))
|
||||
}
|
||||
*/
|
||||
|
||||
func StorageMachine(sub machine.Subscriber) machine.Stater {
|
||||
// start_destroy -> absent -> start_create -> present -> start_destroy
|
||||
stater := machine.New("unknown")
|
||||
|
@ -58,6 +58,8 @@ func TestSchemaValidateJSON(t *testing.T) {
|
||||
testFile := NewFile()
|
||||
e := testFile.LoadDecl(decl)
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, file, testFile.ResolveId(ctx))
|
||||
|
||||
fileApplyErr := testFile.Apply()
|
||||
assert.Nil(t, fileApplyErr)
|
||||
|
||||
@ -71,8 +73,10 @@ func TestSchemaValidateJSON(t *testing.T) {
|
||||
assert.NotNil(t, f)
|
||||
|
||||
f.Path = file
|
||||
assert.Nil(t, f.Init(nil))
|
||||
r, e := f.Read(ctx)
|
||||
assert.Nil(t, e)
|
||||
|
||||
assert.Equal(t, ProcessTestUserName, f.Owner)
|
||||
|
||||
info, statErr := os.Stat(file)
|
||||
|
@ -6,7 +6,7 @@ package resource
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
_ "log/slog"
|
||||
"log/slog"
|
||||
"net/url"
|
||||
"path/filepath"
|
||||
"io"
|
||||
@ -14,8 +14,15 @@ _ "log/slog"
|
||||
"gitea.rosskeen.house/rosskeen.house/machine"
|
||||
"decl/internal/codec"
|
||||
"decl/internal/data"
|
||||
"decl/internal/command"
|
||||
"decl/internal/folio"
|
||||
"encoding/json"
|
||||
"strings"
|
||||
"errors"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrUnsupportedServiceManagerType error = errors.New("Unsupported service manager")
|
||||
)
|
||||
|
||||
const (
|
||||
@ -29,32 +36,72 @@ const (
|
||||
ServiceManagerTypeSysV ServiceManagerType = "sysv"
|
||||
)
|
||||
|
||||
const (
|
||||
SysVStatusRunning int = 0
|
||||
SysVStatusStopped int = 1
|
||||
SysVStatusUnknown int = 2
|
||||
SysVStatusMissing int = 3
|
||||
)
|
||||
|
||||
var (
|
||||
SupportedServiceManagerTypes []ServiceManagerType = []ServiceManagerType{ServiceManagerTypeSystemd, ServiceManagerTypeSysV}
|
||||
SystemServiceManagerType ServiceManagerType = FindSystemServiceManagerType()
|
||||
)
|
||||
|
||||
type Service struct {
|
||||
*Common `yaml:",inline" json:",inline"`
|
||||
stater machine.Stater `yaml:"-" json:"-"`
|
||||
Name string `json:"name" yaml:"name"`
|
||||
ServiceManagerType ServiceManagerType `json:"servicemanager,omitempty" yaml:"servicemanager,omitempty"`
|
||||
|
||||
CreateCommand *Command `yaml:"-" json:"-"`
|
||||
ReadCommand *Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *Command `yaml:"-" json:"-"`
|
||||
CreateCommand *command.Command `yaml:"-" json:"-"`
|
||||
ReadCommand *command.Command `yaml:"-" json:"-"`
|
||||
UpdateCommand *command.Command `yaml:"-" json:"-"`
|
||||
DeleteCommand *command.Command `yaml:"-" json:"-"`
|
||||
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `yaml:"-" json:"-"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
ResourceTypes.Register([]string{"service"}, func(u *url.URL) data.Resource {
|
||||
s := NewService()
|
||||
s.Name = filepath.Join(u.Hostname(), u.Path)
|
||||
s.CreateCommand, s.ReadCommand, s.UpdateCommand, s.DeleteCommand = s.ServiceManagerType.NewCRUD()
|
||||
if err := folio.CastParsedURI(u).ConstructResource(s); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return s
|
||||
})
|
||||
}
|
||||
|
||||
func NewService() *Service {
|
||||
return &Service{ ServiceManagerType: ServiceManagerTypeSystemd, Common: &Common{ resourceType: ServiceTypeName } }
|
||||
func FindSystemServiceManagerType() ServiceManagerType {
|
||||
for _, servicemanagerType := range SupportedServiceManagerTypes {
|
||||
if c := servicemanagerType.NewReadCommand(); c != nil && c.Exists() {
|
||||
return servicemanagerType
|
||||
}
|
||||
}
|
||||
return ServiceManagerTypeSystemd
|
||||
}
|
||||
|
||||
func NewService() (s *Service) {
|
||||
s = &Service{ ServiceManagerType: SystemServiceManagerType }
|
||||
s.Common = NewCommon(ServiceTypeName, false)
|
||||
s.Common.NormalizePath = s.NormalizePath
|
||||
s.CreateCommand, s.ReadCommand, s.UpdateCommand, s.DeleteCommand = s.ServiceManagerType.NewCRUD()
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) Init(u data.URIParser) (err error) {
|
||||
if u == nil {
|
||||
u = folio.URI(s.URI()).Parse()
|
||||
}
|
||||
uri := u.URL()
|
||||
err = s.SetParsedURI(u)
|
||||
s.Name = filepath.Join(uri.Hostname(), uri.Path)
|
||||
s.CreateCommand, s.ReadCommand, s.UpdateCommand, s.DeleteCommand = s.ServiceManagerType.NewCRUD()
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) StateMachine() machine.Stater {
|
||||
@ -69,6 +116,16 @@ func (s *Service) Notify(m *machine.EventMessage) {
|
||||
switch m.On {
|
||||
case machine.ENTERSTATEEVENT:
|
||||
switch m.Dest {
|
||||
case "start_stat":
|
||||
if statErr := s.ReadStat(); statErr == nil {
|
||||
if triggerErr := s.StateMachine().Trigger("exists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if triggerErr := s.StateMachine().Trigger("notexists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
case "start_create":
|
||||
if e := s.Create(ctx); e == nil {
|
||||
if triggerErr := s.stater.Trigger("created"); triggerErr == nil {
|
||||
@ -76,10 +133,24 @@ func (s *Service) Notify(m *machine.EventMessage) {
|
||||
}
|
||||
}
|
||||
s.Common.State = "absent"
|
||||
case "created":
|
||||
case "start_read":
|
||||
if _,readErr := s.Read(ctx); readErr == nil {
|
||||
if triggerErr := s.stater.Trigger("state_read"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
s.Common.State = "absent"
|
||||
panic(triggerErr)
|
||||
}
|
||||
} else {
|
||||
s.Common.State = "absent"
|
||||
panic(readErr)
|
||||
}
|
||||
case "present", "created", "read":
|
||||
s.Common.State = "present"
|
||||
case "running":
|
||||
s.Common.State = "running"
|
||||
case "absent":
|
||||
s.Common.State = "absent"
|
||||
}
|
||||
case machine.EXITSTATEEVENT:
|
||||
}
|
||||
@ -89,20 +160,20 @@ func (s *Service) URI() string {
|
||||
return fmt.Sprintf("service://%s", s.Name)
|
||||
}
|
||||
|
||||
func (s *Service) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == s.Type() {
|
||||
s.Name = filepath.Join(resourceUri.Hostname(), resourceUri.RequestURI())
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a %s", ErrInvalidResourceURI, uri, s.Type())
|
||||
func (s *Service) SetParsedURI(uri data.URIParser) (err error) {
|
||||
if err = s.Common.SetParsedURI(uri); err == nil {
|
||||
err = s.setFieldsFromPath()
|
||||
}
|
||||
}
|
||||
return e
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) UseConfig(config data.ConfigurationValueGetter) {
|
||||
s.config = config
|
||||
func (s *Service) setFieldsFromPath() (err error) {
|
||||
if len(s.Common.Path) > 0 {
|
||||
s.Name = s.Common.Path
|
||||
} else {
|
||||
err = fmt.Errorf("%w: %s is not an iptable rule", ErrInvalidResourceURI, s.Common.URI())
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) JSON() ([]byte, error) {
|
||||
@ -167,7 +238,7 @@ func (s *Service) UnmarshalYAML(value *yaml.Node) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *ServiceManagerType) NewCRUD() (create *Command, read *Command, update *Command, del *Command) {
|
||||
func (s *ServiceManagerType) NewCRUD() (create *command.Command, read *command.Command, update *command.Command, del *command.Command) {
|
||||
switch *s {
|
||||
case ServiceManagerTypeSystemd:
|
||||
return NewSystemdCreateCommand(), NewSystemdReadCommand(), NewSystemdUpdateCommand(), NewSystemdDeleteCommand()
|
||||
@ -178,14 +249,52 @@ func (s *ServiceManagerType) NewCRUD() (create *Command, read *Command, update *
|
||||
return nil, nil, nil, nil
|
||||
}
|
||||
|
||||
|
||||
func (s *Service) Create(ctx context.Context) error {
|
||||
func (s *ServiceManagerType) NewReadCommand() (read *command.Command) {
|
||||
switch *s {
|
||||
case ServiceManagerTypeSystemd:
|
||||
return NewSystemdReadCommand()
|
||||
case ServiceManagerTypeSysV:
|
||||
return NewSysVReadCommand()
|
||||
default:
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) Read(ctx context.Context) ([]byte, error) {
|
||||
func (s *Service) Create(ctx context.Context) (err error) {
|
||||
var out []byte
|
||||
out, err = s.CreateCommand.Execute(s)
|
||||
slog.Info("Service.Create()", "out", out, "error", err)
|
||||
return
|
||||
}
|
||||
|
||||
return yaml.Marshal(s)
|
||||
func (s *Service) ReadStat() (err error) {
|
||||
if s.ReadCommand.Exists() {
|
||||
_, err = s.ReadCommand.Execute(s)
|
||||
} else {
|
||||
err = ErrUnsupportedServiceManagerType
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) Read(ctx context.Context) (resourceYaml []byte, err error) {
|
||||
if s.ReadCommand.Exists() {
|
||||
var out []byte
|
||||
out, err = s.ReadCommand.Execute(s)
|
||||
if err == nil {
|
||||
err = s.ReadCommand.Extractor(out, s)
|
||||
} else {
|
||||
err = fmt.Errorf("%w - %w", ErrResourceStateAbsent, err)
|
||||
}
|
||||
slog.Info("Service.Read()", "service", s, "error", err)
|
||||
} else {
|
||||
err = ErrUnsupportedServiceManagerType
|
||||
}
|
||||
var yamlErr error
|
||||
resourceYaml, yamlErr = yaml.Marshal(s)
|
||||
if err == nil {
|
||||
err = yamlErr
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Service) Update(ctx context.Context) error {
|
||||
@ -202,22 +311,22 @@ func (s *Service) ResolveId(ctx context.Context) string {
|
||||
return ""
|
||||
}
|
||||
|
||||
func NewSystemdCreateCommand() *Command {
|
||||
c := NewCommand()
|
||||
func NewSystemdCreateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "systemctl"
|
||||
c.Args = []CommandArg{
|
||||
CommandArg("enable"),
|
||||
CommandArg("{{ .Name }}"),
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("enable"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewSystemdReadCommand() *Command {
|
||||
c := NewCommand()
|
||||
func NewSystemdReadCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "systemctl"
|
||||
c.Args = []CommandArg{
|
||||
CommandArg("show"),
|
||||
CommandArg("{{ .Name }}"),
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("show"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
s := target.(*Service)
|
||||
@ -253,26 +362,63 @@ func NewSystemdReadCommand() *Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func NewSystemdUpdateCommand() *Command {
|
||||
func NewSystemdUpdateCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSystemdDeleteCommand() *Command {
|
||||
func NewSystemdDeleteCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSysVCreateCommand() *Command {
|
||||
func NewSysVCreateCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSysVReadCommand() *Command {
|
||||
func NewSysVReadCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "service"
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("status"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) (err error) {
|
||||
s := target.(*Service)
|
||||
/*
|
||||
serviceStatus := strings.Split(string(out), "\n")
|
||||
for _, statusLine := range(serviceStatus) {
|
||||
if len(statusLine) > 1 {
|
||||
statusFields := strings.Fields(statusLine)
|
||||
}
|
||||
}
|
||||
*/
|
||||
if len(out) < 1 {
|
||||
if err = s.stater.Trigger("notexist"); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
switch c.ExitCode {
|
||||
case SysVStatusRunning:
|
||||
if err = s.stater.Trigger("running"); err != nil {
|
||||
return
|
||||
}
|
||||
case SysVStatusStopped:
|
||||
if err = s.stater.Trigger("created"); err != nil {
|
||||
return
|
||||
}
|
||||
case SysVStatusUnknown, SysVStatusMissing:
|
||||
if err = s.stater.Trigger("notexist"); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSysVUpdateCommand() *Command {
|
||||
func NewSysVUpdateCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSysVDeleteCommand() *Command {
|
||||
func NewSysVDeleteCommand() *command.Command {
|
||||
return nil
|
||||
}
|
||||
|
@ -4,10 +4,10 @@ package resource
|
||||
|
||||
import (
|
||||
"context"
|
||||
_ "decl/tests/mocks"
|
||||
_ "fmt"
|
||||
"decl/internal/command"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
"decl/internal/folio"
|
||||
)
|
||||
|
||||
func TestNewServiceResource(t *testing.T) {
|
||||
@ -17,20 +17,35 @@ func TestNewServiceResource(t *testing.T) {
|
||||
|
||||
func TestUriServiceResource(t *testing.T) {
|
||||
c := NewService()
|
||||
assert.Nil(t, c.SetURI("service://ssh"))
|
||||
uri := folio.URI("service://ssh").Parse()
|
||||
|
||||
assert.Nil(t, c.Init(uri))
|
||||
assert.Equal(t, "ssh", c.Name)
|
||||
}
|
||||
|
||||
func TestReadServiceResource(t *testing.T) {
|
||||
|
||||
s := NewService()
|
||||
s.Name = "ssh"
|
||||
|
||||
yamlResult := `
|
||||
name: "ssh"
|
||||
servicemanager: "systemd"
|
||||
state: "present"
|
||||
`
|
||||
c := NewService()
|
||||
c.Name = "ssh"
|
||||
c.State = "present"
|
||||
yamlData, err := c.Read(context.Background())
|
||||
m := &MockCommand{
|
||||
CommandExists: func() error { return nil },
|
||||
Executor: func(value any) ([]byte, error) {
|
||||
return nil, nil
|
||||
},
|
||||
Extractor: func(output []byte, target any) error {
|
||||
s.Common.State = "present"
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
s.ReadCommand = (*command.Command)(m)
|
||||
yamlData, err := s.Read(context.Background())
|
||||
assert.Nil(t, err)
|
||||
assert.YAMLEq(t, yamlResult, string(yamlData))
|
||||
}
|
||||
|
@ -28,13 +28,15 @@ type UserType string
|
||||
|
||||
const (
|
||||
UserTypeName TypeName = "user"
|
||||
UserTypeAddUser UserType = "adduser"
|
||||
UserTypeUserAdd UserType = "useradd"
|
||||
UserTypeBusyBox UserType = "busybox"
|
||||
UserTypeShadow UserType = "shadow"
|
||||
)
|
||||
|
||||
var ErrUnsupportedUserType error = errors.New("The UserType is not supported on this system")
|
||||
var ErrInvalidUserType error = errors.New("invalid UserType value")
|
||||
|
||||
var SupportedUserTypes []UserType = []UserType{UserTypeShadow, UserTypeBusyBox}
|
||||
|
||||
var SystemUserType UserType = FindSystemUserType()
|
||||
|
||||
type User struct {
|
||||
@ -44,48 +46,65 @@ type User struct {
|
||||
UID string `json:"uid,omitempty" yaml:"uid,omitempty"`
|
||||
Group string `json:"group,omitempty" yaml:"group,omitempty"`
|
||||
Groups []string `json:"groups,omitempty" yaml:"groups,omitempty"`
|
||||
AppendGroups bool `json:"appendgroups,omitempty" yaml:"appendgroups,omitempty"`
|
||||
Gecos string `json:"gecos,omitempty" yaml:"gecos,omitempty"`
|
||||
Home string `json:"home" yaml:"home"`
|
||||
CreateHome bool `json:"createhome,omitempty" yaml:"createhome,omitempty"`
|
||||
Shell string `json:"shell,omitempty" yaml:"shell,omitempty"`
|
||||
UserType UserType `json:"-" yaml:"-"`
|
||||
UserType UserType `json:"type,omitempty" yaml:"type,omitempty"`
|
||||
|
||||
userStatus *user.User `json:"-" yaml:"-"`
|
||||
groupStatus *user.Group `json:"-" yaml:"-"`
|
||||
|
||||
CreateCommand *command.Command `json:"-" yaml:"-"`
|
||||
ReadCommand *command.Command `json:"-" yaml:"-"`
|
||||
UpdateCommand *command.Command `json:"-" yaml:"-"`
|
||||
DeleteCommand *command.Command `json:"-" yaml:"-"`
|
||||
config data.ConfigurationValueGetter
|
||||
Resources data.ResourceMapper `json:"-" yaml:"-"`
|
||||
}
|
||||
|
||||
func NewUser() *User {
|
||||
return &User{ CreateHome: true, Common: &Common{ resourceType: UserTypeName } }
|
||||
u := &User{ CreateHome: true, AppendGroups: true, UserType: SystemUserType }
|
||||
u.Common = NewCommon(UserTypeName, false)
|
||||
u.Common.NormalizePath = u.NormalizePath
|
||||
return u
|
||||
}
|
||||
|
||||
func init() {
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"user"}, func(u *url.URL) data.Resource {
|
||||
user := NewUser()
|
||||
user.Name = u.Hostname()
|
||||
user.UID = LookupUIDString(u.Hostname())
|
||||
if _, addUserPathErr := exec.LookPath("adduser"); addUserPathErr == nil {
|
||||
user.UserType = UserTypeAddUser
|
||||
folio.DocumentRegistry.ResourceTypes.Register([]string{"user"}, func(u *url.URL) (user data.Resource) {
|
||||
user = NewUser()
|
||||
if u != nil {
|
||||
if err := folio.CastParsedURI(u).ConstructResource(user); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if _, pathErr := exec.LookPath("useradd"); pathErr == nil {
|
||||
user.UserType = UserTypeUserAdd
|
||||
}
|
||||
user.CreateCommand, user.ReadCommand, user.UpdateCommand, user.DeleteCommand = user.UserType.NewCRUD()
|
||||
return user
|
||||
return
|
||||
})
|
||||
}
|
||||
|
||||
func FindSystemUserType() UserType {
|
||||
for _, userType := range []UserType{UserTypeAddUser, UserTypeUserAdd} {
|
||||
for _, userType := range SupportedUserTypes {
|
||||
c := userType.NewCreateCommand()
|
||||
if c.Exists() {
|
||||
return userType
|
||||
}
|
||||
}
|
||||
return UserTypeAddUser
|
||||
return UserTypeShadow
|
||||
}
|
||||
|
||||
func (u *User) Init(uri data.URIParser) error {
|
||||
if uri == nil {
|
||||
uri = folio.URI(u.URI()).Parse()
|
||||
} else {
|
||||
u.Name = uri.URL().Hostname()
|
||||
}
|
||||
u.UID = LookupUIDString(uri.URL().Hostname())
|
||||
u.CreateCommand, u.ReadCommand, u.UpdateCommand, u.DeleteCommand = u.UserType.NewCRUD()
|
||||
return u.SetParsedURI(uri)
|
||||
}
|
||||
|
||||
func (u *User) NormalizePath() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *User) SetResourceMapper(resources data.ResourceMapper) {
|
||||
@ -121,6 +140,16 @@ func (u *User) Notify(m *machine.EventMessage) {
|
||||
switch m.On {
|
||||
case machine.ENTERSTATEEVENT:
|
||||
switch m.Dest {
|
||||
case "start_stat":
|
||||
if statErr := u.ReadStat(); statErr == nil {
|
||||
if triggerErr := u.StateMachine().Trigger("exists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if triggerErr := u.StateMachine().Trigger("notexists"); triggerErr == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
case "start_read":
|
||||
if _,readErr := u.Read(ctx); readErr == nil {
|
||||
if triggerErr := u.StateMachine().Trigger("state_read"); triggerErr == nil {
|
||||
@ -133,6 +162,17 @@ func (u *User) Notify(m *machine.EventMessage) {
|
||||
u.Common.State = "absent"
|
||||
panic(readErr)
|
||||
}
|
||||
case "start_update":
|
||||
if updateErr := u.Update(ctx); updateErr == nil {
|
||||
if triggerErr := u.stater.Trigger("updated"); triggerErr == nil {
|
||||
return
|
||||
} else {
|
||||
u.Common.State = "absent"
|
||||
}
|
||||
} else {
|
||||
u.Common.State = "absent"
|
||||
panic(updateErr)
|
||||
}
|
||||
case "start_delete":
|
||||
if deleteErr := u.Delete(ctx); deleteErr == nil {
|
||||
if triggerErr := u.StateMachine().Trigger("deleted"); triggerErr == nil {
|
||||
@ -161,27 +201,16 @@ func (u *User) Notify(m *machine.EventMessage) {
|
||||
}
|
||||
}
|
||||
|
||||
func (u *User) SetURI(uri string) error {
|
||||
resourceUri, e := url.Parse(uri)
|
||||
if e == nil {
|
||||
if resourceUri.Scheme == "user" {
|
||||
u.Name = resourceUri.Hostname()
|
||||
} else {
|
||||
e = fmt.Errorf("%w: %s is not a user", ErrInvalidResourceURI, uri)
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func (u *User) URI() string {
|
||||
return fmt.Sprintf("user://%s", u.Name)
|
||||
}
|
||||
|
||||
func (u *User) UseConfig(config data.ConfigurationValueGetter) {
|
||||
u.config = config
|
||||
}
|
||||
|
||||
func (u *User) ResolveId(ctx context.Context) string {
|
||||
if u.config != nil {
|
||||
if configUser, configUserErr := u.config.GetValue("user"); configUserErr == nil && u.Name == "self" {
|
||||
u.Name = configUser.(string)
|
||||
}
|
||||
}
|
||||
return LookupUIDString(u.Name)
|
||||
}
|
||||
|
||||
@ -236,20 +265,74 @@ func (u *User) Create(ctx context.Context) (error) {
|
||||
return e
|
||||
}
|
||||
|
||||
func (u *User) ReadStat() (err error) {
|
||||
if u.userStatus == nil {
|
||||
if u.userStatus, err = user.Lookup(u.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if len(u.userStatus.Uid) < 1 {
|
||||
return ErrResourceStateAbsent
|
||||
}
|
||||
u.UID = u.userStatus.Uid
|
||||
if len(u.Group) > 1 {
|
||||
if u.groupStatus == nil {
|
||||
if u.groupStatus, err = user.LookupGroup(u.Group); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if len(u.groupStatus.Gid) < 1 {
|
||||
return ErrResourceStateAbsent
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u *User) Read(ctx context.Context) ([]byte, error) {
|
||||
exErr := u.ReadCommand.Extractor(nil, u)
|
||||
if exErr != nil {
|
||||
u.Common.State = "absent"
|
||||
}
|
||||
if yaml, yamlErr := yaml.Marshal(u); yamlErr != nil {
|
||||
return yaml, yamlErr
|
||||
_ = u.ReadGroups()
|
||||
if yamlDoc, yamlErr := yaml.Marshal(u); yamlErr != nil {
|
||||
return yamlDoc, yamlErr
|
||||
} else {
|
||||
return yaml, exErr
|
||||
return yamlDoc, exErr
|
||||
}
|
||||
}
|
||||
|
||||
func (u *User) ReadGroups() (err error) {
|
||||
knownSecondaryGroups := make(map[string]bool)
|
||||
for _, secondaryGroupName := range u.Groups {
|
||||
knownSecondaryGroups[secondaryGroupName] = true
|
||||
}
|
||||
if u.ReadStat() == nil {
|
||||
if groups, groupsErr := u.userStatus.GroupIds(); groupsErr == nil {
|
||||
for _, secondaryGroup := range groups {
|
||||
if readGroup, groupErr := user.LookupGroupId(secondaryGroup); groupErr == nil {
|
||||
if ! knownSecondaryGroups[readGroup.Name] {
|
||||
u.Groups = append(u.Groups, readGroup.Name)
|
||||
knownSecondaryGroups[readGroup.Name] = true
|
||||
}
|
||||
} else {
|
||||
err = groupErr
|
||||
}
|
||||
}
|
||||
} else {
|
||||
err = groupsErr
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u *User) Update(ctx context.Context) (error) {
|
||||
return u.Create(ctx)
|
||||
_, err := u.UpdateCommand.Execute(u)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_,e := u.Read(ctx)
|
||||
return e
|
||||
}
|
||||
|
||||
func (u *User) Delete(ctx context.Context) (error) {
|
||||
@ -279,38 +362,38 @@ func (u *User) UnmarshalYAML(value *yaml.Node) error {
|
||||
|
||||
func (u *UserType) NewCRUD() (create *command.Command, read *command.Command, update *command.Command, del *command.Command) {
|
||||
switch *u {
|
||||
case UserTypeUserAdd:
|
||||
return NewUserAddCreateCommand(), NewUserReadCommand(), NewUserUpdateCommand(), NewUserDelDeleteCommand()
|
||||
case UserTypeAddUser:
|
||||
return NewAddUserCreateCommand(), NewUserReadCommand(), NewUserUpdateCommand(), NewDelUserDeleteCommand()
|
||||
case UserTypeShadow:
|
||||
return NewUserShadowCreateCommand(), NewUserReadCommand(), NewUserShadowUpdateCommand(), NewUserShadowDeleteCommand()
|
||||
case UserTypeBusyBox:
|
||||
return NewUserBusyBoxCreateCommand(), NewUserReadCommand(), NewUserBusyBoxUpdateCommand(), NewUserBusyBoxDeleteCommand()
|
||||
default:
|
||||
if _, addUserPathErr := exec.LookPath("adduser"); addUserPathErr == nil {
|
||||
*u = UserTypeAddUser
|
||||
return NewAddUserCreateCommand(), NewUserReadCommand(), NewUserUpdateCommand(), NewDelUserDeleteCommand()
|
||||
*u = UserTypeBusyBox
|
||||
return NewUserBusyBoxCreateCommand(), NewUserReadCommand(), NewUserBusyBoxUpdateCommand(), NewUserBusyBoxDeleteCommand()
|
||||
}
|
||||
if _, pathErr := exec.LookPath("useradd"); pathErr == nil {
|
||||
*u = UserTypeUserAdd
|
||||
return NewUserAddCreateCommand(), NewUserReadCommand(), NewUserUpdateCommand(), NewUserDelDeleteCommand()
|
||||
*u = UserTypeShadow
|
||||
return NewUserShadowCreateCommand(), NewUserReadCommand(), NewUserShadowUpdateCommand(), NewUserShadowDeleteCommand()
|
||||
}
|
||||
return NewUserAddCreateCommand(), NewUserReadCommand(), NewUserUpdateCommand(), NewUserDelDeleteCommand()
|
||||
return NewUserShadowCreateCommand(), NewUserReadCommand(), NewUserShadowUpdateCommand(), NewUserShadowDeleteCommand()
|
||||
}
|
||||
}
|
||||
|
||||
func (u *UserType) NewReadCommand() (read *command.Command) {
|
||||
return NewUserReadCommand()
|
||||
}
|
||||
|
||||
func (u *UserType) NewCreateCommand() (create *command.Command) {
|
||||
switch *u {
|
||||
case UserTypeUserAdd:
|
||||
return NewUserAddCreateCommand()
|
||||
case UserTypeAddUser:
|
||||
return NewAddUserCreateCommand()
|
||||
case UserTypeShadow:
|
||||
return NewUserShadowCreateCommand()
|
||||
case UserTypeBusyBox:
|
||||
return NewUserBusyBoxCreateCommand()
|
||||
default:
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (u *UserType) NewReadCommand() (*command.Command) {
|
||||
return NewUserReadCommand()
|
||||
}
|
||||
|
||||
func (p *UserType) NewReadUsersCommand() (*command.Command) {
|
||||
return NewReadUsersCommand()
|
||||
}
|
||||
@ -318,7 +401,7 @@ func (p *UserType) NewReadUsersCommand() (*command.Command) {
|
||||
|
||||
func (u *UserType) UnmarshalValue(value string) error {
|
||||
switch value {
|
||||
case string(UserTypeUserAdd), string(UserTypeAddUser):
|
||||
case string(UserTypeShadow), string(UserTypeBusyBox):
|
||||
*u = UserType(value)
|
||||
return nil
|
||||
default:
|
||||
@ -385,7 +468,7 @@ func NewReadUsersCommand() *command.Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func NewUserAddCreateCommand() *command.Command {
|
||||
func NewUserShadowCreateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "useradd"
|
||||
c.Args = []command.CommandArg{
|
||||
@ -399,19 +482,11 @@ func NewUserAddCreateCommand() *command.Command {
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
return nil
|
||||
/*
|
||||
for _,line := range strings.Split(string(out), "\n") {
|
||||
if line == "iptables: Chain already exists." {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return fmt.Errorf(string(out))
|
||||
*/
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewAddUserCreateCommand() *command.Command {
|
||||
func NewUserBusyBoxCreateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "adduser"
|
||||
c.Args = []command.CommandArg{
|
||||
@ -425,14 +500,6 @@ func NewAddUserCreateCommand() *command.Command {
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
return nil
|
||||
/*
|
||||
for _,line := range strings.Split(string(out), "\n") {
|
||||
if line == "iptables: Chain already exists." {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return fmt.Errorf(string(out))
|
||||
*/
|
||||
}
|
||||
return c
|
||||
}
|
||||
@ -471,11 +538,40 @@ func NewUserReadCommand() *command.Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func NewUserUpdateCommand() *command.Command {
|
||||
func NewUserBusyBoxUpdateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "xargs"
|
||||
c.StdinAvailable = true
|
||||
c.Input = command.CommandInput("{{ if .Groups }}{{ range .Groups }}{{ . }}\n{{ end }}{{ end }}")
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("adduser"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
return nil
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewUserDelDeleteCommand() *command.Command {
|
||||
func NewUserShadowUpdateCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "usermod"
|
||||
c.Args = []command.CommandArg{
|
||||
command.CommandArg("{{ if .UID }}-u {{ .UID }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Gecos }}-c {{ .Gecos }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Group }}-g {{ .groupStatus.Gid }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Home }}-d {{ .Home }}{{ if .CreateHome }} -m{{- end }}{{ end }}"),
|
||||
command.CommandArg("{{ if .Groups }}-G {{- range $i, $g := .Groups -}}{{ if $i }}, {{- end }}{{ . }}{{- end }}{{ if .AppendGroups }} -a{{- end }}{{- end }}"),
|
||||
command.CommandArg("{{ if .Shell }}-s {{ .Shell }}{{ end }}"),
|
||||
command.CommandArg("{{ .Name }}"),
|
||||
}
|
||||
c.Extractor = func(out []byte, target any) error {
|
||||
return nil
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewUserShadowDeleteCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "userdel"
|
||||
c.Args = []command.CommandArg{
|
||||
@ -487,7 +583,7 @@ func NewUserDelDeleteCommand() *command.Command {
|
||||
return c
|
||||
}
|
||||
|
||||
func NewDelUserDeleteCommand() *command.Command {
|
||||
func NewUserBusyBoxDeleteCommand() *command.Command {
|
||||
c := command.NewCommand()
|
||||
c.Path = "deluser"
|
||||
c.Args = []command.CommandArg{
|
||||
|
@ -13,6 +13,7 @@ import (
|
||||
_ "os"
|
||||
_ "strings"
|
||||
"testing"
|
||||
"decl/internal/data"
|
||||
)
|
||||
|
||||
func TestNewUserResource(t *testing.T) {
|
||||
@ -77,3 +78,43 @@ func TestCreateUser(t *testing.T) {
|
||||
applyDeleteErr := u.Apply()
|
||||
assert.Equal(t, nil, applyDeleteErr)
|
||||
}
|
||||
|
||||
func TestSystemUser(t *testing.T) {
|
||||
u := NewUser()
|
||||
u.Name = "self"
|
||||
u.UseConfig(MockConfigValueGetter(func(key string) (any, error) {
|
||||
switch key {
|
||||
case "user":
|
||||
return "bar", nil
|
||||
case "group":
|
||||
return "foo", nil
|
||||
case "uid":
|
||||
return "", nil
|
||||
}
|
||||
return nil, fmt.Errorf("%w: %s", data.ErrUnknownConfigurationKey, key)
|
||||
}))
|
||||
|
||||
assert.Nil(t, u.Init(nil))
|
||||
u.ResolveId(context.Background())
|
||||
|
||||
assert.Equal(t, "bar", u.Name)
|
||||
}
|
||||
|
||||
func TestUserSecondaryGroups(t *testing.T) {
|
||||
groupCounts := make(map[string]int)
|
||||
u := NewUser()
|
||||
u.Name = "root"
|
||||
u.ResolveId(context.Background())
|
||||
u.Groups = []string{
|
||||
"root",
|
||||
"wheel",
|
||||
}
|
||||
assert.Nil(t, u.ReadGroups())
|
||||
|
||||
for _, groupName := range u.Groups {
|
||||
groupCounts[groupName]++
|
||||
}
|
||||
assert.Greater(t, len(u.Groups), 2)
|
||||
assert.Equal(t, 1, groupCounts["root"])
|
||||
assert.Equal(t, 1, groupCounts["wheel"])
|
||||
}
|
||||
|
126
internal/system/os.go
Normal file
126
internal/system/os.go
Normal file
@ -0,0 +1,126 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
package system
|
||||
|
||||
import (
|
||||
"os/user"
|
||||
"strconv"
|
||||
"regexp"
|
||||
"log/slog"
|
||||
)
|
||||
|
||||
var MatchId *regexp.Regexp = regexp.MustCompile(`^[0-9]+$`)
|
||||
|
||||
func LookupUIDString(userName string) string {
|
||||
user, userLookupErr := user.Lookup(userName)
|
||||
if userLookupErr != nil {
|
||||
return ""
|
||||
}
|
||||
return user.Uid
|
||||
}
|
||||
|
||||
func LookupUID(userName string) (int, error) {
|
||||
var userLookupErr error
|
||||
var UID string
|
||||
if MatchId.MatchString(userName) {
|
||||
user, err := user.LookupId(userName)
|
||||
slog.Info("LookupUID() numeric", "user", user, "userLookupErr", err)
|
||||
if err != nil {
|
||||
userLookupErr = err
|
||||
UID = userName
|
||||
} else {
|
||||
UID = user.Uid
|
||||
}
|
||||
} else {
|
||||
if user, err := user.Lookup(userName); err != nil {
|
||||
return -1, err
|
||||
} else {
|
||||
UID = user.Uid
|
||||
}
|
||||
}
|
||||
|
||||
uid, uidErr := strconv.Atoi(UID)
|
||||
slog.Info("LookupUID()", "uid", uid, "uidErr", uidErr)
|
||||
if uidErr != nil {
|
||||
if userLookupErr != nil {
|
||||
return -1, userLookupErr
|
||||
} else {
|
||||
return -1, uidErr
|
||||
}
|
||||
}
|
||||
|
||||
return uid, userLookupErr
|
||||
}
|
||||
|
||||
func LookupGID(groupName string) (int, error) {
|
||||
var GID string
|
||||
if MatchId.MatchString(groupName) {
|
||||
group, groupLookupErr := user.LookupGroupId(groupName)
|
||||
if groupLookupErr != nil {
|
||||
//return -1, groupLookupErr
|
||||
GID = groupName
|
||||
} else {
|
||||
GID = group.Gid
|
||||
}
|
||||
} else {
|
||||
group, groupLookupErr := user.LookupGroup(groupName)
|
||||
if groupLookupErr != nil {
|
||||
return -1, groupLookupErr
|
||||
}
|
||||
GID = group.Gid
|
||||
}
|
||||
|
||||
gid, gidErr := strconv.Atoi(GID)
|
||||
if gidErr != nil {
|
||||
return -1, gidErr
|
||||
}
|
||||
|
||||
return gid, nil
|
||||
}
|
||||
|
||||
func LookupGIDString(groupName string) string {
|
||||
group, groupLookupErr := user.LookupGroup(groupName)
|
||||
if groupLookupErr != nil {
|
||||
return ""
|
||||
}
|
||||
return group.Gid
|
||||
}
|
||||
|
||||
func ProcessUser() *user.User {
|
||||
processUser, userErr := user.Current()
|
||||
if userErr != nil {
|
||||
panic(userErr)
|
||||
}
|
||||
return processUser
|
||||
}
|
||||
|
||||
func ProcessGroup(u *user.User) *user.Group {
|
||||
if u == nil {
|
||||
u = ProcessUser()
|
||||
}
|
||||
processGroup, groupErr := user.LookupGroupId(u.Gid)
|
||||
if groupErr != nil {
|
||||
panic(groupErr)
|
||||
}
|
||||
return processGroup
|
||||
}
|
||||
|
||||
func ProcessUserName() string {
|
||||
processUser, userErr := user.Current()
|
||||
if userErr != nil {
|
||||
panic(userErr)
|
||||
}
|
||||
return processUser.Username
|
||||
}
|
||||
|
||||
func ProcessGroupName() string {
|
||||
processUser, userErr := user.Current()
|
||||
if userErr != nil {
|
||||
panic(userErr)
|
||||
}
|
||||
processGroup, groupErr := user.LookupGroupId(processUser.Gid)
|
||||
if groupErr != nil {
|
||||
panic(groupErr)
|
||||
}
|
||||
return processGroup.Name
|
||||
}
|
38
internal/system/os_test.go
Normal file
38
internal/system/os_test.go
Normal file
@ -0,0 +1,38 @@
|
||||
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
||||
|
||||
|
||||
package system
|
||||
|
||||
import (
|
||||
_ "context"
|
||||
_ "encoding/json"
|
||||
"github.com/stretchr/testify/assert"
|
||||
_ "io"
|
||||
_ "net/http"
|
||||
_ "net/http/httptest"
|
||||
_ "net/url"
|
||||
_ "strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestLookupUID(t *testing.T) {
|
||||
uid, e := LookupUID("nobody")
|
||||
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, 65534, uid)
|
||||
|
||||
nuid, ne := LookupUID("10101")
|
||||
assert.Error(t, ne, "user: unknonwn userid ", ne)
|
||||
assert.Equal(t, 10101, nuid)
|
||||
}
|
||||
|
||||
func TestLookupGID(t *testing.T) {
|
||||
gid, e := LookupGID("adm")
|
||||
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, 4, gid)
|
||||
|
||||
ngid, ne := LookupGID("1001")
|
||||
assert.Nil(t, ne)
|
||||
assert.Equal(t, 1001, ngid)
|
||||
}
|
@ -91,6 +91,10 @@ func (h *HTTP) Stat() (info fs.FileInfo, err error) {
|
||||
}
|
||||
defer response.Body.Close()
|
||||
|
||||
if response.StatusCode > 399 {
|
||||
err = ErrTransportResourceAbsent
|
||||
return
|
||||
}
|
||||
fi := NewHTTPFileInfo(uri)
|
||||
contentLength := response.Header.Get("Content-Length")
|
||||
if contentLength != "" {
|
||||
|
@ -3,7 +3,7 @@
|
||||
package transport
|
||||
|
||||
import (
|
||||
_ "errors"
|
||||
"errors"
|
||||
_ "fmt"
|
||||
"net/url"
|
||||
"net/http"
|
||||
@ -15,6 +15,10 @@ _ "os"
|
||||
"context"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrTransportResourceAbsent error = errors.New("Transport resource is absent")
|
||||
)
|
||||
|
||||
type Handler interface {
|
||||
URI() *url.URL
|
||||
ContentType() string
|
||||
@ -162,11 +166,17 @@ func (r *Reader) AddHeader(name string, value string) {
|
||||
}
|
||||
|
||||
func (r *Reader) Status() string {
|
||||
return r.handle.(*HTTPReader).GetResponse().Status
|
||||
if httpReader, ok := r.handle.(*HTTPReader); ok {
|
||||
return httpReader.GetResponse().Status
|
||||
}
|
||||
panic("Unable to get HTTP status from reader")
|
||||
}
|
||||
|
||||
func (r *Reader) StatusCode() int {
|
||||
return r.handle.(*HTTPReader).GetResponse().StatusCode
|
||||
if httpReader, ok := r.handle.(*HTTPReader); ok {
|
||||
return httpReader.GetResponse().StatusCode
|
||||
}
|
||||
panic("Unable to get HTTP status code from reader")
|
||||
}
|
||||
|
||||
func (r *Reader) Response() *http.Response {
|
||||
|
@ -105,6 +105,19 @@ func (t *Types[Product]) NewFromParsedURI(u *url.URL) (result Product, err error
|
||||
return
|
||||
}
|
||||
|
||||
// Returns an uninitialized resource for the given type
|
||||
func (t *Types[Product]) NewFromType(typename string) (result Product, err error) {
|
||||
if r, ok := t.registry[typename]; ok {
|
||||
if result = r(nil); result != any(nil) {
|
||||
return result, nil
|
||||
} else {
|
||||
return result, fmt.Errorf("%w: factory failed creating %s", ErrInvalidProduct, typename)
|
||||
}
|
||||
}
|
||||
err = fmt.Errorf("%w: %s", ErrUnknownType, typename)
|
||||
return
|
||||
}
|
||||
|
||||
func (t *Types[Product]) Has(typename string) bool {
|
||||
if _, ok := t.registry[typename]; ok {
|
||||
return true
|
||||
|
@ -61,3 +61,17 @@ func TestTypeName(t *testing.T) {
|
||||
assert.Equal(t, "file", string(fTypeName.Name))
|
||||
}
|
||||
*/
|
||||
|
||||
func TestNewFromType(t *testing.T) {
|
||||
testTypes := New[string]()
|
||||
assert.NotNil(t, testTypes)
|
||||
|
||||
testTypes.Register([]string{"foo"}, func(*url.URL) string { return "test" })
|
||||
|
||||
r, e := testTypes.NewFromType("foo")
|
||||
assert.Nil(t, e)
|
||||
assert.Equal(t, "test", r)
|
||||
failr, faile := testTypes.NewFromType("bar")
|
||||
assert.ErrorIs(t, faile, ErrUnknownType)
|
||||
assert.Equal(t, "", failr)
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user