@zerops/zcli 1.0.0 → 8.0.1

package/src/command/startVpn/handler.go

@@ -1,150 +0,0 @@
-package startVpn
-
-import (
-	"context"
-	"errors"
-	"net"
-
-	"github.com/zerops-io/zcli/src/helpers"
-
-	"github.com/zerops-io/zcli/src/service/logger"
-	"github.com/zerops-io/zcli/src/service/storage"
-	"github.com/zerops-io/zcli/src/service/sudoers"
-	"github.com/zerops-io/zcli/src/zeropsApiProtocol"
-	"github.com/zerops-io/zcli/src/zeropsVpnProtocol"
-)
-
-const wireguardPort = "51820"
-const vpnApiGrpcPort = ":64510"
-
-type Config struct {
-	VpnAddress string
-	UserId     string
-}
-
-type RunConfig struct {
-	ProjectName string
-}
-
-type Handler struct {
-	config        Config
-	logger        logger.Logger
-	apiGrpcClient zeropsApiProtocol.ZeropsApiProtocolClient
-	sudoers       *sudoers.Handler
-	storage       *storage.Handler
-}
-
-func New(
-	config Config,
-	logger logger.Logger,
-	apiGrpcClient zeropsApiProtocol.ZeropsApiProtocolClient,
-	sudoers *sudoers.Handler,
-	storage *storage.Handler,
-) *Handler {
-	return &Handler{
-		config:        config,
-		logger:        logger,
-		apiGrpcClient: apiGrpcClient,
-		sudoers:       sudoers,
-		storage:       storage,
-	}
-}
-
-func (h *Handler) Run(ctx context.Context, config RunConfig) error {
-
-	if config.ProjectName == "" {
-		return errors.New("project name must be filled")
-	}
-
-	projectsResponse, err := h.apiGrpcClient.GetProjectsByName(ctx, &zeropsApiProtocol.GetProjectsByNameRequest{
-		Name: config.ProjectName,
-	})
-	if err := helpers.HandleGrpcApiError(projectsResponse, err); err != nil {
-		return err
-	}
-
-	projectsResponse.GetOutput().GetProjects()
-
-	projects := projectsResponse.GetOutput().GetProjects()
-	if len(projects) == 0 {
-		return errors.New("project not found")
-	}
-	if len(projects) > 1 {
-		return errors.New("there are multiple project with same name")
-	}
-	project := projects[0]
-
-	if h.storage.Data.ProjectId != "" && project.GetId() != h.storage.Data.ProjectId {
-		if h.isVpnAlive() {
-			return errors.New("vpn is started for another project, use stopVpn first")
-		}
-	}
-
-	err = h.cleanVpn()
-	if err != nil {
-		return err
-	}
-
-	publicKey, privateKey, err := h.generateKeys()
-	if err != nil {
-		return err
-	}
-
-	apiVpnRequestResponse, err := h.apiGrpcClient.PostVpnRequest(ctx, &zeropsApiProtocol.PostVpnRequestRequest{
-		Id:              project.GetId(),
-		ClientPublicKey: publicKey,
-	})
-	if err := helpers.HandleGrpcApiError(apiVpnRequestResponse, err); err != nil {
-		return err
-	}
-
-	expiry := zeropsApiProtocol.FromProtoTimestamp(apiVpnRequestResponse.GetOutput().GetExpiry())
-	signature := apiVpnRequestResponse.GetOutput().GetSignature()
-
-	ipRecords, err := net.LookupIP(h.config.VpnAddress)
-	if err != nil {
-		return err
-	}
-
-	vpnAddress := ""
-	for _, ip := range ipRecords {
-		vpnAddress = helpers.IpToString(ip)
-		break
-	}
-
-	vpnGrpcClient, closeFunc, err := h.startVpnClient(ctx, vpnAddress)
-	if err != nil {
-		return err
-	}
-	defer closeFunc()
-
-	startVpnResponse, err := vpnGrpcClient.StartVpn(ctx, &zeropsVpnProtocol.StartVpnRequest{
-		InstanceId:      project.GetId(),
-		UserId:          h.config.UserId,
-		ClientPublicKey: publicKey,
-		Signature:       signature,
-		Expiry:          zeropsVpnProtocol.ToProtoTimestamp(expiry),
-	})
-	if err := helpers.HandleVpnApiError(startVpnResponse, err); err != nil {
-		return err
-	}
-
-	err = h.setVpn(vpnAddress, privateKey, startVpnResponse)
-	if err != nil {
-		return err
-	}
-
-	if h.isVpnAlive() {
-		h.logger.Info("\nvpn connection is established\n")
-	} else {
-		return errors.New("vpn is not connected")
-	}
-
-	h.storage.Data.ProjectId = project.GetId()
-	err = h.storage.Save()
-	if err != nil {
-		return err
-	}
-
-	return nil
-}

