Esempio n. 1
0
File: env.go Progetto: yannski/cli
func dbURLFromAPI(app, envWord string, urlSchemes []string) (string, error) {
	environ, err := api.VariablesListWithoutAlias(app)
	if err != nil {
		return "", errgo.Mask(err)
	}
	for _, variable := range environ {
		for _, scheme := range urlSchemes {
			if strings.Contains(variable.Name, envWord) && strings.HasPrefix(variable.Value, scheme) {
				return variable.Value, nil
			}
		}
	}

	return "", errgo.Newf("no %v addon detected", strings.ToLower(envWord))
}
Esempio n. 2
0
func Tunnel(app string, dbEnvVar string, identity string, port int) error {
	environ, err := api.VariablesListWithoutAlias(app)
	if err != nil {
		return errgo.Mask(err)
	}

	dbUrlStr := dbEnvVarValue(dbEnvVar, environ)
	if dbUrlStr == "" {
		return errgo.Newf("no such environment variable: %s", dbEnvVar)
	}

	dbUrl, err := url.Parse(dbUrlStr)
	if err != nil {
		return errgo.Notef(err, "invalid database 'URL': %s", dbUrlStr)
	}
	fmt.Printf("Building tunnel to %s\n", dbUrl.Host)

	var privateKeys []ssh.Signer
	if identity == "ssh-agent" {
		var agentConnection io.Closer
		privateKeys, agentConnection, err = sshkeys.ReadPrivateKeysFromAgent()
		if err != nil {
			return errgo.Mask(err)
		}
		defer agentConnection.Close()
	}

	if len(privateKeys) == 0 {
		identity = sshkeys.DefaultKeyPath
		privateKey, err := sshkeys.ReadPrivateKey(identity)
		if err != nil {
			return errgo.Mask(err)
		}
		privateKeys = append(privateKeys, privateKey)
	}

	debug.Println("Identity used:", identity)

	var client *ssh.Client
	for _, privateKey := range privateKeys {
		sshConfig := &ssh.ClientConfig{
			User: "******",
			Auth: []ssh.AuthMethod{ssh.PublicKeys(privateKey)},
		}

		client, err = ssh.Dial("tcp", config.C.SshHost, sshConfig)
		if err == nil {
			break
		} else {
			config.C.Logger.Println("Fail to connect to the SSH server", err)
		}
	}
	if client == nil {
		return errgo.Newf("No authentication method has succeeded, please use the flag '-i /path/to/private/key' to specify your private key")
	}

	tcpAddr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("localhost:%d", port))
	if err != nil {
		return errgo.Mask(err)
	}

	sock, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		return errgo.Mask(err)
	}
	defer sock.Close()
	fmt.Printf("You can access your database on '%v'\n", sock.Addr())

	go startIDGenerator()
	errs := make(chan error)
	for {
		select {
		case err := <-errs:
			return errgo.Mask(err)
		default:
		}

		connToTunnel, err := sock.Accept()
		if err != nil {
			return errgo.Mask(err)
		}
		go handleConnToTunnel(client, dbUrl, connToTunnel, errs)
	}
}