new revenue getters

internal/database/revenue.go

@@ -57,3 +57,59 @@ func GetPeriodByID(db *sql.DB, periodID int) (model.Period, error) {
 	p.End, _ = time.Parse("2006-01-02", end)
 	return p, nil
 }
+
+func GetRevenueByPeriod(db *sql.DB, companyID int, periodID int) ([]model.Revenue, error) {
+	rows, err := db.Query(
+		`SELECT id, company_id, currency_id, period_id, value FROM revenue_entries WHERE company_id = ? AND period_id = ?`,
+		companyID, periodID,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("query revenue by period: %w", err)
+	}
+	defer rows.Close()
+
+	var revenues []model.Revenue
+	for rows.Next() {
+		var rc model.Revenue
+		if err := rows.Scan(&rc.ID, &rc.Company, &rc.Currency, &rc.Period, &rc.Value); err != nil {
+			return nil, fmt.Errorf("scan revenue row: %w", err)
+		}
+		revenues = append(revenues, rc)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("iterate revenue rows: %w", err)
+	}
+	if len(revenues) == 0 {
+		return nil, fmt.Errorf("revenue by company %d and period_id %d not found", companyID, periodID)
+	}
+
+	return revenues, nil
+}
+
+func GetRevenueByCategory(db *sql.DB, companyID int, categoryID int) ([]model.Revenue, error) {
+	rows, err := db.Query(
+		`SELECT id, company_id, currency_id, category_id, value FROM revenue_entries WHERE company_id = ? AND category_id = ?`,
+		companyID, categoryID,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("query revenue by category: %w", err)
+	}
+	defer rows.Close()
+
+	var revenues []model.Revenue
+	for rows.Next() {
+		var rc model.Revenue
+		if err := rows.Scan(&rc.ID, &rc.Company, &rc.Currency, &rc.Category, &rc.Value); err != nil {
+			return nil, fmt.Errorf("scan revenue row: %w", err)
+		}
+		revenues = append(revenues, rc)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("iterate revenue rows: %w", err)
+	}
+	if len(revenues) == 0 {
+		return nil, fmt.Errorf("revenue by company %d and category_id %d not found", companyID, categoryID)
+	}
+
+	return revenues, nil
+}