package/src/command/startVpn/handler_clean_linux.go

@@ -1,24 +0,0 @@
-// +build linux
-
-package startVpn
-
-import (
-	"errors"
-	"os/exec"
-
-	"github.com/zerops-io/zcli/src/service/sudoers"
-)
-
-func (h *Handler) cleanVpn() error {
-
-	var err error
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "link", "del", "dev", "wg0"))
-	if err != nil {
-		if !errors.Is(err, sudoers.CannotFindDeviceErr) {
-			return err
-		}
-	}
-
-	return nil
-}

package/src/command/startVpn/handler_clean_linux_macos.go

@@ -1,19 +0,0 @@
-// +build darwin
-
-package startVpn
-
-import "os/exec"
-
-func (h *Handler) cleanVpn() error {
-
-	var err error
-
-	cmd := "ps aux | grep wireguard | grep -v grep | awk '{print $2}' | xargs sudo kill"
-
-	_, err = h.sudoers.RunCommand(exec.Command("bash", "-c", cmd))
-	if err != nil {
-		return err
-	}
-
-	return nil
-}

package/src/command/startVpn/handler_generateKeys.go

@@ -1,26 +0,0 @@
-package startVpn
-
-import (
-	"bytes"
-	"os/exec"
-)
-
-func (h *Handler) generateKeys() (public, private string, err error) {
-
-	privateKeyOutput, err := h.sudoers.RunCommand(exec.Command("wg", "genkey"))
-	if err != nil {
-		return
-	}
-	privateKey := privateKeyOutput[0 : len(privateKeyOutput)-1]
-
-	cmd := exec.Command("wg", "pubkey")
-	cmd.Stdin = bytes.NewReader(privateKey)
-	publicKeyOutput, err := h.sudoers.RunCommand(cmd)
-	if err != nil {
-		return
-	}
-
-	publicKey := publicKeyOutput[0 : len(publicKeyOutput)-1]
-
-	return string(publicKey), string(privateKey), nil
-}

package/src/command/startVpn/handler_isVpnAlive.go

@@ -1,23 +0,0 @@
-package startVpn
-
-import (
-	"os/exec"
-)
-
-func (h *Handler) isVpnAlive() bool {
-
-	if h.storage.Data.ServerIp == "" {
-		return false
-	}
-
-	for i := 0; i < 3; i++ {
-		_, err := exec.Command("ping6", h.storage.Data.ServerIp, "-c", "1").Output()
-		if err != nil {
-			continue
-		}
-
-		return true
-	}
-
-	return false
-}

package/src/command/startVpn/handler_setVpn_linux.go

