mirror of
https://github.com/Luzifer/korvike.git
synced 2024-12-20 11:11:19 +00:00
add initial version
This commit is contained in:
parent
d6112c11c0
commit
a101eeb060
3 changed files with 113 additions and 0 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
@ -0,0 +1 @@
|
||||||
|
korvike
|
96
app.go
Normal file
96
app.go
Normal file
|
@ -0,0 +1,96 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"sync"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/Luzifer/go_helpers/env"
|
||||||
|
"github.com/Luzifer/rconfig"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
cfg = struct {
|
||||||
|
Input string `flag:"in,i" default:"-" description:"Source to read the template from ('-' = stdin)"`
|
||||||
|
KeyPairs []string `flag:"key-value,v" default:"" description:"Key-value pairs to use in templating (-v key=value)"`
|
||||||
|
Output string `flag:"out,o" default:"-" description:"Destination to write the output to ('-' = stdout)"`
|
||||||
|
VersionAndExit bool `flag:"version" default:"false" description:"Prints current version and exits"`
|
||||||
|
}{}
|
||||||
|
|
||||||
|
templateFunctions = template.FuncMap{}
|
||||||
|
templateFunctionsLock sync.Mutex
|
||||||
|
|
||||||
|
version = "dev"
|
||||||
|
)
|
||||||
|
|
||||||
|
func registerFunction(name string, f interface{}) error {
|
||||||
|
templateFunctionsLock.Lock()
|
||||||
|
defer templateFunctionsLock.Unlock()
|
||||||
|
if _, ok := templateFunctions[name]; ok {
|
||||||
|
return errors.New("Duplicate function for name " + name)
|
||||||
|
}
|
||||||
|
templateFunctions[name] = f
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
if err := rconfig.Parse(&cfg); err != nil {
|
||||||
|
log.Fatalf("Unable to parse commandline options: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.VersionAndExit {
|
||||||
|
fmt.Printf("korvike %s\n", version)
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func openInput(f string) (io.Reader, error) {
|
||||||
|
if f == "-" {
|
||||||
|
return os.Stdin, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := os.Stat(f); err != nil {
|
||||||
|
return nil, errors.New("Could not find file " + f)
|
||||||
|
}
|
||||||
|
|
||||||
|
return os.Open(f)
|
||||||
|
}
|
||||||
|
|
||||||
|
func openOutput(f string) (io.Writer, error) {
|
||||||
|
if f == "-" {
|
||||||
|
return os.Stdout, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return os.Create(f)
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
in, err := openInput(cfg.Input)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to open input: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
out, err := openOutput(cfg.Output)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to open output: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rawTpl, err := ioutil.ReadAll(in)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to read from input: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tpl, err := template.New("in").Funcs(templateFunctions).Parse(string(rawTpl))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to parse template: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := tpl.Execute(out, env.ListToMap(cfg.KeyPairs)); err != nil {
|
||||||
|
log.Fatalf("Unable to execute template: %s", err)
|
||||||
|
}
|
||||||
|
}
|
16
func_env.go
Normal file
16
func_env.go
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import "os"
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
registerFunction("env", func(name string, v ...string) string {
|
||||||
|
defaultValue := ""
|
||||||
|
if len(v) > 0 {
|
||||||
|
defaultValue = v[0]
|
||||||
|
}
|
||||||
|
if value, ok := os.LookupEnv(name); ok {
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
return defaultValue
|
||||||
|
})
|
||||||
|
}
|
Loading…
Reference in a new issue