evcc-io/vehicle/saic/api.go

package saic

import (
	"encoding/json"
	"fmt"
	"net/http"
	"time"

	"github.com/evcc-io/evcc/api"
	"github.com/evcc-io/evcc/util"
	"github.com/evcc-io/evcc/util/request"
	"github.com/evcc-io/evcc/util/transport"
	"github.com/evcc-io/evcc/vehicle/saic/requests"
	"golang.org/x/oauth2"
)

const (
	StatRunning = iota
	StatValid
	StatInvalid
)

type ConcurrentRequest struct {
	Status int
	Result requests.ChargeStatus
}

// API is an api.Vehicle implementation for SAIC cars
type API struct {
	*request.Helper
	identity *Identity
	request  ConcurrentRequest
	Logger   *util.Logger
}

// NewAPI creates a new vehicle
func NewAPI(log *util.Logger, identity *Identity) *API {
	v := &API{
		Helper:   request.NewHelper(log),
		identity: identity,
		Logger:   log,
	}

	v.Client.Transport = &transport.Decorator{
		Decorator: requests.Decorate,
		Base:      v.Client.Transport,
	}
	v.request.Status = StatInvalid

	return v
}

/*
func (v *API) printAnswer() {
	v.Logger.DEBUG.Printf("SOC:%d ", v.request.Result.ChrgMgmtData.BmsPackSOCDsp)
	v.Logger.DEBUG.Printf("GUN:%d ", v.request.Result.RvsChargeStatus.ChargingGunState)
	v.Logger.DEBUG.Printf("Chrg State:%d ", v.request.Result.ChrgMgmtData.BmsChrgSts)
	v.Logger.DEBUG.Printf("Mileage:%d ", v.request.Result.RvsChargeStatus.Mileage)
	v.Logger.DEBUG.Printf("Range:%d ", v.request.Result.RvsChargeStatus.FuelRangeElec)
}
*/

func (v *API) doRepeatedRequest(url string, event_id string) error {
	var req *http.Request

	answer := requests.Answer{
		Data: &v.request.Result,
	}

	token, err := v.identity.Token()
	if err != nil {
		v.request.Status = StatInvalid
		return err
	}

	req, err = requests.CreateRequest(url,
		http.MethodGet,
		"",
		request.JSONContent,
		token.AccessToken,
		event_id)
	if err != nil {
		v.request.Status = StatInvalid
		return err
	}

	_, err = v.DoRequest(req, &answer)
	if err == nil {
		v.request.Status = StatValid
	} else if err != api.ErrMustRetry {
		v.request.Status = StatInvalid
	}
	return err
}

// This is running concurrently
func (v *API) repeatRequest(url string, event_id string) {
	var err error
	var count = 0

	v.request.Status = StatRunning
	for err = api.ErrMustRetry; err == api.ErrMustRetry && count < 20; {
		time.Sleep(2 * time.Second)
		v.Logger.DEBUG.Printf("Starting repeated query. Count: %d\n", count)
		err = v.doRepeatedRequest(url, event_id)
		count++
	}

	v.Logger.DEBUG.Printf("Exitig repeated query. Count: %d\n", count)
	//v.printAnswer()
}

func (v *API) DoRequest(req *http.Request, result *requests.Answer) (string, error) {
	var body []byte

	resp, err := v.Do(req)
	if err != nil {
		return "", err
	}

	defer resp.Body.Close()
	if resp.StatusCode == http.StatusUnauthorized {
		v.Logger.DEBUG.Printf("DoRequest: %s", resp.Status)
		v.identity.Login()
		return "", api.ErrMustRetry
	}

	event_id := resp.Header.Get("event-id")

	if result != nil {
		body, err = requests.DecryptAnswer(resp)
		if err == nil {
			err = json.Unmarshal(body, result)
			if err == nil && result.Code != 0 {
				if result.Code == 4 {
					err = api.ErrMustRetry
				} else {
					err = fmt.Errorf("%d: %s", result.Code, result.Message)
				}
				v.Logger.DEBUG.Printf("%d: %s", result.Code, result.Message)
			}
		} else {
			v.Logger.DEBUG.Printf("decrypt: %s", err.Error())
		}
	}

	return event_id, err
}

/* Vehicles implements returns the /user/vehicles api
func (v *API) Vehicles() ([]Vehicle, error) {
	var res []Vehicle
	uri := fmt.Sprintf("%s/eadrax-vcs/v4/vehicles?apptimezone=120&appDateTime=%d", regions[v.region].CocoApiURI, time.Now().UnixMilli())
	err := v.GetJSON(uri, &res)
	return res, err
}
*/

func (v *API) Wakeup(vin string) error {
	var req *http.Request
	var err error
	var token *oauth2.Token

	token, err = v.identity.Token()
	if err != nil {
		return err
	}

	url := requests.BASE_URL_P + "vehicle/status?vin=" + requests.Sha256(vin)
	req, err = requests.CreateRequest(url,
		http.MethodGet,
		"",
		request.JSONContent,
		token.AccessToken,
		"")
	if err != nil {
		return err
	}

	v.DoRequest(req, nil)

	return nil
}

// Status implements the /user/vehicles/<vin>/status api
func (v *API) Status(vin string) (requests.ChargeStatus, error) {
	var req *http.Request
	var res requests.ChargeStatus
	var event_id string
	var err error
	var token *oauth2.Token
	answer := requests.Answer{
		Data: &res,
	}

	// Check if we are already running in the background
	if v.request.Status == StatValid {
		v.request.Status = StatInvalid
		v.Logger.DEBUG.Printf("StatVaild. Returning stored value\n")
		//v.printAnswer()
		return v.request.Result, nil
	} else if v.request.Status == StatRunning {
		v.Logger.DEBUG.Printf("StatRunning. Exiting\n")
		return res, api.ErrMustRetry
	}
	v.Logger.DEBUG.Printf("StatInvaild. Starting query\n")

	token, err = v.identity.Token()
	if err != nil {
		return res, err
	}

	url := requests.BASE_URL_P + "vehicle/charging/mgmtData?vin=" + requests.Sha256(vin)

	// get charging status of vehicle
	req, err = requests.CreateRequest(url,
		http.MethodGet,
		"",
		request.JSONContent,
		token.AccessToken,
		"")
	if err != nil {
		return res, err
	}

	event_id, err = v.DoRequest(req, &answer)

	if err != nil {
		v.Logger.DEBUG.Printf("Getting event id failed")
		return res, err
	}

	if event_id == "" {
		v.Logger.ERROR.Printf("Answer without event ID")
		return res, api.ErrMustRetry
	}

	req, err = requests.CreateRequest(url,
		http.MethodGet,
		"",
		request.JSONContent,
		token.AccessToken,
		event_id)
	if err != nil {
		v.Logger.ERROR.Printf("Could not create request %s", err.Error())
		return res, err
	}

	_, err = v.DoRequest(req, &answer)

	// Continue checking....
	if err == api.ErrMustRetry {
		v.request.Status = StatRunning
		v.Logger.DEBUG.Printf(" No answer yet. Continue status query in background\n")
		go v.repeatRequest(url, event_id)
	} else if err != nil {
		v.Logger.ERROR.Printf("doRequest failed with %s", err.Error())
	}

	return res, err
}