@@ -1,97 +0,0 @@
-// +build linux
-
-package startVpn
-
-import (
-	"errors"
-	"io/ioutil"
-	"os"
-	"os/exec"
-	"path"
-	"strconv"
-
-	"github.com/google/uuid"
-	"github.com/zerops-io/zcli/src/service/sudoers"
-	"github.com/zerops-io/zcli/src/zeropsVpnProtocol"
-)
-
-func (h *Handler) setVpn(selectedVpnAddress, privateKey string, response *zeropsVpnProtocol.StartVpnResponse) error {
-	var err error
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "link", "add", "wg0", "type", "wireguard"))
-	if err != nil {
-		if !errors.Is(err, sudoers.IpAlreadySetErr) {
-			return err
-		}
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "link", "set", "mtu", "1420", "up", "dev", "wg0"))
-	if err != nil {
-		return err
-	}
-
-	{
-		privateKeyName := uuid.New().String()
-		tempPrivateKeyFile := path.Join(os.TempDir(), privateKeyName)
-		err = ioutil.WriteFile(tempPrivateKeyFile, []byte(privateKey), 0755)
-		if err != nil {
-			return err
-		}
-		_, err = h.sudoers.RunCommand(exec.Command("wg", "set", "wg0", "private-key", tempPrivateKeyFile))
-		if err != nil {
-			return err
-		}
-		err = os.Remove(tempPrivateKeyFile)
-		if err != nil {
-			return err
-		}
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "link", "set", "wg0", "up"))
-	if err != nil {
-		return err
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("wg", "set", "wg0", "listen-port", wireguardPort))
-	if err != nil {
-		return err
-	}
-
-	clientIp := zeropsVpnProtocol.FromProtoIP(response.GetVpn().GetAssignedClientIp())
-	serverIp := zeropsVpnProtocol.FromProtoIP(response.GetVpn().GetServerIp())
-	vpnRange := zeropsVpnProtocol.FromProtoIPRange(response.GetVpn().GetVpnIpRange())
-
-	args := []string{
-		"set", "wg0",
-		"peer", response.GetVpn().GetServerPublicKey(),
-		"allowed-ips", vpnRange.String(),
-		"endpoint", selectedVpnAddress + ":" + strconv.Itoa(int(response.GetVpn().GetPort())),
-		"persistent-keepalive", "25",
-	}
-	_, err = h.sudoers.RunCommand(exec.Command("wg", args...))
-	if err != nil {
-		if !errors.Is(err, sudoers.IpAlreadySetErr) {
-			panic(err)
-		}
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "-6", "address", "add", clientIp.String(), "dev", "wg0"))
-	if err != nil {
-		return err
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "route", "add", vpnRange.String(), "dev", "wg0"))
-	if err != nil {
-		return err
-	}
-
-	h.logger.Debug("assigned client address: " + clientIp.String())
-	h.logger.Debug("assigned vpn server: " + selectedVpnAddress + ":" + strconv.Itoa(int(response.GetVpn().GetPort())))
-	h.logger.Debug("server public key: " + response.GetVpn().GetServerPublicKey())
-	h.logger.Debug("serverIp address: " + serverIp.String())
-	h.logger.Debug("vpnRange: " + vpnRange.String())
-
-	h.storage.Data.ServerIp = serverIp.String()
-
-	return nil
-}

package/src/command/startVpn/handler_setVpn_macos.go

@@ -1,98 +0,0 @@
-// +build darwin
-
-package startVpn
-
-import (
-	"errors"
-	"fmt"
-	"io/ioutil"
-	"os"
-	"os/exec"
-	"path"
-	"regexp"
-	"strconv"
-
-	"github.com/zerops-io/zcli/src/service/sudoers"
-
-	"github.com/google/uuid"
-	"github.com/zerops-io/zcli/src/zeropsVpnProtocol"
-)
-
-func (h *Handler) setVpn(selectedVpnAddress, privateKey string, response *zeropsVpnProtocol.StartVpnResponse) error {
-	var err error
-
-	output, err := h.sudoers.RunCommand(exec.Command("wireguard-go", "utun"))
-	if err != nil {
-		return err
-	}
-
-	re := regexp.MustCompile(`INFO: \((.*)\)`)
-	submatches := re.FindSubmatch(output)
-	if len(submatches) != 2 {
-		return errors.New("vpn interface not found")
-	}
-
-	interfaceName := string(submatches[1])
-
-	{
-		privateKeyName := uuid.New().String()
-		tempPrivateKeyFile := path.Join(os.TempDir(), privateKeyName)
-
-		fmt.Println(tempPrivateKeyFile)
-		err = ioutil.WriteFile(tempPrivateKeyFile, []byte(privateKey), 0755)
-		if err != nil {
-			return err
-		}
-		_, err = h.sudoers.RunCommand(exec.Command("wg", "set", interfaceName, "private-key", tempPrivateKeyFile))
-		if err != nil {
-			return err
-		}
-		err = os.Remove(tempPrivateKeyFile)
-		if err != nil {
-			return err
-		}
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("wg", "set", interfaceName, "listen-port", wireguardPort))
-	if err != nil {
-		return err
-	}
-
-	clientIp := zeropsVpnProtocol.FromProtoIP(response.GetVpn().GetAssignedClientIp())
-	serverIp := zeropsVpnProtocol.FromProtoIP(response.GetVpn().GetServerIp())
-	vpnRange := zeropsVpnProtocol.FromProtoIPRange(response.GetVpn().GetVpnIpRange())
-
-	args := []string{
-		"set", interfaceName,
-		"peer", response.GetVpn().GetServerPublicKey(),
-		"allowed-ips", vpnRange.String(),
-		"endpoint", selectedVpnAddress + ":" + strconv.Itoa(int(response.GetVpn().GetPort())),
-		"persistent-keepalive", "25",
-	}
-	_, err = h.sudoers.RunCommand(exec.Command("wg", args...))
-	if err != nil {
-		if !errors.Is(err, sudoers.IpAlreadySetErr) {
-			panic(err)
-		}
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("ifconfig", interfaceName, "inet6", clientIp.String(), "mtu", "1420"))
-	if err != nil {
-		return err
-	}
-
-	_, err = h.sudoers.RunCommand(exec.Command("route", "add", "-inet6", vpnRange.String(), serverIp.String()))
-	if err != nil {
-		return err
-	}
-
-	h.logger.Debug("assigned client address: " + clientIp.String())
-	h.logger.Debug("assigned vpn server: " + selectedVpnAddress + ":" + strconv.Itoa(int(response.GetVpn().GetPort())))
-	h.logger.Debug("server public key: " + response.GetVpn().GetServerPublicKey())
-	h.logger.Debug("serverIp address: " + serverIp.String())
-	h.logger.Debug("vpnRange: " + vpnRange.String())
-
-	h.storage.Data.ServerIp = serverIp.String()
-
-	return nil
-}

