evcc-io/vehicle/niu.go

package vehicle

import (
	"crypto/md5"
	"encoding/hex"
	"errors"
	"io"
	"net/http"
	"net/url"
	"strings"
	"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/vehicle/niu"
)

// Niu is an api.Vehicle implementation for Niu vehicles
type Niu struct {
	*embed
	*request.Helper
	user, password string
	serial         string
	token          niu.Token
	apiG           func() (niu.Response, error)
}

func init() {
	registry.Add("niu", NewNiuFromConfig)
}

// NewFordFromConfig creates a new vehicle
func NewNiuFromConfig(other map[string]any) (api.Vehicle, error) {
	cc := struct {
		embed                  `mapstructure:",squash"`
		User, Password, Serial string
		Cache                  time.Duration
	}{
		Cache: interval,
	}

	if err := util.DecodeOther(other, &cc); err != nil {
		return nil, err
	}

	if cc.User == "" || cc.Password == "" {
		return nil, api.ErrMissingCredentials
	}

	if cc.Serial == "" {
		return nil, errors.New("missing serial")
	}

	log := util.NewLogger("niu").Redact(cc.User, cc.Password)

	v := &Niu{
		embed:    &cc.embed,
		Helper:   request.NewHelper(log),
		user:     cc.User,
		password: cc.Password,
		serial:   strings.ToUpper(cc.Serial),
	}

	v.apiG = util.Cached(v.batteryAPI, cc.Cache)

	return v, nil
}

// login implements the Niu oauth2 api
func (v *Niu) login() error {
	hash := md5.New()
	hash.Write([]byte(v.password))
	md5hash := hex.EncodeToString(hash.Sum(nil))

	data := url.Values{
		"account":    {v.user},
		"password":   {md5hash},
		"grant_type": {"password"},
		"scope":      {"base"},
		"app_id":     {"niu_8xt1afu6"},
	}

	uri := niu.AuthURI + "/v3/api/oauth2/token"
	req, err := request.New(http.MethodPost, uri, strings.NewReader(data.Encode()), map[string]string{
		"Content-Type": "application/x-www-form-urlencoded",
	})

	if err == nil {
		var token niu.Token
		if err = v.DoJSON(req, &token); err == nil {
			v.token = token
		}
	}

	return err
}

func (v *Niu) tokenRefresh() error {
	if v.token.AccessToken == "" || time.Until(v.token.Expiry) < time.Minute {
		if err := v.login(); err != nil {
			return err
		}
	}
	return nil
}

func (v *Niu) newRequest(method, uri string, body io.Reader) (*http.Request, error) {
	if err := v.tokenRefresh(); err != nil {
		return nil, err
	}
	req, err := request.New(method, uri, body, map[string]string{
		"Accept-Language": "en-US",
		"Content-Type":    "application/x-www-form-urlencoded",
		"token":           v.token.AccessToken,
	})
	return req, err
}

func (v *Niu) get(uri string) (*http.Request, error) {
	return v.newRequest(http.MethodGet, uri, nil)
}

func (v *Niu) post(uri string) (*http.Request, error) {
	data := url.Values{"sn": {v.serial}}
	return v.newRequest(http.MethodPost, uri, strings.NewReader(data.Encode()))
}

// batteryAPI provides battery api response
func (v *Niu) batteryAPI() (niu.Response, error) {
	var res niu.Response

	req, err := v.get(niu.ApiURI + "/v3/motor_data/index_info?sn=" + v.serial)
	if err == nil {
		err = v.DoJSON(req, &res)
	}

	return res, err
}

// Soc implements the api.Vehicle interface
func (v *Niu) Soc() (float64, error) {
	res, err := v.apiG()
	return float64(res.Data.Batteries.CompartmentA.BatteryCharging), err
}

var _ api.ChargeState = (*Niu)(nil)

// Status implements the api.ChargeState interface
func (v *Niu) Status() (api.ChargeStatus, error) {
	status := api.StatusA // disconnected

	res, err := v.apiG()
	if err == nil {
		if res.Data.IsConnected {
			status = api.StatusB
		}
		if res.Data.IsCharging > 0 {
			status = api.StatusC
		}
	}

	return status, err
}

var _ api.VehicleRange = (*Niu)(nil)

// Range implements the api.VehicleRange interface
func (v *Niu) Range() (int64, error) {
	res, err := v.apiG()
	return res.Data.EstimatedMileage, err
}

var _ api.VehicleOdometer = (*Niu)(nil)

// Odometer implements the api.VehicleOdometer interface
func (v *Niu) Odometer() (float64, error) {
	var res niu.Response

	req, err := v.post(niu.ApiURI + "/motoinfo/overallTally")
	if err == nil {
		err = v.DoJSON(req, &res)
	}

	return res.Data.TotalMileage, err
}