rev cat getter by company id

internal/database/category.go

@@ -52,3 +52,54 @@ func InsertCategory(db *sql.DB, rc model.RevenueCategory) error {
 	}
 	return nil
 }
+
+func GetCategoriesByCompanyID(db *sql.DB, companyID int) ([]string, error) {
+	rows, err := db.Query("SELECT name FROM category WHERE company_id = ?", companyID)
+	if err != nil {
+		return []string{}, err
+	}
+	defer rows.Close()
+
+	var list []string
+	for rows.Next() {
+		var Name string
+		err := rows.Scan(&Name)
+		if err != nil {
+			return []string{}, err
+		}
+
+		list = append(list, Name)
+	}
+	return list, nil
+}
+
+func GetCategoryByID(db *sql.DB, companyID int, ID int) (*model.RevenueCategory, error) {
+	var rc model.RevenueCategory
+	err := db.QueryRow(
+		`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND id = ?`,
+		companyID, ID,
+	).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
+	if err == sql.ErrNoRows {
+		return &rc, fmt.Errorf("category %q not found for company %d", ID, companyID)
+	}
+	if err != nil {
+		return &rc, fmt.Errorf("get category by name: %w", err)
+	}
+	return &rc, nil
+}
+
+func GetCategoryByName(db *sql.DB, companyID int, name string) (model.RevenueCategory, error) {
+
+	var rc model.RevenueCategory
+	err := db.QueryRow(
+		`SELECT id, company_id, parent_id, name FROM category WHERE company_id = ? AND name = ?`,
+		companyID, name,
+	).Scan(&rc.ID, &rc.CompanyID, &rc.ParentID, &rc.Name)
+	if err == sql.ErrNoRows {
+		return rc, fmt.Errorf("category %q not found for company %d", name, companyID)
+	}
+	if err != nil {
+		return rc, fmt.Errorf("get category by name: %w", err)
+	}
+	return rc, nil
+}