evcc-io/cmd/tesla.go

package cmd

import (
	"context"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"strings"

	"github.com/andig/evcc/server"
	"github.com/andig/evcc/util"
	"github.com/andig/evcc/util/request"
	"github.com/bogosj/tesla"
	"github.com/manifoldco/promptui"
	"github.com/skratchdot/open-golang/open"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"golang.org/x/oauth2"
)

// teslaCmd represents the vehicle command
var teslaCmd = &cobra.Command{
	Use:   "tesla-token",
	Short: "Generate Tesla token credentials",
	Run:   runTeslaToken,
}

func init() {
	rootCmd.AddCommand(teslaCmd)
}

// copied from https://github.com/bogosj/tesla
func getUsernameAndPassword() (string, string, error) {
	user, err := (&promptui.Prompt{
		Label:   "Username",
		Pointer: promptui.PipeCursor,
		Validate: func(s string) error {
			if len(s) == 0 {
				return errors.New("len(s) == 0")
			}
			return nil
		},
	}).Run()
	if err != nil {
		return "", "", err
	}

	password, err := (&promptui.Prompt{
		Label:   "Password",
		Mask:    '*',
		Pointer: promptui.PipeCursor,
		Validate: func(s string) error {
			if len(s) == 0 {
				return errors.New("len(s) == 0")
			}
			return nil
		},
	}).Run()
	if err != nil {
		return "", "", err
	}

	return user, password, nil
}

func codePrompt(ctx context.Context, devices []tesla.Device) (tesla.Device, string, error) {
	var i int
	if len(devices) > 1 {
		var err error
		i, _, err = (&promptui.Select{
			Label:   "Device",
			Items:   devices,
			Pointer: promptui.PipeCursor,
		}).Run()
		if err != nil {
			return tesla.Device{}, "", fmt.Errorf("select device: %w", err)
		}
	}

	code, err := (&promptui.Prompt{
		Label:   "Passcode",
		Pointer: promptui.PipeCursor,
		Validate: func(s string) error {
			if len(s) != 6 {
				return errors.New("len(s) != 6")
			}
			return nil
		},
	}).Run()
	if err != nil {
		return tesla.Device{}, "", err
	}

	return devices[i], strings.TrimSpace(code), nil
}

func captchaPrompt(ctx context.Context, svg io.Reader) (string, error) {
	tmpFile, err := ioutil.TempFile(os.TempDir(), "evcc-*.svg")
	if err != nil {
		return "", fmt.Errorf("cannot create temp file: %w", err)
	}

	if _, err := io.Copy(tmpFile, svg); err != nil {
		return "", fmt.Errorf("cannot write temp file: %w", err)
	}

	_ = tmpFile.Close()

	if err := open.Run(tmpFile.Name()); err != nil {
		return "", fmt.Errorf("cannot open captcha for display: %w", err)
	}

	fmt.Println("Captcha is now being opened in default application for svg files.")

	captcha, err := (&promptui.Prompt{
		Label:   "Captcha",
		Pointer: promptui.PipeCursor,
		Validate: func(s string) error {
			if len(s) < 4 {
				return errors.New("len(s) < 4")
			}
			return nil
		},
	}).Run()

	return strings.TrimSpace(captcha), err
}

func generateToken(username, password string) {
	ctx := context.WithValue(context.Background(), oauth2.HTTPClient, request.NewHelper(log).Client)
	client, err := tesla.NewClient(
		ctx,
		tesla.WithMFAHandler(codePrompt),
		tesla.WithCaptchaHandler(captchaPrompt),
		tesla.WithCredentials(username, password),
	)
	if err != nil {
		log.FATAL.Fatalln(err)
	}

	token, err := client.Token()
	if err != nil {
		log.FATAL.Fatalln(err)
	}

	fmt.Println()
	fmt.Println("Add the following tokens to the tesla vehicle config:")
	fmt.Println()
	fmt.Println("  tokens:")
	fmt.Println("    access:", token.AccessToken)
	fmt.Println("    refresh:", token.RefreshToken)
}

func runTeslaToken(cmd *cobra.Command, args []string) {
	util.LogLevel(viper.GetString("log"), viper.GetStringMapString("levels"))
	log.INFO.Printf("evcc %s (%s)", server.Version, server.Commit)

	user, password, err := getUsernameAndPassword()
	if err != nil {
		log.FATAL.Fatal(err)
	}

	generateToken(user, password)
}