add budget feature

core/model/budget.go

@@ -0,0 +1,58 @@
+package model
+
+import (
+	"errors"
+	"github.com/openaccounting/oa-server/core/model/types"
+)
+
+type BudgetInterface interface {
+	GetBudget(string, string) (*types.Budget, error)
+	CreateBudget(*types.Budget, string) error
+	DeleteBudget(string, string) error
+}
+
+func (model *Model) GetBudget(orgId string, userId string) (*types.Budget, error) {
+	belongs, err := model.UserBelongsToOrg(userId, orgId)
+
+	if err != nil {
+		return nil, err
+	}
+
+	if belongs == false {
+		return nil, errors.New("User does not belong to org")
+	}
+
+	return model.db.GetBudget(orgId)
+}
+
+func (model *Model) CreateBudget(budget *types.Budget, userId string) error {
+	belongs, err := model.UserBelongsToOrg(userId, budget.OrgId)
+
+	if err != nil {
+		return err
+	}
+
+	if belongs == false {
+		return errors.New("User does not belong to org")
+	}
+
+	if budget.OrgId == "" {
+		return errors.New("orgId required")
+	}
+
+	return model.db.InsertAndReplaceBudget(budget)
+}
+
+func (model *Model) DeleteBudget(orgId string, userId string) error {
+	belongs, err := model.UserBelongsToOrg(userId, orgId)
+
+	if err != nil {
+		return err
+	}
+
+	if belongs == false {
+		return errors.New("User does not belong to org")
+	}
+
+	return model.db.DeleteBudget(orgId)
+}