2023-05-18 08:55:50 -04:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
2023-06-04 16:21:41 -04:00
|
|
|
"path/filepath"
|
2023-05-18 08:55:50 -04:00
|
|
|
|
2023-06-04 16:21:41 -04:00
|
|
|
"github.com/Nathan13888/mirror-checker2/v2/checkers"
|
2023-05-18 08:55:50 -04:00
|
|
|
"github.com/Nathan13888/mirror-checker2/v2/config"
|
|
|
|
"github.com/Nathan13888/mirror-checker2/v2/web"
|
2023-06-04 16:21:41 -04:00
|
|
|
"github.com/rs/zerolog/log"
|
2023-05-18 08:55:50 -04:00
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
)
|
|
|
|
|
2023-06-04 16:21:41 -04:00
|
|
|
func getAbsPath(path string) string {
|
|
|
|
absPath, err := filepath.Abs(path)
|
|
|
|
if err != nil {
|
|
|
|
log.Error().Err(err).Msg("Failed to get absolute path.")
|
|
|
|
return path
|
|
|
|
}
|
|
|
|
return absPath
|
|
|
|
}
|
|
|
|
|
2023-05-18 08:55:50 -04:00
|
|
|
func main() {
|
2023-06-04 16:21:41 -04:00
|
|
|
var err error
|
|
|
|
config.SetupLogger(true) // TODO: flag for debug mode
|
|
|
|
|
2023-05-18 08:55:50 -04:00
|
|
|
// Start CLI
|
|
|
|
// TODO: documentation - https://cli.urfave.org/v2/examples/full-api-example/
|
|
|
|
app := &cli.App{
|
|
|
|
Name: "CSC Mirror Checker 2",
|
|
|
|
Usage: "sees if the mirror is up!",
|
|
|
|
Version: config.BuildVersion,
|
|
|
|
EnableBashCompletion: true,
|
|
|
|
// https://cli.urfave.org/v2/examples/combining-short-options/
|
2023-06-04 16:21:41 -04:00
|
|
|
// TODO: flags for config file (mirrors.json), defaults to mirrors.json
|
2023-05-18 08:55:50 -04:00
|
|
|
Commands: []*cli.Command{
|
|
|
|
{
|
2023-06-04 16:21:41 -04:00
|
|
|
Name: "daemon",
|
|
|
|
Aliases: []string{"s", "d", "serve", "web", "server"},
|
2023-05-18 08:55:50 -04:00
|
|
|
Usage: "starts web API",
|
|
|
|
Action: func(cCtx *cli.Context) error {
|
|
|
|
// TODO: flags for port, listen address
|
2023-06-04 16:21:41 -04:00
|
|
|
// TODO: flag --config or --stdin-config
|
|
|
|
|
|
|
|
// TODO: enable all projects by default
|
|
|
|
// checkers.LoadDefaultProjects()
|
|
|
|
|
|
|
|
path := getAbsPath("data/mirrors.json")
|
|
|
|
log.Info().Str("path", path).Msg("Loading config file.")
|
|
|
|
err = config.LoadFromFile(path)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal().Err(err).Msg("Failed to load config file.")
|
|
|
|
}
|
2023-05-18 08:55:50 -04:00
|
|
|
|
|
|
|
return web.StartServer()
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Name: "check",
|
|
|
|
Aliases: []string{"c"},
|
2023-06-04 16:21:41 -04:00
|
|
|
Usage: "checks particular mirror (once)",
|
2023-05-18 08:55:50 -04:00
|
|
|
Action: func(cCtx *cli.Context) error {
|
2023-06-04 16:21:41 -04:00
|
|
|
checkers.LoadDefaultProjects()
|
|
|
|
|
|
|
|
// verify all projects are enabled
|
|
|
|
// TODO:
|
|
|
|
|
|
|
|
// attempt to look up and check all projects
|
|
|
|
for _, arg := range cCtx.Args().Slice() {
|
|
|
|
log.Printf("\nPulling project information for '%s'\n\n", arg)
|
|
|
|
proj, err := checkers.GetProject(arg)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal().Err(err).
|
|
|
|
Str("project", arg).
|
|
|
|
Msg("Failed to load project.")
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
proj.RunChecksAsync()
|
|
|
|
}
|
|
|
|
|
2023-05-18 08:55:50 -04:00
|
|
|
return nil
|
|
|
|
},
|
|
|
|
// TODO: auto complete available mirrors, https://cli.urfave.org/v2/examples/combining-short-options/
|
|
|
|
// BashComplete: func(cCtx *cli.Context) {
|
|
|
|
// // This will complete if no args are passed
|
|
|
|
// if cCtx.NArg() > 0 {
|
|
|
|
// return
|
|
|
|
// }
|
|
|
|
// for _, t := range config.Mirrors {
|
2023-06-04 16:21:41 -04:00
|
|
|
// log.Println(t)
|
2023-05-18 08:55:50 -04:00
|
|
|
// }
|
|
|
|
// },
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
2023-06-04 16:21:41 -04:00
|
|
|
log.Fatal().Err(err).Msg("An error occurred.")
|
2023-05-18 08:55:50 -04:00
|
|
|
}
|
|
|
|
}
|