main.go

// Copyright (c) Liam Stanley <liam@liam.sh>. All rights reserved. Use of
// this source code is governed by the MIT license that can be found in
// the LICENSE file.

package main

import (
	"context"

	"github.com/apex/log"
	"github.com/lrstanley/chix"
	"github.com/lrstanley/clix"
	"github.com/lrstanley/geoip/internal/dns"
	"github.com/lrstanley/geoip/internal/lookup"
	"github.com/lrstanley/geoip/internal/models"
)

// Should be automatically added by goreleaser.
var (
	version = "master"
	commit  = "master"
	date    = "unknown"
)

var (
	logger log.Interface
	cli    = &clix.CLI[models.Flags]{
		Links: clix.GithubLinks("github.com/lrstanley/geoip", "master", "https://liam.sh"),
		VersionInfo: &clix.VersionInfo[models.Flags]{
			Version: version,
			Commit:  commit,
			Date:    date,
		},
	}

	lookupSvc *lookup.Service
)

func main() {
	cli.Parse()
	logger = cli.Logger

	ctx := log.NewContext(context.Background(), logger)
	resolver := dns.NewResolver(cli.Flags.DNS)
	lookupSvc = lookup.NewService(ctx, logger, cli.Flags.DB, resolver)

	geoIPUpdater := lookup.NewUpdater(cli.Flags.DB, logger, lookupSvc, models.DatabaseGeoIP)
	asnUpdater := lookup.NewUpdater(cli.Flags.DB, logger, lookupSvc, models.DatabaseASN)

	if err := chix.RunCtx(
		ctx,
		httpServer(ctx),
		geoIPUpdater.Start,
		asnUpdater.Start,
	); err != nil {
		logger.WithError(err).Fatal("shutting down")
	}
}