chore: import upstream snapshot with attribution

2026-07-13 12:22:32 +08:00
commit 39dbe3a57d
1131 changed files with 272709 additions and 0 deletions
@@ -0,0 +1,128 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 conf
+
+import (
+	"bytes"
+	"os"
+	"path/filepath"
+
+	"github.com/apache/answer/internal/base/data"
+	"github.com/apache/answer/internal/base/path"
+	"github.com/apache/answer/internal/base/server"
+	"github.com/apache/answer/internal/base/translator"
+	"github.com/apache/answer/internal/router"
+	"github.com/apache/answer/internal/service/service_config"
+	"github.com/apache/answer/pkg/writer"
+	"github.com/segmentfault/pacman/contrib/conf/viper"
+	"gopkg.in/yaml.v3"
+)
+
+// AllConfig all config
+type AllConfig struct {
+	Debug         bool                          `json:"debug" mapstructure:"debug" yaml:"debug"`
+	Server        *Server                       `json:"server" mapstructure:"server" yaml:"server"`
+	Data          *Data                         `json:"data" mapstructure:"data" yaml:"data"`
+	I18n          *translator.I18n              `json:"i18n" mapstructure:"i18n" yaml:"i18n"`
+	ServiceConfig *service_config.ServiceConfig `json:"service_config" mapstructure:"service_config" yaml:"service_config"`
+	Swaggerui     *router.SwaggerConfig         `json:"swaggerui" mapstructure:"swaggerui" yaml:"swaggerui"`
+	UI            *server.UI                    `json:"ui" mapstructure:"ui" yaml:"ui"`
+}
+
+type envConfigOverrides struct {
+	SwaggerHost        string
+	SwaggerAddressPort string
+	SiteAddr           string
+}
+
+func loadEnvs() (envOverrides *envConfigOverrides) {
+	return &envConfigOverrides{
+		SwaggerHost:        os.Getenv("SWAGGER_HOST"),
+		SwaggerAddressPort: os.Getenv("SWAGGER_ADDRESS_PORT"),
+		SiteAddr:           os.Getenv("SITE_ADDR"),
+	}
+}
+
+type PathIgnore struct {
+	Users []string `yaml:"users"`
+}
+
+// Server server config
+type Server struct {
+	HTTP *server.HTTP `json:"http" mapstructure:"http" yaml:"http"`
+}
+
+// Data data config
+type Data struct {
+	Database *data.Database  `json:"database" mapstructure:"database" yaml:"database"`
+	Cache    *data.CacheConf `json:"cache" mapstructure:"cache" yaml:"cache"`
+}
+
+// SetDefault set default config
+func (c *AllConfig) SetDefault() {
+	if c.UI == nil {
+		c.UI = &server.UI{}
+	}
+}
+
+func (c *AllConfig) SetEnvironmentOverrides() {
+	envs := loadEnvs()
+	if envs.SiteAddr != "" {
+		c.Server.HTTP.Addr = envs.SiteAddr
+	}
+	if envs.SwaggerHost != "" {
+		c.Swaggerui.Host = envs.SwaggerHost
+	}
+	if envs.SwaggerAddressPort != "" {
+		c.Swaggerui.Address = envs.SwaggerAddressPort
+	}
+}
+
+// ReadConfig read config
+func ReadConfig(configFilePath string) (c *AllConfig, err error) {
+	if len(configFilePath) == 0 {
+		configFilePath = filepath.Join(path.ConfigFileDir, path.DefaultConfigFileName)
+	}
+	c = &AllConfig{}
+	config, err := viper.NewWithPath(configFilePath)
+	if err != nil {
+		return nil, err
+	}
+	if err = config.Parse(&c); err != nil {
+		return nil, err
+	}
+	c.SetDefault()
+	c.SetEnvironmentOverrides()
+	return c, nil
+}
+
+// RewriteConfig rewrite config file path
+func RewriteConfig(configFilePath string, allConfig *AllConfig) error {
+	buf := bytes.Buffer{}
+	enc := yaml.NewEncoder(&buf)
+	defer func() {
+		_ = enc.Close()
+	}()
+	enc.SetIndent(2)
+	if err := enc.Encode(allConfig); err != nil {
+		return err
+	}
+	return writer.ReplaceFile(configFilePath, buf.String())
+}