main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"os"
	"path/filepath"

	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"github.com/zulkhair/taxi-fares/controller/console"
	"github.com/zulkhair/taxi-fares/domain/config"
	"gopkg.in/yaml.v2"
)

func main() {
	// mode for future improvement
	var mode string
	flag.StringVar(&mode, "mode", "console", "console/http")
	flag.Parse()

	// Read Config file
	c := readConfigFile()
	if c == nil {
		log.Info().Msgf("Using default values")
	}

	// Setup log
	logFile := "log/app.log"
	if c != nil && c.Log.File != "" {
		logFile = c.Log.File
	}
	// Check if log file exists. if dir not exist then create
	if _, err := os.Stat(logFile); os.IsNotExist(err) {
		os.MkdirAll(filepath.Dir(logFile), 0700)
	}
	file, err := os.OpenFile(logFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0664)
	if err != nil {
		log.Fatal().Msgf("Cannot open log file, err : %v", err)
	}
	log.Logger = zerolog.New(file).With().Timestamp().Logger()

	// mode switch
	switch mode {
	case "http":
		startHttp(c)
	case "console":
		startConsole(c)
	default:
		log.Fatal().Msgf("Mode '%s' not found", mode)
	}
}

// readConfigFile is function to read and parse config file
func readConfigFile() *config.Config {
	// Open file
	file, err := os.Open("config.yaml")
	if err != nil {
		log.Error().Msgf("Cannot open config file, err : %v", err)
		return nil
	}
	// Read the content
	content, err := io.ReadAll(file)
	if err != nil {
		log.Error().Msgf("Cannot read config file, err : %v", err)
		return nil
	}
	// Parse the content
	var cfg config.Config
	err = yaml.Unmarshal(content, &cfg)
	if err != nil {
		log.Error().Msgf("Cannot unmarshal config file, err : %v", err)
		return nil
	}
	return &cfg
}

// startHttp is function to start and serve HTTP server
func startHttp(config *config.Config) {
	// Todo on the next improvement
	fmt.Println("HTTP mode coming soon, still on development")
	fmt.Println("Switching to console mode")
	startConsole(config)
}

// startConsole is function to start console mode, open stdin and print stdout
func startConsole(config *config.Config) {
	log.Info().Msgf("Starting console")

	// Create console instance
	c, err := console.New(config)
	if err != nil {
		log.Fatal().Msgf("Cannot start console, err : %v", err)
	}

	// Start the app
	err = c.StartApp()
	if err != nil {
		log.Error().Msg(err.Error())
		os.Exit(1)
	}
}