main.go

package main

import (
	"flag"
	"fmt"

	"github.com/pritunl/pritunl-link/cmd"
	"github.com/pritunl/pritunl-link/constants"
	"github.com/pritunl/pritunl-link/logger"
	"github.com/pritunl/pritunl-link/requires"
)

const help = `
Usage: pritunl-link COMMAND

Commands:
  version                   Show version
  start                     Start link service
  add                       Add a Pritunl server URI
  remove                    Remove a Pritunl server URI
  clear                     Clear all configured Pritunl server URIs
  list                      List Pritunl server URIs
  default-interface         Manually set default interface
  default-gateway           Manually set default gateway
  local-address             Manually set local IP address
  public-address            Manually set public IP address
  remove-routes-on          Remove unused routes from routing table
  remove-routes-off         Leave unused routes in routing table
  direct-ssh-on             Enable direct SSH
  direct-ssh-off            Disable direct SSH
  firewall-on               Allow access to ipsec ports only from other pritunl-link hosts
  firewall-off              Do not modify system firewall
  verify-on                 Enable HTTPS certificate verification when connecting to Pritunl server
  verify-off                Disable HTTPS certificate verification when connecting to Pritunl server
  host-check-on             Enable link host checking
  host-check-off            Disable disable link host checking
  disconnected-timeout-on   Enable restart when disconnected for duration of timeout
  disconnected-timeout-off  Disable restart when disconnected for duration of timeout
  advertise-update-on       Enable recurring checks and updates of routing table and port forwarding
  advertise-update-off      Disable recurring checks and updates of routing table and port forwarding
  custom-option-add         Add custom ipsec option
  custom-option-clear       Clear custom ipsec options
  provider                  Manually set network provider
  oracle-user-ocid          Set Oracle user ocid
  oracle-private-key        Set Oracle base64 private key
  oracle-region             Set Oracle region
  oracle-tenancy-ocid       Set Oracle tenancy ocid
  oracle-compartment-ocid   Set Oracle compartment ocid
  oracle-vnic-ocid          Set Oracle vnic ocid
  oracle-private-ip-ocid    Set Oracle private IP ocid
  unifi-username            Set Unifi username
  unifi-password            Set Unifi password
  unifi-controller          Set URL of Unifi controller
  unifi-site                Set the Unifi site if different then default
  unifi-interface           Set the Unifi forward interface if different then default
  unifi-port-on             Enable automatic port forwarding on Unifi
  unifi-port-off            Disable automatic port forwarding on Unifi
  edge-username             Set EdgeRouter username
  edge-password             Set EdgeRouter password
  edge-hostname             Set hostname of EdgeRouter
  edge-port-on              Enable automatic port forwarding on EdgeRouter
  edge-port-off             Disable automatic port forwarding on EdgeRouter
  pritunl-hostname          Set hostname of Pritunl Cloud server
  pritunl-organization      Set Pritunl Cloud organization ID
  pritunl-vpc               Set Pritunl Cloud VPC ID
  pritunl-token             Set Pritunl Cloud token
  pritunl-secret            Set Pritunl Cloud secret
  hetzner-token             Set Hetzner token
  hetzner-network-id        Set Hetzner network id
`

func Init() {
	logger.Init()
	requires.Init()
}

