package access

import (
	"net/url"

	"github.com/cloudflare/cloudflared/carrier"
	"github.com/cloudflare/cloudflared/cmd/cloudflared/config"
	"github.com/cloudflare/cloudflared/validation"
	"github.com/pkg/errors"
	cli "gopkg.in/urfave/cli.v2"
)

// ssh will start a WS proxy server for server mode
// or copy from stdin/stdout for client mode
// useful for proxying other protocols (like ssh) over websockets
// (which you can put Access in front of)
func ssh(c *cli.Context) error {
	hostname, err := validation.ValidateHostname(c.String("hostname"))
	if err != nil || c.String("hostname") == "" {
		return cli.ShowCommandHelp(c, "ssh")
	}

	if c.NArg() > 0 || c.IsSet("url") {
		localForwarder, err := config.ValidateUrl(c)
		if err != nil {
			logger.WithError(err).Error("Error validating origin URL")
			return errors.Wrap(err, "error validating origin URL")
		}
		forwarder, err := url.Parse(localForwarder)
		if err != nil {
			logger.WithError(err).Error("Error validating origin URL")
			return errors.Wrap(err, "error validating origin URL")
		}
		return carrier.StartServer(logger, forwarder.Host, "https://"+hostname, shutdownC)
	}

	return carrier.StartClient(logger, "https://"+hostname, &carrier.StdinoutStream{})
}