Merge pull request #1 from cdr/initial-port

Export source from monorepo

Author: sreya

.gitignore

@@ -5,3 +5,4 @@ node_modules
 .dbtemp
 .vscode
 vendor
+cloud-agent

go.mod

@@ -0,0 +1,14 @@
+module go.coder.com/cloud-agent
+
+go 1.15
+
+require (
+	cdr.dev/slog v1.3.0
+	github.com/pkg/browser v0.0.0-20180916011732-0a3d74bf9ce4
+	github.com/spf13/pflag v1.0.5
+	go.coder.com/cli v0.4.0
+	go.coder.com/flog v0.0.0-20200908145530-d7adc3802a47
+	golang.org/x/sync v0.0.0-20190423024810-112230192c58
+	golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1
+	nhooyr.io/websocket v1.8.6
+)

go.sum

@@ -0,0 +1,34 @@
+package client
+
+import (
+	"net/url"
+
+	"go.coder.com/cloud-agent/internal/config"
+)
+
+type Client struct {
+	token   string
+	baseURL *url.URL
+}
+
+func FromEnv() (*Client, error) {
+	token, err := config.SessionToken.Read()
+	if err != nil {
+		return nil, err
+	}
+
+	u, err := config.URL.Read()
+	if err != nil {
+		return nil, err
+	}
+
+	parsed, err := url.Parse(u)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Client{
+		token:   token,
+		baseURL: parsed,
+	}, nil
+}

internal/client/client.go

