64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// NewBudget creates a budget and adds it to the current user.
|
|
func (s *Database) NewBudget(context context.Context, name string, userID uuid.UUID) (*Budget, error) {
|
|
tx, err := s.BeginTx(context, &sql.TxOptions{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("begin transaction: %w", err)
|
|
}
|
|
|
|
transaction := s.WithTx(tx)
|
|
budget, err := transaction.CreateBudget(context, CreateBudgetParams{
|
|
Name: name,
|
|
IncomeCategoryID: uuid.New(),
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create budget: %w", err)
|
|
}
|
|
|
|
ub := LinkBudgetToUserParams{UserID: userID, BudgetID: budget.ID}
|
|
_, err = transaction.LinkBudgetToUser(context, ub)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("link budget to user: %w", err)
|
|
}
|
|
|
|
group, err := transaction.CreateCategoryGroup(context, CreateCategoryGroupParams{
|
|
Name: "Inflow",
|
|
BudgetID: budget.ID,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create inflow category_group: %w", err)
|
|
}
|
|
|
|
cat, err := transaction.CreateCategory(context, CreateCategoryParams{
|
|
Name: "Ready to Assign",
|
|
CategoryGroupID: group.ID,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create ready to assign category: %w", err)
|
|
}
|
|
|
|
err = transaction.SetInflowCategory(context, SetInflowCategoryParams{
|
|
IncomeCategoryID: cat.ID,
|
|
ID: budget.ID,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("set inflow category: %w", err)
|
|
}
|
|
|
|
err = tx.Commit()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("commit: %w", err)
|
|
}
|
|
|
|
return &budget, nil
|
|
}
|