mirror of
https://github.com/aclindsa/moneygo.git
synced 2024-10-30 15:50:04 -04:00
155 lines
3.5 KiB
Go
155 lines
3.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"github.com/aclindsa/moneygo/internal/models"
|
|
"github.com/aclindsa/moneygo/internal/store/db"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func CreatePriceIfNotExist(tx *db.Tx, price *models.Price) error {
|
|
if len(price.RemoteId) == 0 {
|
|
// Always create a new price if we can't match on the RemoteId
|
|
err := tx.Insert(price)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
exists, err := tx.PriceExists(price)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exists {
|
|
return nil // price already exists
|
|
}
|
|
|
|
err = tx.InsertPrice(price)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Return the price for security in currency closest to date
|
|
func GetClosestPrice(tx *db.Tx, security, currency *models.Security, date *time.Time) (*models.Price, error) {
|
|
earliest, _ := tx.GetEarliestPrice(security, currency, date)
|
|
latest, err := tx.GetLatestPrice(security, currency, date)
|
|
|
|
// Return early if either earliest or latest are invalid
|
|
if earliest == nil {
|
|
return latest, err
|
|
} else if err != nil {
|
|
return earliest, nil
|
|
}
|
|
|
|
howlate := earliest.Date.Sub(*date)
|
|
howearly := date.Sub(latest.Date)
|
|
if howearly < howlate {
|
|
return latest, nil
|
|
} else {
|
|
return earliest, nil
|
|
}
|
|
}
|
|
|
|
func PriceHandler(r *http.Request, context *Context, user *models.User, securityid int64) ResponseWriterWriter {
|
|
security, err := context.Tx.GetSecurity(securityid, user.UserId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
if r.Method == "POST" {
|
|
var price models.Price
|
|
if err := ReadJSON(r, &price); err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
price.PriceId = -1
|
|
|
|
if price.SecurityId != security.SecurityId {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
_, err = context.Tx.GetSecurity(price.CurrencyId, user.UserId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
err = context.Tx.Insert(&price)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return NewError(999 /*Internal Error*/)
|
|
}
|
|
|
|
return ResponseWrapper{201, &price}
|
|
} else if r.Method == "GET" {
|
|
if context.LastLevel() {
|
|
//Return all this security's prices
|
|
var pl models.PriceList
|
|
|
|
prices, err := context.Tx.GetPrices(security.SecurityId)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return NewError(999 /*Internal Error*/)
|
|
}
|
|
|
|
pl.Prices = prices
|
|
return &pl
|
|
}
|
|
|
|
priceid, err := context.NextID()
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
price, err := context.Tx.GetPrice(priceid, security.SecurityId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
return price
|
|
} else {
|
|
priceid, err := context.NextID()
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
if r.Method == "PUT" {
|
|
var price models.Price
|
|
if err := ReadJSON(r, &price); err != nil || price.PriceId != priceid {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
_, err = context.Tx.GetSecurity(price.SecurityId, user.UserId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
_, err = context.Tx.GetSecurity(price.CurrencyId, user.UserId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
err = context.Tx.UpdatePrice(&price)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return NewError(999 /*Internal Error*/)
|
|
}
|
|
|
|
return &price
|
|
} else if r.Method == "DELETE" {
|
|
price, err := context.Tx.GetPrice(priceid, security.SecurityId)
|
|
if err != nil {
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|
|
|
|
err = context.Tx.DeletePrice(price)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return NewError(999 /*Internal Error*/)
|
|
}
|
|
|
|
return SuccessWriter{}
|
|
}
|
|
}
|
|
return NewError(3 /*Invalid Request*/)
|
|
}
|