Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add kfwd command #11

Merged
merged 3 commits into from
Nov 29, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion .travis.yml
Original file line number Diff line number Diff line change
@@ -1,9 +1,12 @@
sudo: required
language: go

go:
- '1.11'
- 1.12.x

script:
- make dist

before_deploy:
- "./ci/hashgen.sh"

Expand All @@ -25,3 +28,6 @@ deploy:
skip_cleanup: true
on:
tags: true

env:
- GO111MODULE=off
9 changes: 9 additions & 0 deletions Gopkg.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

31 changes: 4 additions & 27 deletions Gopkg.toml
Original file line number Diff line number Diff line change
@@ -1,30 +1,3 @@
# Gopkg.toml example
#
# Refer to https://golang.github.io/dep/docs/Gopkg.toml.html
# for detailed Gopkg.toml documentation.
#
# required = ["github.com/user/thing/cmd/thing"]
# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
#
# [[constraint]]
# name = "github.com/user/project"
# version = "1.0.0"
#
# [[constraint]]
# name = "github.com/user/project2"
# branch = "dev"
# source = "github.com/myfork/project2"
#
# [[override]]
# name = "github.com/x/y"
# version = "2.4.0"
#
# [prune]
# non-go = false
# go-tests = true
# unused-packages = true


[[constraint]]
name = "github.com/digitalocean/godo"
version = "1.27.0"
Expand Down Expand Up @@ -61,6 +34,10 @@
branch = "master"
name = "golang.org/x/oauth2"

[[constraint]]
name = "github.com/alexellis/go-execute"
version = "0.2.1"

[prune]
go-tests = true
unused-packages = true
176 changes: 176 additions & 0 deletions cmd/kfwd.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,176 @@
// Copyright (c) Inlets Author(s) 2019. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

package cmd

import (
"fmt"
"io/ioutil"
"log"
"os"
"os/signal"
"path"
"strings"
"syscall"

v1 "github.com/alexellis/go-execute/pkg/v1"
"github.com/spf13/cobra"
)

func init() {
inletsCmd.AddCommand(kfwdCmd)

kfwdCmd.Flags().StringP("from", "f", "", "From service for the inlets client to forward")
kfwdCmd.Flags().StringP("if", "i", "", "Destination interface for the inlets server")
kfwdCmd.Flags().StringP("namespace", "n", "default", "Source service namespace")

}

// clientCmd represents the client sub command.
var kfwdCmd = &cobra.Command{
Use: "kfwd",
Short: "Forward a Kubernetes service to the local machine",
Long: `Forward a Kubernetes service to the local machine`,
Example: ` inletsctl kfwd --from test-app-expressjs-k8s:8080
inletsctl kfwd --from test-app-expressjs-k8s:8080 --if 192.168.0.14
`,
RunE: runKfwd,
SilenceUsage: true,
SilenceErrors: true,
}

func runKfwd(cmd *cobra.Command, _ []string) error {
ns, err := cmd.Flags().GetString("namespace")
if err != nil {
return err
}

eth, err := cmd.Flags().GetString("if")
if err != nil {
return err
}

if len(eth) == 0 {
return fmt.Errorf("give --if with the IP of your local network from ifconfig or similar")
}

from, err := cmd.Flags().GetString("from")
if err != nil {
return err
}

if len(from) == 0 {
return fmt.Errorf("give a --from service")
}

portSep := strings.Index(from, ":")
if portSep < 0 {
return fmt.Errorf("no port given in --from flag")
}

upstream := from[:portSep]
port := from[portSep+1:]

fmt.Println(upstream, "=", port)

deployment := makeDeployment(eth, port, upstream, ns)
tmpPath := path.Join(os.TempDir(), "inlets-"+upstream+".yaml")
ioutil.WriteFile(tmpPath, []byte(deployment), 0600)

task := v1.ExecTask{
Command: "kubectl",
Args: []string{"apply", "-f", tmpPath},
}
res, err := task.Execute()

if err != nil {
return err
}

if res.ExitCode != 0 {
return fmt.Errorf("exit code unexpected: %d, stderr: %s", res.ExitCode, res.Stderr)
}

fmt.Println("Inlets client scheduled inside your cluster.")

go func() {
sig := make(chan os.Signal, 1)
signal.Notify(sig, syscall.SIGTERM)
signal.Notify(sig, syscall.SIGINT)

<-sig

log.Printf("Interrupt received..\n")

task := v1.ExecTask{
Command: "kubectl",
Args: []string{"delete", "-f", tmpPath},
}
res, err := task.Execute()

if err != nil {
fmt.Fprintf(os.Stderr, err.Error())
return
}

if res.ExitCode != 0 {
fmt.Fprintf(os.Stderr, fmt.Errorf("exit code unexpected from kubectl delete: %d, stderr: %s", res.ExitCode, res.Stderr).Error())
return
}
}()

fmt.Printf(`Inlets server now listening.

http://%s:%s

Hit Control+C to cancel.`, eth, port)

serverTask := v1.ExecTask{
Command: "inlets",
Args: []string{
"server",
"--port=" + fmt.Sprintf("%s", port),
},
}
serverRes, serverErr := serverTask.Execute()

if serverErr != nil {
return fmt.Errorf("error with server: %s", serverErr.Error())
}

if serverRes.ExitCode != 0 {
return fmt.Errorf("exit code unexpected from inlets server: %d, stderr: %s, stdout: %s", serverRes.ExitCode, serverRes.Stderr, serverRes.Stdout)

}

return nil
}

func makeDeployment(remote, port, upstream, ns string) string {

return fmt.Sprintf(`apiVersion: apps/v1
kind: Deployment
metadata:
name: inlets-%s
namespace: %s
spec:
replicas: 1
selector:
matchLabels:
app.kubernetes.io/name: inlets
template:
metadata:
labels:
app.kubernetes.io/name: inlets
spec:
containers:
- name: inlets
image: inlets/inlets:2.6.3
imagePullPolicy: IfNotPresent
command: ["inlets"]
args:
- "client"
- "--remote=ws://%s:%s"
- "--upstream=http://%s:%s"
`, upstream, ns, remote, port, upstream, port)
}
21 changes: 21 additions & 0 deletions vendor/github.com/alexellis/go-execute/LICENSE

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

113 changes: 113 additions & 0 deletions vendor/github.com/alexellis/go-execute/pkg/v1/exec.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.