@@ -0,0 +1,75 @@
+package client
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"golang.org/x/xerrors"
+)
+
+func (c *Client) CodeServer(id string) (*CodeServer, error) {
+	path := fmt.Sprintf("/api/servers/%v", id)
+
+	var response CodeServer
+	err := c.requestBody("GET", path, nil, &response)
+	if err != nil {
+		return nil, err
+	}
+
+	return &response, nil
+}
+
+type AccessURLResponse struct {
+	URL string `json:"url"`
+}
+
+func (c *Client) AccessURL(id string) (string, error) {
+	path := fmt.Sprintf("/api/servers/%v/access-url", id)
+
+	var response AccessURLResponse
+	err := c.requestBody("GET", path, nil, &response)
+	if err != nil {
+		return "", err
+	}
+
+	return response.URL, nil
+}
+
+type CodeServer struct {
+	ID               string    `json:"id"`
+	UserID           string    `json:"user_id"`
+	Name             string    `json:"name"`
+	Hostname         string    `json:"hostname"`
+	CreatedAt        time.Time `json:"created_at"`
+	LastConnectionAt time.Time `json:"last_connection_at"`
+}
+
+// RegisterServerRequest is the request body sent in a
+// register server request.
+type RegisterServerRequest struct {
+	Name     string `json:"name"`
+	Hostname string `json:"hostname"`
+}
+
+func (c *Client) RegisterCodeServer(name string) (*CodeServer, error) {
+	const path = "/api/servers"
+	hostname, err := os.Hostname()
+	if err != nil {
+		return nil, xerrors.Errorf("get hostname: %w", err)
+	}
+
+	var response CodeServer
+	err = c.requestBody("POST", path,
+		&RegisterServerRequest{
+			Name:     name,
+			Hostname: hostname,
+		},
+		&response,
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	return &response, nil
+}

internal/client/code_server.go

@@ -0,0 +1,68 @@
+package client
+
+import (
+	"context"
+	"io/ioutil"
+	"net/url"
+
+	"github.com/pkg/browser"
+	"golang.org/x/xerrors"
+	"nhooyr.io/websocket"
+
+	"go.coder.com/cloud-agent/pkg/agentlogin"
+	"go.coder.com/flog"
+)
+
+func init() {
+	browser.Stderr = ioutil.Discard
+	browser.Stdout = ioutil.Discard
+}
+
+// Login performs the login flow for an agent. It returns the resulting
+// session token to use for authenticated routes.
+func Login(addr, serverName string) (string, error) {
+	ctx := context.Background()
+
+	u, err := url.Parse(addr)
+	if err != nil {
+		return "", err
+	}
+
+	query := url.Values{}
+	query.Add(agentlogin.ServerNameQueryParam, serverName)
+
+	loginURL := &url.URL{
+		Scheme:   u.Scheme,
+		Host:     u.Host,
+		Path:     "/login",
+		RawQuery: query.Encode(),
+	}
+
+	conn, _, err := websocket.Dial(ctx, loginURL.String(), nil)
+	if err != nil {
+		return "", err
+	}
+	defer conn.Close(websocket.StatusInternalError, "")
+
+	client := &agentlogin.Client{
+		Ctx:  ctx,
+		Conn: conn,
+	}
+
+	url, err := client.ReadAuthURL()
+	if err != nil {
+		return "", xerrors.Errorf("read auth url: %w", err)
+	}
+
+	err = browser.OpenURL(url)
+	if err != nil {
+		flog.Info("visit %s to login", url)
+	}
+
+	token, err := client.ReadSessionToken()
+	if err != nil {
+		return "", xerrors.Errorf("read session token: %w", err)
+	}
+
+	return token, nil
+}

internal/client/login.go

@@ -0,0 +1,62 @@
+package client
+
+import (
+	"bytes"
+	"encoding/json"
+	"net/http"
+
+	"golang.org/x/xerrors"
+)
+
+const sessionHeader = "Session-Token"
+
+func (c *Client) request(method, path string, body interface{}) (*http.Response, error) {
+	b, err := json.Marshal(body)
+	if err != nil {
+		return nil, xerrors.Errorf("marshal body: %w", err)
+	}
+
+	req, err := http.NewRequest(method, c.baseURL.String()+path, bytes.NewReader(b))
+	if err != nil {
+		return nil, xerrors.Errorf("new request: %w", err)
+	}
+
+	req.Header.Set(sessionHeader, c.token)
+
+	return http.DefaultClient.Do(req)
+}
+
+func (c *Client) requestBody(method, path string, request, response interface{}) error {
+	resp, err := c.request(method, path, request)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 && resp.StatusCode != 201 {
+		return bodyError(resp)
+	}
+
+	err = json.NewDecoder(resp.Body).Decode(&response)
+	if err != nil {
+		return xerrors.Errorf("unmarshal response: %w", err)
+	}
+	return nil
+}
+
+type apiError struct {
+	Err struct {
+		Msg string `json:"msg"`
+	} `json:"error"`
+}
+
+func bodyError(resp *http.Response) error {
+	var apiErr apiError
+
+	err := json.NewDecoder(resp.Body).Decode(&apiErr)
+	if err != nil {
+		return xerrors.Errorf("decode err: %w", err)
+	}
+
+	return xerrors.New(apiErr.Err.Msg)
+}

internal/client/request.go

@@ -0,0 +1,27 @@
+package client
+
+import (
+	"time"
+)
+
+// User describe a Coder Cloud user.
+type User struct {
+	ID        string    `json:"id" `
+	Name      string    `json:"name"`
+	Username  string    `json:"username"`
+	Email     string    `json:"email"`
+	CreatedAt time.Time `json:"created_at"`
+}
+
+func (c *Client) Me() (*User, error) {
+	const path = "/api/users/me"
+
+	var response User
+	err := c.requestBody("GET", path, nil, &response)
+	if err != nil {
+		return nil, err
+	}
+
+	return &response, nil
+
+}

internal/client/user.go

@@ -0,0 +1,38 @@
+package cmd
+
+import (
+	"github.com/spf13/pflag"
+	"go.coder.com/cli"
+)
+
+func Make() cli.Command {
+	return &rootCmd{}
+}
+
+var _ interface {
+	cli.Command
+	cli.ParentCommand
+} = &rootCmd{}
+
+type rootCmd struct {
+}
+
+func (c *rootCmd) Spec() cli.CommandSpec {
+	return cli.CommandSpec{
+		Name:  "agent",
+		Usage: "[GLOBAL FLAGS] COMMAND [COMMAND FLAGS] [ARGS...]",
+		Desc:  `Run the Coder Cloud Agent.`,
+	}
+}
+
+func (c *rootCmd) Subcommands() []cli.Command {
+	return []cli.Command{
+		&linkCmd{},
+		&proxyCmd{},
+		&versionCmd{},
+	}
+}
+
+func (c *rootCmd) Run(fl *pflag.FlagSet) {
+	fl.Usage()
+}