added dividend db table and insert by trade

internal/database/main.go

@@ -17,14 +17,26 @@ func InitDB(db *sql.DB) {
     );
 
     CREATE TABLE IF NOT EXISTS trades (
-        id          INTEGER PRIMARY KEY AUTOINCREMENT,
-        symbol      TEXT NOT NULL,
-        currency_code TEXT NOT NULL,
-        shares      INTEGER NOT NULL,
-        product     INTEGER NOT NULL CHECK(product IN (0, 1, 2, 3)),
-        type        INTEGER NOT NULL CHECK(type IN (0, 1)),
-        price       REAL    NOT NULL,
-        traded_at   DATETIME NOT NULL
+        id            INTEGER PRIMARY KEY AUTOINCREMENT,
+        symbol        TEXT    NOT NULL,
+        currency_code TEXT    NOT NULL,
+        shares        INTEGER NOT NULL,
+        product       INTEGER NOT NULL CHECK(product IN (0, 1, 2, 3, 4)), -- added 4 for BondTrade
+        type          INTEGER NOT NULL CHECK(type IN (0, 1)),             -- Buy=0, Sell=1 only; Dividend has its own table
+        price         REAL    NOT NULL,
+        traded_at     DATETIME NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS dividends (
+        id            INTEGER PRIMARY KEY AUTOINCREMENT,
+        symbol        TEXT    NOT NULL,
+        currency_code TEXT    NOT NULL,
+        product       INTEGER NOT NULL CHECK(product IN (0, 1, 2, 3, 4)),
+        value         REAL    NOT NULL,
+        tax_amount    REAL    NOT NULL DEFAULT 0,
+        tax_rate      REAL    NOT NULL DEFAULT 0,
+        net_value     REAL    NOT NULL,
+        payment_date  DATETIME NOT NULL
     );
 
     CREATE TABLE IF NOT EXISTS position (

internal/database/portfolio.go

@@ -25,14 +25,7 @@ func GetTrades(db *sql.DB) ([]model.Trade, error) {
 			return nil, err
 		}
 
-		switch typeInt {
-		case 0:
-			t.Type = model.TradeType(false)
-		case 1:
-			t.Type = model.TradeType(true)
-		default:
-			return nil, fmt.Errorf("failed to convert given Type int to bool of trade type.")
-		}
+		t.Type = model.TradeType(typeInt)
 
 		trades = append(trades, t)
 	}
@@ -81,6 +74,21 @@ func InsertTrade(db *sql.DB, trade model.Trade) error {
 	return err
 }
 
+func InsertDividend(db *sql.DB, div model.Dividend) error {
+	_, err := db.Exec(
+		"INSERT INTO trades (symbol, currency_code, shares, product, value, tax_amount, tax_rate, net_value, payment_date) VALUES (?, ?, ?, ?, ?, ?, ?)",
+		div.Symbol,
+		div.CurrencyCode,
+		div.Product,
+		div.Value,
+		div.TaxAmount,
+		div.TaxRate,
+		div.NetValue,
+		div.PaymentDate,
+	)
+	return err
+}
+
 func UpdatePositions(db *sql.DB, positions []model.Position) error {
 	// Complete overwrite of the db positions
 	_, err := db.Exec("DELETE FROM position")