Add Add GCE support and GCE provisioning files to pkg

Adds GCE support for exit-node and will aid to merge the PR inlets/inlets-operator#33 Signed-off-by: Utsav Anand <[email protected]>
inlets · Nov 30, 2019 · 3bf1b42 · 3bf1b42
1 parent 9df2ee7
commit 3bf1b42
Show file tree

Hide file tree

Showing 708 changed files with 640,452 additions and 11 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/Gopkg.toml b/Gopkg.toml
@@ -41,3 +41,12 @@
 [prune]
   go-tests = true
   unused-packages = true
+
+[[constraint]]
+  name = "google.golang.org/api"
+  version = "0.14.0"
+
+[[override]]
+  name = "golang.org/x/sys"
+  branch = "master"
+
diff --git a/cmd/create.go b/cmd/create.go
@@ -6,6 +6,7 @@ package cmd
 import (
 	"fmt"
 	"io/ioutil"
+	"strconv"
 	"strings"
 	"time"
 
@@ -20,8 +21,9 @@ import (
 
 func init() {
 	inletsCmd.AddCommand(createCmd)
-	createCmd.Flags().StringP("provider", "p", "digitalocean", "The cloud provider - digitalocean, packet, scaleway, or civo")
+	createCmd.Flags().StringP("provider", "p", "digitalocean", "The cloud provider - digitalocean, gce, packet, scaleway, or civo")
 	createCmd.Flags().StringP("region", "r", "lon1", "The region for your cloud provider")
+	createCmd.Flags().StringP("zone", "z", "us-central1-a", "The zone for the exit node (Google Compute Engine)")
 
 	createCmd.Flags().StringP("inlets-token", "t", "", "The inlets auth token for your exit node")
 	createCmd.Flags().StringP("access-token", "a", "", "The access token for your cloud")
@@ -30,7 +32,7 @@ func init() {
 	createCmd.Flags().String("secret-key", "", "The access token for your cloud (Scaleway)")
 	createCmd.Flags().String("secret-key-file", "", "Read this file for the access token for your cloud (Scaleway)")
 	createCmd.Flags().String("organisation-id", "", "Organisation ID (Scaleway)")
-	createCmd.Flags().String("project-id", "", "Project ID (Packet.com)")
+	createCmd.Flags().String("project-id", "", "Project ID (Packet.com, Google Compute Engine)")
 
 	createCmd.Flags().StringP("remote-tcp", "c", "", `Remote host for inlets-pro to use for forwarding TCP connections`)
 
@@ -97,6 +99,11 @@ func runCreate(cmd *cobra.Command, _ []string) error {
 		region = "ams1"
 	}
 
+	var zone string
+	if provider == "gce" {
+		zone, err = cmd.Flags().GetString("zone")
+	}
+
 	var secretKey string
 	var organisationID string
 	if provider == "scaleway" {
@@ -129,12 +136,17 @@ func runCreate(cmd *cobra.Command, _ []string) error {
 
 	projectID, _ := cmd.Flags().GetString("project-id")
 
-	hostReq, err := createHost(provider, name, region, projectID, userData)
+	hostReq, err := createHost(provider, name, region, zone, projectID, userData, strconv.Itoa(inletsControlPort))
 	if err != nil {
 		return err
 	}
 
-	fmt.Printf("Requesting host: %s in %s, from %s\n", name, region, provider)
+	if provider == "gce" {
+		fmt.Printf("Requesting host: %s in %s, from %s\n", name, zone, provider)
+	} else {
+		fmt.Printf("Requesting host: %s in %s, from %s\n", name, region, provider)
+	}
+
 	hostRes, err := provisioner.Provision(*hostReq)
 	if err != nil {
 		return err
@@ -201,6 +213,8 @@ func getProvisioner(provider, accessToken, secretKey, organisationID, region str
 		return provision.NewCivoProvisioner(accessToken)
 	} else if provider == "scaleway" {
 		return provision.NewScalewayProvisioner(accessToken, secretKey, organisationID, region)
+	} else if provider == "gce" {
+		return provision.NewGCEProvisioner(accessToken)
 	}
 	return nil, fmt.Errorf("no provisioner for provider: %s", provider)
 }
@@ -210,7 +224,7 @@ func generateAuth() (string, error) {
 	return pwdRes, pwdErr
 }
 
-func createHost(provider, name, region, projectID, userData string) (*provision.BasicHost, error) {
+func createHost(provider, name, region, zone, projectID, userData, inletsPort string) (*provision.BasicHost, error) {
 	if provider == "digitalocean" {
 		return &provision.BasicHost{
 			Name:       name,
@@ -249,6 +263,20 @@ func createHost(provider, name, region, projectID, userData string) (*provision.
 			UserData:   userData,
 			Additional: map[string]string{},
 		}, nil
+	} else if provider == "gce" {
+		return &provision.BasicHost{
+			Name:     name,
+			OS:       "projects/debian-cloud/global/images/debian-9-stretch-v20191121",
+			Plan:     "f1-micro",
+			Region:   "",
+			UserData: userData,
+			Additional: map[string]string{
+				"projectid":     projectID,
+				"zone":          zone,
+				"firewall-name": "inlets",
+				"firewall-port": inletsPort,
+			},
+		}, nil
 	}
 
 	return nil, fmt.Errorf("no provisioner for provider: %q", provider)

diff --git a/cmd/delete.go b/cmd/delete.go
@@ -12,7 +12,7 @@ import (
 
 func init() {
 	inletsCmd.AddCommand(deleteCmd)
-	deleteCmd.Flags().StringP("provider", "p", "digitalocean", "The cloud provider")
+	deleteCmd.Flags().StringP("provider", "p", "digitalocean", "The cloud provider - digitalocean, gce, packet, scaleway, or civo")
 
 	deleteCmd.Flags().StringP("inlets-token", "t", "", "The inlets auth token for your exit node")
 	deleteCmd.Flags().StringP("access-token", "a", "", "The access token for your cloud")