func main() {
	flag.Usage = func() {
		fmt.Println(help)
	}

	flag.Parse()

	switch flag.Arg(0) {
	case "version":
		fmt.Printf("pritunl-link v%s\n", constants.Version)
		break
	case "start":
		Init()
		err := cmd.Start()
		if err != nil {
			panic(err)
		}
		break
	case "add":
		Init()
		err := cmd.Add(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "remove":
		Init()
		err := cmd.Remove(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "clear":
		Init()
		err := cmd.Clear()
		if err != nil {
			panic(err)
		}
		break
	case "list":
		Init()
		err := cmd.List()
		if err != nil {
			panic(err)
		}
		break
	case "default-interface":
		Init()
		err := cmd.DefaultInterface(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "default-gateway":
		Init()
		err := cmd.DefaultGateway(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "local-address":
		Init()
		err := cmd.LocalAddress(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "public-address":
		Init()
		err := cmd.PublicAddress(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "remove-routes-on":
		Init()
		err := cmd.RemoveRoutesOn()
		if err != nil {
			panic(err)
		}
		break
	case "remove-routes-off":
		Init()
		err := cmd.RemoveRoutesOff()
		if err != nil {
			panic(err)
		}
		break
	case "direct-ssh-on":
		Init()
		err := cmd.DirectSshOn()
		if err != nil {
			panic(err)
		}
		break
	case "direct-ssh-off":
		Init()
		err := cmd.DirectSshOff()
		if err != nil {
			panic(err)
		}
		break
	case "firewall-on":
		Init()
		err := cmd.FirewallOn()
		if err != nil {
			panic(err)
		}
		break
	case "firewall-off":
		Init()
		err := cmd.FirewallOff()
		if err != nil {
			panic(err)
		}
		break
	case "verify-on":
		Init()
		err := cmd.VerifyOn()
		if err != nil {
			panic(err)
		}
		break
	case "verify-off":
		Init()
		err := cmd.VerifyOff()
		if err != nil {
			panic(err)
		}
		break
	case "host-check-on":
		Init()
		err := cmd.HostCheckOn()
		if err != nil {
			panic(err)
		}
		break
	case "host-check-off":
		Init()
		err := cmd.HostCheckOff()
		if err != nil {
			panic(err)
		}
		break
	case "disconnected-timeout-on":
		Init()
		err := cmd.DisconnectedTimeoutOn()
		if err != nil {
			panic(err)
		}
		break
	case "disconnected-timeout-off":
		Init()
		err := cmd.DisconnectedTimeoutOff()
		if err != nil {
			panic(err)
		}
		break
	case "advertise-update-on":
		Init()
		err := cmd.AdvertiseUpdateOn()
		if err != nil {
			panic(err)
		}
		break
	case "advertise-update-off":
		Init()
		err := cmd.AdvertiseUpdateOff()
		if err != nil {
			panic(err)
		}
		break
	case "custom-option-add":
		Init()
		err := cmd.AddCustomOption(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "custom-option-clear":
		Init()
		err := cmd.ClearCustomOptions()
		if err != nil {
			panic(err)
		}
		break
	case "provider":
		Init()
		err := cmd.Provider(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-username":
		Init()
		err := cmd.UnifiUsername(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-password":
		Init()
		err := cmd.UnifiPassword(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-controller":
		Init()
		err := cmd.UnifiController(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-site":
		Init()
		err := cmd.UnifiSite(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-interface":
		Init()
		err := cmd.UnifiInterface(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "unifi-port-on":
		Init()
		err := cmd.UnifiPortOn()
		if err != nil {
			panic(err)
		}
		break
	case "unifi-port-off":
		Init()
		err := cmd.UnifiPortOff()
		if err != nil {
			panic(err)
		}
		break
	case "edge-username":
		Init()
		err := cmd.EdgeUsername(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "edge-password":
		Init()
		err := cmd.EdgePassword(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "edge-hostname":
		Init()
		err := cmd.EdgeHostname(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "edge-port-on":
		Init()
		err := cmd.EdgePortOn()
		if err != nil {
			panic(err)
		}
		break
	case "edge-port-off":
		Init()
		err := cmd.EdgePortOff()
		if err != nil {
			panic(err)
		}
		break
	case "pritunl-hostname":
		Init()
		err := cmd.PritunlHostname(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "pritunl-organization":
		Init()
		err := cmd.PritunlOrganization(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "pritunl-vpc":
		Init()
		err := cmd.PritunlVpc(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "pritunl-token":
		Init()
		err := cmd.PritunlToken(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "pritunl-secret":
		Init()
		err := cmd.PritunlSecret(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-region":
		Init()
		err := cmd.OracleRegion(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-private-key":
		Init()
		err := cmd.OraclePrivateKey(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-user-ocid":
		Init()
		err := cmd.OracleUserOcid(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-tenancy-ocid":
		Init()
		err := cmd.OracleTenancyOcid(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-compartment-ocid":
		Init()
		err := cmd.OracleCompartmentOcid(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "oracle-vnic-ocid":
		Init()
		err := cmd.OracleVnicOcid(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "hetzner-token":
		Init()
		err := cmd.HetznerToken(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	case "hetzner-network-id":
		Init()
		err := cmd.HetznerNetworkId(flag.Arg(1))
		if err != nil {
			panic(err)
		}
		break
	default:
		fmt.Println(help)
	}
}