package/src/command/startVpn/handler_startVpnClient.go

@@ -1,27 +0,0 @@
-package startVpn
-
-import (
-	"context"
-
-	"github.com/zerops-io/zcli/src/zeropsVpnProtocol"
-	"google.golang.org/grpc"
-)
-
-func (h *Handler) startVpnClient(ctx context.Context, targetAddress string) (_ zeropsVpnProtocol.ZeropsVpnProtocolClient, closeFunc func(), _ error) {
-
-	var dialOpts []grpc.DialOption
-	dialOpts = append(dialOpts, grpc.WithInsecure())
-
-	connection, err := grpc.DialContext(
-		ctx,
-		targetAddress+vpnApiGrpcPort,
-		dialOpts...,
-	)
-	if err != nil {
-		return nil, nil, err
-	}
-
-	return zeropsVpnProtocol.NewZeropsVpnProtocolClient(connection), func() {
-		_ = connection.Close()
-	}, nil
-}

package/src/command/stopVpn/handler.go

@@ -1,55 +0,0 @@
-package stopVpn
-
-import (
-	"context"
-
-	"github.com/zerops-io/zcli/src/service/logger"
-	"github.com/zerops-io/zcli/src/service/storage"
-	"github.com/zerops-io/zcli/src/service/sudoers"
-)
-
-type Config struct {
-}
-
-type RunConfig struct {
-}
-
-type Handler struct {
-	config  Config
-	logger  logger.Logger
-	sudoers *sudoers.Handler
-	storage *storage.Handler
-}
-
-func New(
-	config Config,
-	logger logger.Logger,
-	sudoers *sudoers.Handler,
-	storage *storage.Handler,
-) *Handler {
-	return &Handler{
-		config:  config,
-		logger:  logger,
-		sudoers: sudoers,
-		storage: storage,
-	}
-}
-
-func (h *Handler) Run(_ context.Context, _ RunConfig) error {
-
-	err := h.cleanVpn()
-	if err != nil {
-		return err
-	}
-
-	h.storage.Data.ProjectId = ""
-	h.storage.Data.ServerIp = ""
-	err = h.storage.Save()
-	if err != nil {
-		return err
-	}
-
-	h.logger.Info("\nvpn connection was closed\n")
-
-	return nil
-}

package/src/command/stopVpn/handler_clean_linux.go

@@ -1,24 +0,0 @@
-// +build linux
-
-package stopVpn
-
-import (
-	"errors"
-	"os/exec"
-
-	"github.com/zerops-io/zcli/src/service/sudoers"
-)
-
-func (h *Handler) cleanVpn() error {
-
-	var err error
-
-	_, err = h.sudoers.RunCommand(exec.Command("ip", "link", "del", "dev", "wg0"))
-	if err != nil {
-		if !errors.Is(err, sudoers.CannotFindDeviceErr) {
-			return err
-		}
-	}
-
-	return nil
-}

