added dividend db table and insert by trade

internal/handlers/portfolio.go

@@ -20,45 +20,57 @@ func AddTradeHandler(db *sql.DB) http.HandlerFunc {
 			return
 		}
 
-		err := req.Validate()
-		if err != nil {
-			http.Error(w, fmt.Sprintf("failed to validate trade: %s", err), http.StatusInternalServerError)
+		if err := req.Validate(); err != nil {
+			http.Error(w, fmt.Sprintf("failed to validate trade: %s", err), http.StatusBadRequest)
 			return
 		}
 
-		// check if currency is in the db.
 		currency, err := database.GetCurrencyByCode(db, req.CurrencyCode)
 		if err != nil {
 			http.Error(w, fmt.Sprintf("failed to find currency: %s", err), http.StatusInternalServerError)
 			return
 		}
 
-		trade := model.Trade{
-			Symbol:       req.Symbol,
-			Shares:       req.Shares,
-			Product:      model.TradeProduct(req.Product),
-			Type:         model.TradeType(req.Type),
-			Price:        req.Price,
-			CurrencyCode: currency.Code,
-			Date:         req.Date,
-		}
+		switch model.TradeType(req.Type) {
+		case model.DividendType:
+			dividend, err := req.ToDividend()
+			if err != nil {
+				http.Error(w, fmt.Sprintf("failed to build dividend: %s", err), http.StatusBadRequest)
+				return
+			}
+			dividend.CurrencyCode = currency.Code
 
-		err = database.InsertTrade(db, trade)
-		if err != nil {
-			http.Error(w, fmt.Sprintf("failed to insert trade into db: %s", err), http.StatusInternalServerError)
-			return
-		}
+			if err := database.InsertDividend(db, dividend); err != nil {
+				http.Error(w, fmt.Sprintf("failed to insert dividend: %s", err), http.StatusInternalServerError)
+				return
+			}
 
-		err = service.UpdatePositionByTradeList(db)
-		update := true
-		if err != nil {
-			update = false
-		}
+			w.Header().Set("Content-Type", "application/json")
+			json.NewEncoder(w).Encode(map[string]any{"success": true})
 
-		w.Header().Set("Content-Type", "application/json")
-		if err := json.NewEncoder(w).Encode(map[string]any{"success": true, "position update": update}); err != nil {
-			http.Error(w, fmt.Sprintf("failed to encode trades: %s", err), http.StatusInternalServerError)
-			return
+		case model.BuyType, model.SellType:
+			trade, err := req.ToTrade()
+			if err != nil {
+				http.Error(w, fmt.Sprintf("failed to build trade: %s", err), http.StatusBadRequest)
+				return
+			}
+			trade.CurrencyCode = currency.Code
+
+			if err := database.InsertTrade(db, trade); err != nil {
+				http.Error(w, fmt.Sprintf("failed to insert trade: %s", err), http.StatusInternalServerError)
+				return
+			}
+
+			update := true
+			if err := service.UpdatePositionByTradeList(db); err != nil {
+				update = false
+			}
+
+			w.Header().Set("Content-Type", "application/json")
+			json.NewEncoder(w).Encode(map[string]any{"success": true, "position_update": update})
+
+		default:
+			http.Error(w, fmt.Sprintf("unknown trade type: %d", req.Type), http.StatusBadRequest)
 		}
 	}
 }