hackathon/cmd/srv/main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"net"
	"os"

	"github.com/hhhapz/codequest/api"
	"github.com/hhhapz/codequest/db"
	"github.com/peterbourgon/ff/v3"
	"google.golang.org/grpc/grpclog"
)

func run() error {
	fs := flag.NewFlagSet("codequest", flag.ExitOnError)
	port := fs.Int("port", 10000, "GRPC Server Port")
	secretFile := fs.String("secret", "client.secret.json", "Path to google oauth2 secret credentials JSON file.")
	dbFile := fs.String("db", "db.sqlite", "Path to sqlite3 database file")
	if err := ff.Parse(fs, os.Args[1:], ff.WithEnvVarPrefix("HK")); err != nil {
		return err
	}

	if *secretFile == "" {
		return fmt.Errorf("location to secret credentials file not provided.\nTry -h")
	}

	database, err := db.NewDB(*dbFile)
	if err != nil {
		return fmt.Errorf("could not open db: %v", err)
	}

	oaStore, err := db.NewOAuthState(*secretFile)
	if err != nil {
		return fmt.Errorf("could not create oauth config: %w", err)
	}

	server, err := api.NewServer(oaStore, database)

	log := grpclog.NewLoggerV2(os.Stderr, io.Discard, os.Stderr)
	grpclog.SetLoggerV2(log)

	addr := fmt.Sprintf("0.0.0.0:%d", *port)
	lis, err := net.Listen("tcp", addr)
	if err != nil {
		return fmt.Errorf("failed to listen: %v", err)
	}

	log.Info("Serving gRPC on http://", addr)
	return server.Serve(lis)
}

func main() {
	if err := run(); err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}