load_saseul.go

package main

import (
	"bufio"
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"

	"github.com/joho/godotenv"
)

// SASEUL_VERSION은 SASEUL의 현재 버전을 나타냅니다.
const SASEUL_VERSION = "2.1.9.6"

// DEBUG_LOG는 디버그 로그 파일의 경로를 나타냅니다.
const DEBUG_LOG = "debug.log"

var (
	ROOT       = filepath.Dir(filepath.Dir(os.Args[0])) // 프로젝트의 루트 디렉터리 경로입니다.
	SOURCE     = filepath.Dir(os.Args[0])                // 현재 소스 파일의 디렉터리 경로입니다.
	SCRIPT_BIN = filepath.Join(SOURCE, "saseul-script")  // 스크립트 실행 파일의 경로입니다.
	SERVICE_BIN = filepath.Join(SOURCE, "saseulsvc")     // 서비스 실행 파일의 경로입니다.
	SASEUL_INI map[string]string                        // SASEUL 설정 파일에서 로드된 설정들을 저장합니다.
)

func init() {
	// SASEUL 설정 파일인 saseul.ini 파일을 로드합니다.
	godotenv.Load(filepath.Join(ROOT, "saseul.ini"))

	// SASEUL_INI 맵에 설정 값을 저장합니다.
	SASEUL_INI = map[string]string{
		"log":             getEnv("log", DEBUG_LOG),                     // 로그 파일 경로
		"version":         getEnv("version", SASEUL_VERSION),           // 버전 정보
		"data":            getEnv("data", filepath.Join(ROOT, "data")), // 데이터 디렉터리 경로
		"peers":           getEnv("peers", ""),                         // 피어 설정
		"network_name":    getEnv("network_name", ""),                  // 네트워크 이름 설정
		"system_nonce":    getEnv("system_nonce", ""),                  // 시스템 논스 설정
		"genesis_address": getEnv("genesis_address", ""),               // genesis 주소 설정
		"manager_addresses": getEnv("manager_addresses", ""),           // 매니저 주소 설정
		"ledger":          getEnv("ledger", ""),                        // 분산장부 설정
		"database":        getEnv("database", ""),                      // 데이터베이스 설정
		"collect":         getEnv("collect", ""),                       // 데이터 수집 설정
		"consensus":       getEnv("consensus", ""),                     // 합의 설정
		"mining":          getEnv("mining", ""),                        // 채굴 설정
		"mysql_host":      getEnv("mysql_host", ""),                    // MySQL 호스트 설정
		"mysql_port":      getEnv("mysql_port", ""),                    // MySQL 포트 설정
		"mysql_user":      getEnv("mysql_user", ""),                    // MySQL 사용자 설정
		"mysql_database":  getEnv("mysql_database", ""),                // MySQL 데이터베이스 설정
		"mysql_password":  getEnv("mysql_password", ""),                // MySQL 비밀번호 설정
		"err_file":        getEnv("err_file", DEBUG_LOG),               // 오류 로그 파일 경로
	}
}

func main() {
	// saseul.ini 파일이 존재하는지 확인합니다.
	if !fileExists(filepath.Join(ROOT, "saseul.ini")) {
		fmt.Println("\nsaseul.ini file does not exist.")
		os.Exit(1)
	}

	// 로그 파일을 설정합니다.
	setLogFile()

	// 예제: 전역 변수 설정
	// Config.Version = SASEUL_INI["version"]
	// Config.Data = SASEUL_INI["data"]

	// 오류 처리 핸들러를 설정합니다.
	setErrorHandler()
	setFatalErrorHandler()

	// 메인 로직의 나머지 부분을 여기에 작성합니다.
}

// getEnv 함수는 환경 변수에서 값을 가져오거나 기본 값을 반환합니다.
func getEnv(key, defaultValue string) string {
	if value, exists := os.LookupEnv(key); exists {
		return value
	}
	return defaultValue
}

// fileExists 함수는 파일이 존재하는지 확인합니다.
func fileExists(filename string) bool {
	info, err := os.Stat(filename)
	return err == nil && !info.IsDir()
}

// setLogFile 함수는 로그 파일을 설정합니다.
func setLogFile() {
	logFile := SASEUL_INI["log"]
	// 로그 파일이 존재하지 않으면 기본 디버그 로그 파일을 사용합니다.
	if !fileExists(filepath.Dir(logFile)) {
		logFile = DEBUG_LOG
	}
	LoggerSetLogFile(logFile)
}

// setErrorHandler 함수는 일반적인 오류 처리 핸들러를 설정합니다.
func setErrorHandler() {
	defer func() {
		if err := recover(); err != nil {
			handleError(err)
		}
	}()
}

// setFatalErrorHandler 함수는 치명적인 오류 처리 핸들러를 설정합니다.
func setFatalErrorHandler() {
	defer func() {
		if r := recover(); r != nil {
			handleFatalError(r)
		}
	}()
}

// handleError 함수는 발생한 일반적인 오류를 처리하고 로그에 기록합니다.
func handleError(err interface{}) {
	msg := fmt.Sprintf("\n[%s] [ERROR] %v\n\n", time.Now().Format("2006-01-02 15:04:05"), err)
	writeToFile(SASEUL_INI["err_file"], msg)
}

// handleFatalError 함수는 발생한 치명적인 오류를 처리하고 로그에 기록합니다.
func handleFatalError(err interface{}) {
	msg := fmt.Sprintf("[%s] [Fatal Error] %v\n\n", time.Now().Format("2006-01-02 15:04:05"), err)
	writeToFile(SASEUL_INI["err_file"], msg)
}

// writeToFile 함수는 지정된 파일에 데이터를 씁니다.
func writeToFile(filename, data string) {
	f, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		fmt.Println("Error opening file:", err)
		return
	}
	defer f.Close()

	writer := bufio.NewWriter(f)
	_, err = writer.WriteString(data)
	if err != nil {
		fmt.Println("Error writing to file:", err)
	}
	writer.Flush()
}

// LoggerSetLogFile 함수는 로거의 로그 파일을 설정합니다.
func LoggerSetLogFile(logFile string) {
	// 로거의 로그 파일 설정 로직을 여기에 구현합니다.
}