init

Signed-off-by: storyicon <yuanchao@bilibili.com>

pkg/configs/configs.go

@@ -0,0 +1,109 @@
+// Copyright 2021 storyicon@foxmail.com
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// 	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package configs
+
+import (
+	"bytes"
+	"io/fs"
+	"io/ioutil"
+	"path/filepath"
+
+	"gopkg.in/yaml.v3"
+
+	"github.com/storyicon/powerproto/pkg/consts"
+	"github.com/storyicon/powerproto/pkg/util"
+)
+
+// Config defines the config model
+type Config struct {
+	Scopes        []string          `json:"scopes" yaml:"scopes"`
+	Protoc        string            `json:"protoc" yaml:"protoc"`
+	ProtocWorkDir string            `json:"protocWorkDir" yaml:"protocWorkDir"`
+	Plugins       map[string]string `json:"plugins" yaml:"plugins"`
+	Options       []string          `json:"options" yaml:"options"`
+	ImportPaths   []string          `json:"importPaths" yaml:"importPaths"`
+	PostActions   []*PostAction     `json:"postActions" yaml:"postActions"`
+	PostShell     string            `json:"postShell" yaml:"postShell"`
+}
+
+// PostAction defines the Action model
+type PostAction struct {
+	Name string   `json:"name" yaml:"name"`
+	Args []string `json:"args" yaml:"args"`
+}
+
+// SaveConfigs is used to save configs into files
+func SaveConfigs(path string, configs ...*Config) error {
+	parts := make([][]byte, 0, len(configs))
+	for _, config := range configs {
+		data, err := yaml.Marshal(config)
+		if err != nil {
+			return err
+		}
+		parts = append(parts, data)
+	}
+	data := bytes.Join(parts, []byte("\r\n---\r\n"))
+	if err := ioutil.WriteFile(path, data, fs.ModePerm); err != nil {
+		return err
+	}
+	return nil
+}
+
+// LoadConfigs is used to load config from specified path
+func LoadConfigs(path string) ([]*Config, error) {
+	raw, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+	items, err := util.SplitYAML(raw)
+	if err != nil {
+		return nil, err
+	}
+	var ret []*Config
+	for _, item := range items {
+		var config Config
+		if err := yaml.Unmarshal(item, &config); err != nil {
+			return nil, err
+		}
+		ret = append(ret, &config)
+	}
+	return ret, nil
+}
+
+// Similar to LoadConfigs, but obtains the abstraction of the Config prototype structure
+func LoadConfigItems(path string) ([]ConfigItem, error) {
+	data, err := LoadConfigs(path)
+	if err != nil {
+		return nil, err
+	}
+	return GetConfigItems(data, path), nil
+}
+
+// ListConfigPaths is used to list all possible config paths
+func ListConfigPaths(sourceDir string) []string {
+	var paths []string
+	cur := sourceDir
+	for {
+		filePath := filepath.Join(cur, consts.ConfigFileName)
+		paths = append(paths, filePath)
+		next := filepath.Dir(cur)
+		if next == cur {
+			break
+		}
+		cur = next
+	}
+	paths = append(paths, consts.PathForGlobalConfig())
+	return paths
+}