astro-domu/main.go

package main

import (
	"os"
	"time"

	"github.com/jordanorelli/belt-mud/internal/exit"
	"github.com/jordanorelli/blammo"
)

func newLog(path string) *blammo.Log {
	f, err := os.OpenFile("./belt.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
	if err != nil {
		exit.WithMessage(1, "unable to open log file %q for writing: %v", err)
	}

	w := blammo.NewLineWriter(f)

	options := []blammo.Option{
		blammo.DebugWriter(w),
		blammo.InfoWriter(w),
		blammo.ErrorWriter(w),
	}

	return blammo.NewLog("belt", options...)
}

func main() {
	if len(os.Args) < 2 {
		exit.WithMessage(1, "client or server?")
	}

	switch os.Args[1] {
	case "client":
		runClient()
	case "server":
		stdout := blammo.NewLineWriter(os.Stdout)
		stderr := blammo.NewLineWriter(os.Stderr)

		options := []blammo.Option{
			blammo.DebugWriter(stdout),
			blammo.InfoWriter(stdout),
			blammo.ErrorWriter(stderr),
		}

		log := blammo.NewLog("belt", options...).Child("server")
		s := server{Log: log, host: "127.0.0.1", port: 12805}
		err := s.listen()
		log.Error("listen error: %v", err)
	default:
		exit.WithMessage(1, "supported options are [client|server]")
	}

}

func runClient() {
	log := newLog("./belt.log").Child("client")

	start := time.Now()
	log.Info("starting at: %v", start)
	defer func() {
		finished := time.Now()
		log.Info("finished at: %v", finished)
		log.Info("total play time: %v", finished.Sub(start))
	}()

	ui := ui{
		Log: log.Child("ui"),
	}
	ui.run()
}