utils.go

package goibcp

import (
	"errors"
	"fmt"
	"time"

	"github.com/go-resty/resty/v2"
)

func logMsg(level int, fn string, msg string, err ...error) {
	if level <= Settings.LogLevel {
		switch level {
		case 0:
			fmt.Println("GOIBCP ERROR in function", fn, ":", msg, err)
		case 1:
			fmt.Println("GOBICP ERROR in function", fn, ":", msg, err)
		case 2:
			fmt.Println("GOIBCP", fn, ":", msg)
		case 3:
			fmt.Println("GOIBCP Debug:", fn, msg)
		}
	}
}

//GetEndpoint - Get an endpoint from IBCP and return a map
func (c *IBClient) GetEndpoint(endp string, res interface{}, qs ...string) error {
	epURL := Settings.CPURL + endpoints[endp]
	var req *resty.Request
	if len(qs) > 0 {
		req = rClient.R().SetResult(res).SetQueryString(qs[0])
	} else {
		req = rClient.R().SetResult(res).SetHeader("Accept", "application/json")
	}
	resp, err := req.Get(epURL)
	if err != nil {
		logMsg(ERROR, endp, "Failed to get", err)
		return err
	}
	if resp.StatusCode() != 200 {
		err = errors.New("HTTP " + fmt.Sprintf("%d", resp.StatusCode()) + "error")
		logMsg(ERROR, endp, "StatusCode ", err)
		return err
	}
	logMsg(DEBUG, endp, resp.String())
	return nil
}

//PostEndpoint - Post to an endpoint from IBCP
func (c *IBClient) PostEndpoint(endp string, res interface{}) error {
	epURL := Settings.CPURL + endpoints[endp]
	resp, err := rClient.R().SetResult(res).SetHeader("Content-Type", "application/json").Post(epURL)
	if err != nil {
		logMsg(ERROR, endp, "Failed to post", err)
		return err
	}
	logMsg(DEBUG, endp, resp.String())
	return nil
}

//KeepAlive - Keeps the sesssion alive by tickeling the server every minute unless an error is encountered or session expires
//reauthorize session on nobridge error
func KeepAlive(c *IBClient) error {
	KAStatus = true
	var treply = IBTickle{}
	var sessionInfo = IBSession{}
	var err error
	for {
		time.Sleep(55 * time.Second)
		err = c.GetSessionInfo(&sessionInfo)
		//exit if http response is not 200 from sessionInfo endpoint.
		//TODO: send disconnected event to a channel
		if err != nil {
			logMsg(ERROR, "KeepAlive", "No Active Session found")
			KAStatus = false
			return err
		}
		// fmt.Printf("%+v\n", sessionInfo)
		time.Sleep(5 * time.Second)
		err = c.GetEndpoint("sessionTickle", &treply)
		// var l = IBAccountLedger{}
		// c.GetAccountLedger(&l)
		logMsg(INFO, "KeepAlive", fmt.Sprintf("%+v", treply))
		if err != nil {
			logMsg(ERROR, "KeepAlive", "Error on tickle, Reauthenticating as session is still valid")
			c.Reauthenticate()
		}
		if treply.Iserver.Error != "" {
			//try to reconnect if "no bridge error is recieved"
			if treply.Iserver.Error == "no bridge" {
				logMsg(ERROR, "KeepAlive", "No Bridge error on Tickle..trying to rauthenticate")
				c.Reauthenticate()
			}
			//return errors.New(treply.Iserver.Error)
		}
		// if treply.Iserver.AuthStatus.Connected == false || treply.Iserver.AuthStatus.Authenticated == false {
		// 	return errors.New("IB Session disconnected")
		// }
		c.UserID = sessionInfo.UserID
		c.IsAuthenticated = treply.Iserver.AuthStatus.Authenticated
		c.IsConnected = treply.Iserver.AuthStatus.Connected
		c.IsCompeting = treply.Iserver.AuthStatus.Competing
		c.Message = treply.Iserver.AuthStatus.Message
	}
}

//TODO: create Helper methods to place simple market and limit orders for stocks and futures