added the adjust, view and remove budget funcationality

cmd/budget/adjust.go

@@ -1,8 +1,9 @@
 package budget
 
 import (
-	"fmt"
+	"log"
 
+	"github.com/ibilalkayy/flow/internal/app"
 	"github.com/spf13/cobra"
 )
 
@@ -14,9 +15,10 @@ var adjustCmd = &cobra.Command{
 		oldCategory, _ := cmd.Flags().GetString("oldcategory")
 		newCategory, _ := cmd.Flags().GetString("newcategory")
 		amount, _ := cmd.Flags().GetString("amount")
-		fmt.Println(oldCategory)
-		fmt.Println(newCategory)
-		fmt.Println(amount)
+		err := app.UpdateBudget(oldCategory, newCategory, amount)
+		if err != nil {
+			log.Fatal(err)
+		}
 	},
 }
 

cmd/budget/remove.go

@@ -1,8 +1,9 @@
 package budget
 
 import (
-	"fmt"
+	"log"
 
+	"github.com/ibilalkayy/flow/internal/app"
 	"github.com/spf13/cobra"
 )
 
@@ -12,7 +13,10 @@ var removeCmd = &cobra.Command{
 	Short: "Remove the budget details",
 	Run: func(cmd *cobra.Command, args []string) {
 		category, _ := cmd.Flags().GetString("category")
-		fmt.Println(category)
+		err := app.RemoveBudget(category)
+		if err != nil {
+			log.Fatal(err)
+		}
 	},
 }
 

cmd/budget/view.go

@@ -2,7 +2,9 @@ package budget
 
 import (
 	"fmt"
+	"log"
 
+	"github.com/ibilalkayy/flow/internal/app"
 	"github.com/spf13/cobra"
 )
 
@@ -12,7 +14,11 @@ var viewCmd = &cobra.Command{
 	Short: "View the budget details",
 	Run: func(cmd *cobra.Command, args []string) {
 		category, _ := cmd.Flags().GetString("category")
-		fmt.Println(category)
+		details, err := app.ViewBudget(category)
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Budget Details:\n\tCategory: %s\n\tAmount: %s", details[0], details[1])
 	},
 }
 

db/connection.go

@@ -18,7 +18,7 @@ type Variables struct {
 	DBName   string
 }
 
-func connection() (*sql.DB, error) {
+func Connection() (*sql.DB, error) {
 	v := Variables{
 		Host:     middleware.LoadEnvVariable("host"),
 		Port:     middleware.LoadEnvVariable("port"),
@@ -41,7 +41,7 @@ func connection() (*sql.DB, error) {
 }
 
 func Table(table_name, filename string, number int) (*sql.DB, error) {
-	db, err := connection()
+	db, err := Connection()
 	if err != nil {
 		return nil, err
 	}

internal/app/budget.go

@@ -1,6 +1,7 @@
 package app
 
 import (
+	"errors"
 	"fmt"
 
 	"github.com/ibilalkayy/flow/db"
@@ -32,3 +33,71 @@ func CreateBudget(bv *BudgetVariables) error {
 	fmt.Println("Budget data is successfully inserted!")
 	return nil
 }
+
+func ViewBudget(category string) ([2]string, error) {
+	bv := new(BudgetVariables)
+	db, err := db.Connection()
+	if err != nil {
+		return [2]string{}, err
+	}
+
+	query := "SELECT categories, amounts FROM Budget WHERE categories=$1"
+	if err := db.QueryRow(query, category).Scan(&bv.Category, &bv.Amount); err != nil {
+		return [2]string{}, err
+	}
+	return [2]string{bv.Category, bv.Amount}, err
+}
+
+func RemoveBudget(category string) error {
+	db, err := db.Connection()
+	if err != nil {
+		return err
+	}
+
+	query := "DELETE FROM Budget WHERE categories=$1"
+	remove, err := db.Prepare(query)
+	if err != nil {
+		return err
+	}
+
+	defer remove.Close()
+
+	if len(category) != 0 {
+		_, err = remove.Exec(category)
+		if err != nil {
+			return err
+		}
+		fmt.Printf("%s category is successfully removed!", category)
+	} else {
+		fmt.Println("First enter the category and then remove it")
+	}
+	return nil
+}
+
+func UpdateBudget(old, new, amount string) error {
+	db, err := db.Connection()
+	if err != nil {
+		return err
+	}
+
+	// Check if the old category exists
+	var count int
+	err = db.QueryRow("SELECT COUNT(*) FROM Budget WHERE categories = $1", old).Scan(&count)
+	if err != nil {
+		return err
+	}
+
+	// If the old category does not exist, return an error
+	if count == 0 {
+		return errors.New("'" + old + "'" + " category does not exist")
+	}
+
+	// If the old category exists, execute the update query
+	query := "UPDATE Budget SET categories=$1, amounts=$2 WHERE categories=$3"
+	_, err = db.Exec(query, new, amount, old)
+	if err != nil {
+		return err
+	}
+	fmt.Println("Your budget category is successfully updated!")
+	return nil
+}