package/src/command/stopVpn/handler_clean_linux_macos.go

@@ -1,19 +0,0 @@
-// +build darwin
-
-package stopVpn
-
-import "os/exec"
-
-func (h *Handler) cleanVpn() error {
-
-	var err error
-
-	cmd := "ps aux | grep wireguard | grep -v grep | awk '{print $2}' | xargs sudo kill"
-
-	_, err = h.sudoers.RunCommand(exec.Command("bash", "-c", cmd))
-	if err != nil {
-		return err
-	}
-
-	return nil
-}

package/src/command/userInfo/handler.go

@@ -1,21 +0,0 @@
-package userInfo
-
-import (
-	"context"
-	"fmt"
-
-	"github.com/zerops-io/zcli/src/helpers"
-	"github.com/zerops-io/zcli/src/zeropsApiProtocol"
-)
-
-func Run(ctx context.Context, apiGrpcClient zeropsApiProtocol.ZeropsApiProtocolClient) error {
-
-	userInfoResponse, err := apiGrpcClient.GetUserInfo(ctx, &zeropsApiProtocol.GetUserInfoRequest{})
-	if err := helpers.HandleGrpcApiError(userInfoResponse, err); err != nil {
-		return err
-	}
-
-	fmt.Println(userInfoResponse.GetOutput())
-
-	return nil
-}

package/src/helpers/cmdRunner/run.go

@@ -1,40 +0,0 @@
-package cmdRunner
-
-import (
-	"bytes"
-	"errors"
-	"os/exec"
-	"strings"
-)
-
-var IpAlreadySetErr = errors.New("RTNETLINK answers: File exists")
-var CannotFindDeviceErr = errors.New(`Cannot find device "wg0"`)
-var OperationNotPermitted = errors.New(`Operation not permitted`)
-
-func Run(cmd *exec.Cmd) ([]byte, error) {
-	output := &bytes.Buffer{}
-	errOutput := &bytes.Buffer{}
-	cmd.Stdout = output
-	cmd.Stderr = errOutput
-
-	if err := cmd.Run(); err != nil {
-
-
-		if errOutput.Len() > 0 {
-			errOutputString := string(errOutput.Bytes()[0 : errOutput.Len()-1])
-
-			if strings.Contains(errOutputString, OperationNotPermitted.Error()) {
-				return nil, OperationNotPermitted
-			}
-
-			for _, e := range []error{IpAlreadySetErr, CannotFindDeviceErr} {
-				if errOutputString == e.Error() {
-					return nil, e
-				}
-			}
-		}
-		return nil, err
-	}
-
-	return output.Bytes(), nil
-}

package/src/helpers/grpcErrors.go

@@ -1,40 +0,0 @@
-package helpers
-
-import (
-	"errors"
-
-	"github.com/zerops-io/zcli/src/zeropsApiProtocol"
-	"github.com/zerops-io/zcli/src/zeropsVpnProtocol"
-)
-
-func HandleGrpcApiError(
-	response interface {
-		GetError() *zeropsApiProtocol.Error
-	},
-	err error,
-) error {
-	if err != nil {
-		return err
-	}
-	if response.GetError().GetCode() != zeropsApiProtocol.ErrorCode_NO_ERROR {
-		return errors.New(response.GetError().GetMessage())
-	}
-
-	return nil
-}
-
-func HandleVpnApiError(
-	response interface {
-		GetError() *zeropsVpnProtocol.Error
-	},
-	err error,
-) error {
-	if err != nil {
-		return err
-	}
-	if response.GetError().GetCode() != zeropsVpnProtocol.ErrorCode_NO_ERROR {
-		return errors.New(response.GetError().GetMessage())
-	}
-
-	return nil
-}

package/src/helpers/ip.go

@@ -1,11 +0,0 @@
-package helpers
-
-import "net"
-
-func IpToString(ip net.IP) string {
-	if ip.To16() != nil {
-		return "[" + ip.String() + "]"
-	}
-
-	return ip.String()
-}