gitivity/main.go

package main

import (
	"fmt"
	"os"
	"strings"

	"code.gitea.io/sdk/gitea"
	"github.com/akamensky/argparse"
)

// Arguments is a place to store arguments / options / flags passed from the command line.
type Arguments struct {
	Times struct {
	}
	Summary struct {
	}
	Feed struct {
	}
	global struct {
		config_path *string
		server      *string
	}
}

var arguments Arguments
var config Configuration
var Servers map[string]*gitea.Client

func main() {
	parser := argparse.NewParser("gitivity", "Command line tool to get Gitea statistics")
	arguments.global.config_path = parser.String("c", "config", &argparse.Options{
		Required: false,
		Help:     "Configuration file",
		Default:  "./config.lua",
	})
	arguments.global.server = parser.String("s", "server", &argparse.Options{
		Required: false,
		Help:     "Specific server to use",
	})

	Times := parser.NewCommand("times", "Get a user's tracked times.")
	Times.Flag("t", "today", &argparse.Options{})
	Summary := parser.NewCommand("summary", "Display a summary of a user's activity.")
	Feed := parser.NewCommand("feed", "Display the user's activity feed.")

	parse_err := parser.Parse(os.Args)
	if parse_err != nil {
		fmt.Print(parser.Usage(parse_err))
		return
	}

	config = Configuration{}
	Servers = make(map[string]*gitea.Client)
	if err := config.Parse(*arguments.global.config_path); err != nil {
		panic("Failed to parse configuration file")
	}

	for _, server := range config.servers {
		if *arguments.global.server != "" && strings.ToLower(server.servername) != strings.ToLower(*arguments.global.server) {
			continue
		}
		client_opts := []gitea.ClientOption{
			gitea.SetToken(server.token),
		}
		client, err := gitea.NewClient(server.url, client_opts...)
		if err != nil {
			fmt.Printf("Failed to create Gitea client! (%s)\n", err)
			os.Exit(1)
		}
		Servers[server.servername] = client
	}
	if len(Servers) == 0 {
		println("No servers configured / specified")
		os.Exit(0)
	}

	if Times.Happened() {
		times()
	} else if Summary.Happened() {
		summary()
	} else if Feed.Happened() {
		feed()
	}
}