prices: Implement initial API and tests

internal/handlers/prices.go

@@ -1,6 +1,10 @@
 package handlers
 
 import (
+	"encoding/json"
+	"log"
+	"net/http"
+	"strings"
 	"time"
 )
 
@@ -13,18 +17,34 @@ type Price struct {
 	RemoteId   string // unique ID from source, for detecting duplicates
 }
 
-func InsertPrice(tx *Tx, p *Price) error {
-	err := tx.Insert(p)
-	if err != nil {
-		return err
-	}
-	return nil
+type PriceList struct {
+	Prices *[]*Price `json:"prices"`
+}
+
+func (p *Price) Read(json_str string) error {
+	dec := json.NewDecoder(strings.NewReader(json_str))
+	return dec.Decode(p)
+}
+
+func (p *Price) Write(w http.ResponseWriter) error {
+	enc := json.NewEncoder(w)
+	return enc.Encode(p)
+}
+
+func (pl *PriceList) Read(json_str string) error {
+	dec := json.NewDecoder(strings.NewReader(json_str))
+	return dec.Decode(pl)
+}
+
+func (pl *PriceList) Write(w http.ResponseWriter) error {
+	enc := json.NewEncoder(w)
+	return enc.Encode(pl)
 }
 
 func CreatePriceIfNotExist(tx *Tx, price *Price) error {
 	if len(price.RemoteId) == 0 {
 		// Always create a new price if we can't match on the RemoteId
-		err := InsertPrice(tx, price)
+		err := tx.Insert(price)
 		if err != nil {
 			return err
 		}
@@ -42,13 +62,32 @@ func CreatePriceIfNotExist(tx *Tx, price *Price) error {
 		return nil // price already exists
 	}
 
-	err = InsertPrice(tx, price)
+	err = tx.Insert(price)
 	if err != nil {
 		return err
 	}
 	return nil
 }
 
+func GetPrice(tx *Tx, priceid, userid int64) (*Price, error) {
+	var p Price
+	err := tx.SelectOne(&p, "SELECT * from prices where PriceId=? AND SecurityId IN (SELECT SecurityId FROM securities WHERE UserId=?)", priceid, userid)
+	if err != nil {
+		return nil, err
+	}
+	return &p, nil
+}
+
+func GetPrices(tx *Tx, userid int64) (*[]*Price, error) {
+	var prices []*Price
+
+	_, err := tx.Select(&prices, "SELECT * from prices where SecurityId IN (SELECT SecurityId FROM securities WHERE UserId=?)", userid)
+	if err != nil {
+		return nil, err
+	}
+	return &prices, nil
+}
+
 // Return the latest price for security in currency units before date
 func GetLatestPrice(tx *Tx, security, currency *Security, date *time.Time) (*Price, error) {
 	var p Price
@@ -89,3 +128,113 @@ func GetClosestPrice(tx *Tx, security, currency *Security, date *time.Time) (*Pr
 		return earliest, nil
 	}
 }
+
+func PriceHandler(r *http.Request, tx *Tx) ResponseWriterWriter {
+	user, err := GetUserFromSession(tx, r)
+	if err != nil {
+		return NewError(1 /*Not Signed In*/)
+	}
+
+	if r.Method == "POST" {
+		price_json := r.PostFormValue("price")
+		if price_json == "" {
+			return NewError(3 /*Invalid Request*/)
+		}
+
+		var price Price
+		err := price.Read(price_json)
+		if err != nil {
+			return NewError(3 /*Invalid Request*/)
+		}
+		price.PriceId = -1
+
+		_, err = GetSecurity(tx, price.SecurityId, user.UserId)
+		if err != nil {
+			return NewError(3 /*Invalid Request*/)
+		}
+		_, err = GetSecurity(tx, price.CurrencyId, user.UserId)
+		if err != nil {
+			return NewError(3 /*Invalid Request*/)
+		}
+
+		err = tx.Insert(&price)
+		if err != nil {
+			log.Print(err)
+			return NewError(999 /*Internal Error*/)
+		}
+
+		return ResponseWrapper{201, &price}
+	} else if r.Method == "GET" {
+		var priceid int64
+		n, err := GetURLPieces(r.URL.Path, "/price/%d", &priceid)
+
+		if err != nil || n != 1 {
+			//Return all prices
+			var pl PriceList
+
+			prices, err := GetPrices(tx, user.UserId)
+			if err != nil {
+				log.Print(err)
+				return NewError(999 /*Internal Error*/)
+			}
+
+			pl.Prices = prices
+			return &pl
+		} else {
+			price, err := GetPrice(tx, priceid, user.UserId)
+			if err != nil {
+				return NewError(3 /*Invalid Request*/)
+			}
+
+			return price
+		}
+	} else {
+		priceid, err := GetURLID(r.URL.Path)
+		if err != nil {
+			return NewError(3 /*Invalid Request*/)
+		}
+		if r.Method == "PUT" {
+			price_json := r.PostFormValue("price")
+			if price_json == "" {
+				return NewError(3 /*Invalid Request*/)
+			}
+
+			var price Price
+			err := price.Read(price_json)
+			if err != nil || price.PriceId != priceid {
+				return NewError(3 /*Invalid Request*/)
+			}
+
+			_, err = GetSecurity(tx, price.SecurityId, user.UserId)
+			if err != nil {
+				return NewError(3 /*Invalid Request*/)
+			}
+			_, err = GetSecurity(tx, price.CurrencyId, user.UserId)
+			if err != nil {
+				return NewError(3 /*Invalid Request*/)
+			}
+
+			count, err := tx.Update(&price)
+			if err != nil || count != 1 {
+				log.Print(err)
+				return NewError(999 /*Internal Error*/)
+			}
+
+			return &price
+		} else if r.Method == "DELETE" {
+			price, err := GetPrice(tx, priceid, user.UserId)
+			if err != nil {
+				return NewError(3 /*Invalid Request*/)
+			}
+
+			count, err := tx.Delete(price)
+			if err != nil || count != 1 {
+				log.Print(err)
+				return NewError(999 /*Internal Error*/)
+			}
+
+			return SuccessWriter{}
+		}
+	}
+	return NewError(3 /*Invalid Request*/)
+}