128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"regexp"
|
|
"syscall"
|
|
|
|
"github.com/bedrock-tool/bedrocktool/utils"
|
|
|
|
_ "github.com/bedrock-tool/bedrocktool/subcommands"
|
|
_ "github.com/bedrock-tool/bedrocktool/subcommands/skins"
|
|
_ "github.com/bedrock-tool/bedrocktool/subcommands/world"
|
|
|
|
"github.com/google/subcommands"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var version string
|
|
|
|
func exit() {
|
|
logrus.Info("\nExiting\n")
|
|
for i := len(utils.G_exit) - 1; i >= 0; i-- { // go through cleanup functions reversed
|
|
utils.G_exit[i]()
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
func main() {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
if version != "" {
|
|
logrus.Infof("bedrocktool version: %s\n", version)
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
flag.BoolVar(&utils.G_debug, "debug", false, "debug mode")
|
|
flag.BoolVar(&utils.G_preload_packs, "preload", false, "preload resourcepacks for proxy")
|
|
enable_dns := flag.Bool("dns", false, "enable dns server for consoles")
|
|
println(utils.A)
|
|
println("")
|
|
subcommands.Register(subcommands.HelpCommand(), "")
|
|
subcommands.ImportantFlag("debug")
|
|
subcommands.ImportantFlag("dns")
|
|
subcommands.ImportantFlag("preload")
|
|
subcommands.HelpCommand()
|
|
|
|
{ // interactive input
|
|
if len(os.Args) < 2 {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
fmt.Println("Available commands:")
|
|
for name, desc := range utils.ValidCMDs {
|
|
fmt.Printf("\t%s\t%s\n", name, desc)
|
|
}
|
|
fmt.Printf("Use '%s <command>' to run a command\n", os.Args[0])
|
|
|
|
fmt.Printf("Input Command: ")
|
|
reader := bufio.NewReader(os.Stdin)
|
|
target, _ := reader.ReadString('\n')
|
|
r, _ := regexp.Compile(`[\n\r]`)
|
|
target = string(r.ReplaceAll([]byte(target), []byte("")))
|
|
os.Args = append(os.Args, target)
|
|
}
|
|
}
|
|
}
|
|
|
|
flag.Parse()
|
|
|
|
if *enable_dns {
|
|
utils.InitDNS()
|
|
}
|
|
|
|
// exit cleanup
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
go func() {
|
|
<-sigs
|
|
cancel()
|
|
exit()
|
|
}()
|
|
|
|
ret := subcommands.Execute(ctx)
|
|
exit()
|
|
os.Exit(int(ret))
|
|
}
|
|
|
|
type TransCMD struct {
|
|
auth bool
|
|
}
|
|
|
|
func (*TransCMD) Name() string { return "trans" }
|
|
func (*TransCMD) Synopsis() string { return "" }
|
|
|
|
func (c *TransCMD) SetFlags(f *flag.FlagSet) {
|
|
f.BoolVar(&c.auth, "auth", false, "if it should login to xbox")
|
|
}
|
|
|
|
func (c *TransCMD) Usage() string {
|
|
return c.Name() + ": " + c.Synopsis() + "\n"
|
|
}
|
|
|
|
func (c *TransCMD) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
|
|
const (
|
|
BLACK_FG = "\033[30m"
|
|
BOLD = "\033[1m"
|
|
BLUE = "\033[46m"
|
|
PINK = "\033[45m"
|
|
WHITE = "\033[47m"
|
|
RESET = "\033[0m"
|
|
)
|
|
if c.auth {
|
|
utils.GetTokenSource()
|
|
}
|
|
fmt.Println(BLACK_FG + BOLD + BLUE + " Trans " + PINK + " Rights " + WHITE + " Are " + PINK + " Human " + BLUE + " Rights " + RESET)
|
|
return 0
|
|
}
|
|
|
|
func init() {
|
|
utils.RegisterCommand(&TransCMD{})
|
|
}
|