main.go

// Copyright 2023 tsuru authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"log"
	"net"
	"strings"

	"github.com/spf13/cobra"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
	"github.com/tsuru/acl-api/api/version"
	"github.com/tsuru/acl-plugin/cmd"
)

func initConfig(rootCmd *cobra.Command) func() {
	return func() {
		viper.SetConfigName(".acl")
		viper.AddConfigPath("$HOME")
		viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
		viper.AutomaticEnv()

		if err := viper.BindPFlags(rootCmd.PersistentFlags()); err != nil {
			log.Fatalf("unable to bind flags to viper: %v\n", err)
		}

		if err := viper.ReadInConfig(); err == nil {
			log.Printf("Using config file: %v", viper.ConfigFileUsed())
		}
	}
}

func main() {
	rootCmd := &cobra.Command{
		Version: version.Version,
	}
	cobra.OnInitialize(initConfig(rootCmd))
	rulesCmd := &cobra.Command{
		Use: "rules",
	}
	rootCmd.AddCommand(rulesCmd)
	rulesCmd.AddCommand(cmd.AddRuleCmd)
	rulesCmd.AddCommand(cmd.RemoveRuleCmd)
	rulesCmd.AddCommand(cmd.ListRuleCmd)
	rulesCmd.AddCommand(cmd.ForceSyncCmd)
	rulesCmd.AddCommand(cmd.SyncDNSCmd)

	adminCmd := &cobra.Command{
		Use: "admin",
	}
	rootCmd.AddCommand(adminCmd)
	adminCmd.AddCommand(cmd.ListAllRulesCmd)
	adminCmd.AddCommand(cmd.AddCustomRuleCmd)

	rootCmd.PersistentFlags().String("tsuru.target", "", "Tsuru Target URL")
	rootCmd.PersistentFlags().String("tsuru.token", "", "Tsuru Token")

	dstFlags := pflag.NewFlagSet("", pflag.ExitOnError)
	dstFlags.IPNet("ip", net.IPNet{}, "Destination IP Network [10.0.0.1/32]")
	dstFlags.String("dns", "", "Destination DNS name [example.org]")
	dstFlags.String("app", "", "Destination Tsuru App Name [myapp]")
	dstFlags.String("app-pool", "", "Destination Tsuru Pool Name [dev]")
	dstFlags.String("rpaas", "", "Destination RPAAS ServiceName/Instance [rpaasv2-be/myrpaas]")
	dstFlags.String("service", "", "Destination Kubernetes Service [namespace/service]")
	dstFlags.StringSlice("port", nil, "Destination Ports [tcp:443]")

	adminFlags := pflag.NewFlagSet("", pflag.ExitOnError)
	adminFlags.AddFlagSet(dstFlags)

	adminFlags.String("src-app", "", "Source Tsuru App Name [myapp]")
	adminFlags.String("src-app-pool", "", "Source Tsuru Pool Name [dev]")
	adminFlags.String("owner", "", "Rule owner")

	cmd.AddRuleCmd.Flags().AddFlagSet(dstFlags)
	cmd.AddCustomRuleCmd.Flags().AddFlagSet(adminFlags)

	cmd.ListRuleCmd.Flags().Bool("show-sync", false, "Show rules latest sync attempt")
	cmd.ListRuleCmd.Flags().Bool("show-extra-sync", false, "Show rules with latest sync attempt details.")
	cmd.ListRuleCmd.Flags().Bool("json", false, "Return the raw JSON output instead of the formatted table")
	cmd.ListAllRulesCmd.Flags().Bool("json", false, "Return the raw JSON output instead of the formatted table")
	cmd.ListAllRulesCmd.Flags().Bool("show-extra-sync", false, "Show rules with latest sync attempt details.")

	if err := rootCmd.Execute(); err != nil {
		log.Fatal(err)
	}
}