hackathon/cmd/srv/main.go

67 lines
1.6 KiB
Go

package main
import (
"flag"
"fmt"
"io"
"log"
"net"
"os"
"github.com/hhhapz/codequest/api"
"github.com/hhhapz/codequest/db"
"github.com/hhhapz/codequest/models"
"github.com/peterbourgon/ff/v3"
"google.golang.org/grpc/grpclog"
// questions
_ "github.com/hhhapz/codequest/question/q01"
_ "github.com/hhhapz/codequest/question/q02"
)
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)
}
models.SetLogger(log.Printf)
oaStore, err := db.NewOAuthState(*secretFile)
if err != nil {
return fmt.Errorf("could not create oauth config: %w", err)
}
server, err := api.NewServer(oaStore, database, 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)
}
}