Initial commit

config/config.go

@@ -0,0 +1,54 @@
+// maubot - A plugin-based Matrix bot system written in Go.
+// Copyright (C) 2018 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package config
+
+import (
+	"io/ioutil"
+
+	"gopkg.in/yaml.v2"
+	"maubot.xyz/database"
+)
+
+type MainConfig struct {
+	Logging    LogConfig         `yaml:"logging"`
+	Database   database.Database `yaml:"database"`
+	Server     ServerConfig      `yaml:"server"`
+	PluginDirs []string          `yaml:"plugin_directories"`
+}
+
+func (config *MainConfig) Load(path string) error {
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		return err
+	}
+
+	return yaml.Unmarshal(data, config)
+}
+
+func (config *MainConfig) Save(path string) error {
+	data, err := yaml.Marshal(config)
+	if err != nil {
+		return err
+	}
+
+	return ioutil.WriteFile(path, data, 0644)
+}
+
+type ServerConfig struct {
+	Listen   string `yaml:"listen"`
+	BasePath string `yaml:"base_path"`
+}

config/logging.go

@@ -0,0 +1,68 @@
+// maubot - A plugin-based Matrix bot system written in Go.
+// Copyright (C) 2018 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package config
+
+import (
+	"maunium.net/go/maulogger"
+	"os"
+	"fmt"
+)
+
+// LogConfig contains configs for the logger.
+type LogConfig struct {
+	Directory       string `yaml:"directory"`
+	FileNameFormat  string `yaml:"file_name_format"`
+	FileDateFormat  string `yaml:"file_date_format"`
+	FileMode        uint32 `yaml:"file_mode"`
+	TimestampFormat string `yaml:"timestamp_format"`
+	Debug           bool   `yaml:"print_debug"`
+}
+
+// CreateLogConfig creates a basic LogConfig.
+func CreateLogConfig() LogConfig {
+	return LogConfig{
+		Directory:       "./logs",
+		FileNameFormat:  "%[1]s-%02[2]d.log",
+		TimestampFormat: "Jan _2, 2006 15:04:05",
+		FileMode:        0600,
+		FileDateFormat:  "2006-01-02",
+		Debug:           false,
+	}
+}
+
+// GetFileFormat returns a mauLogger-compatible logger file format based on the data in the struct.
+func (lc LogConfig) GetFileFormat() maulogger.LoggerFileFormat {
+	path := lc.FileNameFormat
+	if len(lc.Directory) > 0 {
+		path = lc.Directory + "/" + path
+	}
+
+	return func(now string, i int) string {
+		return fmt.Sprintf(path, now, i)
+	}
+}
+
+// Configure configures a mauLogger instance with the data in this struct.
+func (lc LogConfig) Configure(log *maulogger.Logger) {
+	log.FileFormat = lc.GetFileFormat()
+	log.FileMode = os.FileMode(lc.FileMode)
+	log.FileTimeFormat = lc.FileDateFormat
+	log.TimeFormat = lc.TimestampFormat
+	if lc.Debug {
+		log.PrintLevel = maulogger.LevelDebug.Severity
+	}
+}