aastocks.go

package aastocks

import (
	"net/http"
	"time"
)

// Quote of AAStocks data
type Quote struct {
	Symbol       string
	Name         string
	Price        float64
	Price52WLow  float64
	Price52WHigh float64
	Yield        float64
	PeRatio      float64
	PbRatio      float64
	Lots         int
	Eps          float64
	UpdateTime   time.Time

	client *http.Client
}

// Get quote from AAStocks with symbol
func Get(symbol string, opts ...Option) (*Quote, error) {
	q := &Quote{
		Symbol: symbol,
		client: defaultClient(),
	}
	for _, opt := range opts {
		opt(q)
	}
	return q, q.details()
}

func defaultClient() *http.Client {
	t := &transport{r: http.DefaultTransport}
	return &http.Client{
		Transport: t,
	}
}

type transport struct {
	r http.RoundTripper
}

func (t *transport) RoundTrip(req *http.Request) (*http.Response, error) {
	req.Header.Add("Referer", req.URL.String())
	return t.r.RoundTrip(req)
}