refactor: change structure project

agent/internal/utils/config_yaml/config.go

@@ -0,0 +1,50 @@
+package config_yaml
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/rs/zerolog"
+	"gopkg.in/yaml.v3"
+)
+
+type AgentConfig struct {
+	AppName    string `yaml:"app_name"`
+	HubConnect struct {
+		Host string `yaml:"host"`
+		Port int    `yaml:"port"`
+	} `yaml:"hub"`
+	LogLevel     string `yaml:"log_level"`
+	SettingsPath string `yaml:"settings_path"`
+}
+
+func NewConfig() (*AgentConfig, error) {
+	yamlFile, err := os.ReadFile("agent.dev.yaml")
+	if err != nil {
+		return nil, fmt.Errorf("failed open file: %v", err)
+	}
+
+	var cfg AgentConfig
+
+	if err = yaml.Unmarshal(yamlFile, &cfg); err != nil {
+		return nil, fmt.Errorf("failed read yaml: %v", err)
+	}
+
+	return &cfg, nil
+}
+
+func (c *AgentConfig) GetLogLevel() zerolog.Level {
+	level, err := zerolog.ParseLevel(c.LogLevel)
+	if err != nil {
+		return zerolog.InfoLevel
+	}
+	return level
+}
+
+func (c *AgentConfig) GetMode() string {
+	return "DEV"
+}
+
+func (c *AgentConfig) GetGRPCAddress() string {
+	return fmt.Sprintf("%v:%v", c.HubConnect.Host, c.HubConnect.Port)
+}

agent/internal/utils/config_yaml/config_yaml_test.go

@@ -0,0 +1,41 @@
+package config_yaml
+
+import (
+	"testing"
+
+	"github.com/rs/zerolog"
+)
+
+func TestAgentConfig_GetLogLevel(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name         string
+		cfg          AgentConfig
+		wantLogLevel zerolog.Level
+	}{
+		{
+			name:         "success",
+			cfg:          AgentConfig{LogLevel: "DEBUG"},
+			wantLogLevel: zerolog.DebugLevel,
+		},
+		{
+			name:         "failed parse",
+			cfg:          AgentConfig{LogLevel: "TEST"},
+			wantLogLevel: zerolog.InfoLevel,
+		},
+	}
+
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			logLevel := tt.cfg.GetLogLevel()
+
+			if logLevel != tt.wantLogLevel {
+				t.Fatalf("expected %v, got: %v", tt.wantLogLevel, logLevel)
+			}
+		})
+	}
+}

agent/internal/utils/settings/settings.go

@@ -0,0 +1,66 @@
+package settings
+
+import (
+	"encoding/json"
+	"errors"
+	"io"
+	"os"
+	"path/filepath"
+)
+
+type Settings struct {
+	AgentID string `json:"agent_id"`
+	path    string
+}
+
+func ReadSettings(path string) (*Settings, error) {
+	if path == "" {
+		homeDir, err := os.UserHomeDir()
+		if err != nil {
+			return nil, err
+		}
+		path = filepath.Join(homeDir, ".config", "homeops")
+	}
+
+	err := os.Mkdir(path, 0755)
+	if err != nil {
+		if !errors.Is(err, os.ErrExist) {
+			return nil, err
+		}
+		err = nil
+	}
+
+	settingsPath := filepath.Join(path, "settings.json")
+	var settings Settings
+
+	file, err := os.Open(settingsPath)
+	if err != nil {
+		if !errors.Is(err, os.ErrNotExist) {
+			return nil, err
+		}
+	} else {
+		defer file.Close()
+		err = json.NewDecoder(file).Decode(&settings)
+		if err != nil && !errors.Is(err, io.EOF) {
+			return nil, err
+		}
+	}
+
+	settings.path = settingsPath
+
+	return &settings, nil
+}
+
+func (s *Settings) Insert(sett Settings) error {
+	file, err := os.OpenFile(s.path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	if err = json.NewEncoder(file).Encode(sett); err != nil {
+		return err
+	}
+
+	return nil
+}