go-serviceman/vendor/github.com/UnnoTed/fileb0x/config/config.go

76 lines
1.3 KiB
Go
Raw Normal View History

2019-07-01 08:44:54 +00:00
package config
import (
"strings"
"github.com/UnnoTed/fileb0x/compression"
"github.com/UnnoTed/fileb0x/custom"
"github.com/UnnoTed/fileb0x/updater"
)
// Config holds the json/yaml/toml data
type Config struct {
Dest string
NoPrefix bool
Pkg string
Fmt bool // gofmt
Compression *compression.Options
Tags string
Output string
Custom []custom.Custom
Spread bool
Unexported bool
Clean bool
Debug bool
Updater updater.Config
Lcf bool
}
// Defaults set the default value for some variables
func (cfg *Config) Defaults() error {
// default destination
if cfg.Dest == "" {
cfg.Dest = "/"
}
// insert "/" at end of dest when it's not found
if !strings.HasSuffix(cfg.Dest, "/") {
cfg.Dest += "/"
}
// default file name
if cfg.Output == "" {
cfg.Output = "b0x.go"
}
// inserts .go at the end of file name
if !strings.HasSuffix(cfg.Output, ".go") {
cfg.Output += ".go"
}
// inserts an A before the output file's name so it can
// run init() before b0xfile's
if !cfg.NoPrefix && !strings.HasPrefix(cfg.Output, "a") {
cfg.Output = "a" + cfg.Output
}
// default package
if cfg.Pkg == "" {
cfg.Pkg = "main"
}
if cfg.Compression == nil {
cfg.Compression = &compression.Options{
Compress: false,
Method: "DefaultCompression",
Keep: false,
}
}
return nil
}