go-nd-portal/cmd/main.go

package cmd

import (
	"flag"
	"fmt"
	"net"
	"net/netip"
	"os"
	"runtime"

	"golang.org/x/term"

	"github.com/sirupsen/logrus"

	"github.com/fumiama/go-nd-portal/helper"
	"github.com/fumiama/go-nd-portal/portal"
)

func outip() (net.IP, error) {
	conn, err := net.Dial("udp", "8.8.8.8:53")
	if err != nil {
		return nil, err
	}
	_ = conn.Close()
	return conn.LocalAddr().(*net.UDPAddr).IP.To4(), nil
}

func line() int {
	_, _, fileLine, ok := runtime.Caller(1)
	if ok {
		return fileLine
	}
	return -1
}

const query = "query"

// Main cmd program
func Main() {
	ip, err := outip()
	ipf := ""
	if err != nil {
		ipf = query
	} else {
		ipf = ip.String()
	}
	ips := flag.String("ip", ipf, "public IP")
	n := flag.String("n", query, "username")
	p := flag.String("p", query, "password")
	h := flag.Bool("h", false, "display this help")
	w := flag.Bool("w", false, "only display warn-or-higher-level log")
	d := flag.Bool("d", false, "display debug-level log")
	x := flag.Bool("x", false, "do dx login")
	flag.Parse()
	if *h {
		fmt.Println("Usage:")
		flag.PrintDefaults()
		os.Exit(0)
	}
	if *d {
		logrus.SetLevel(logrus.DebugLevel)
	} else if *w {
		logrus.SetLevel(logrus.WarnLevel)
	}
	if *ips == query {
		fmt.Printf("ip: ")
		_, err = fmt.Scanln(ips)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
	}
	if *ips != ip.String() {
		ipaddr, err := netip.ParseAddr(*ips)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
		a4 := ipaddr.As4()
		copy(ip, a4[:])
	}
	if *n == query {
		fmt.Printf("username: ")
		_, err = fmt.Scanln(n)
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
	}
	if *p == query {
		fmt.Printf("password: ")
		data, err := term.ReadPassword(int(os.Stdin.Fd()))
		if err != nil {
			logrus.Errorln(err)
			os.Exit(line())
		}
		*p = helper.BytesToString(data)
		fmt.Println()
	}
	ptl, err := portal.NewPortal(*n, *p, ip)
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	u := portal.PortalGetChallenge
	if *x {
		u = portal.PortalGetChallengeDX
	}
	challenge, err := ptl.GetChallenge(u)
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	u = portal.PortalLogin
	dm := portal.PortalDomain
	if *x {
		u = portal.PortalLoginDX
		dm = portal.PortalDomainDX
	}
	err = ptl.Login(u, dm, challenge)
	if err != nil {
		logrus.Errorln(err)
		os.Exit(line())
	}
	logrus.Infoln("success")
}