create and get categories with database for user

backend/database/categories.go

@@ -13,7 +13,9 @@ import (
 
 func CreateCategory(category *model.Category) error {
 	e := Db.Where(model.Category{
-		Title: category.Title,
+		Title:  category.Title,
+		UserID: category.UserID,
+		Type:   category.Type,
 	}).FirstOrCreate(&category).Error
 	if e != nil {
 		return e
@@ -75,9 +77,9 @@ func GetCategoriesFromFile(filename string) []model.Category {
 	return res
 }
 
-func GetAllCategories() ([]model.Category, error) {
+func GetAllCategories(userid string) ([]model.Category, error) {
 	var categories []model.Category
-	e := Db.Find(&categories).Error
+	e := Db.Where(model.Category{UserID: userid}).Find(&categories).Error
 	if e != nil {
 		return nil, e
 	}

backend/database/db.go

@@ -41,13 +41,16 @@ func InitDatabase() {
 		allusers := GetUsersFromFile(os.Getenv("USERS_FILE"))
 		for i := range allusers {
 			CreateUser(&allusers[i])
-		}
-	}
-
-	if os.Getenv("DEFAULT_CATEGORIES_FILE") != "" {
-		default_categories := GetCategoriesFromFile(os.Getenv("DEFAULT_CATEGORIES_FILE"))
-		for i := range default_categories {
-			CreateCategory(&default_categories[i])
+			if os.Getenv("DEFAULT_CATEGORIES_FILE") != "" {
+				default_categories := GetCategoriesFromFile(os.Getenv("DEFAULT_CATEGORIES_FILE"))
+				for j := range default_categories {
+					default_categories[j].UserID = allusers[i].ID
+					for k := range default_categories[j].Subcategories {
+						default_categories[j].Subcategories[k].UserID = allusers[i].ID
+					}
+					CreateCategory(&default_categories[j])
+				}
+			}
 		}
 	}
 }