From a7f3f5b79512c28e0fe9d91b3836b37182ba8e2b Mon Sep 17 00:00:00 2001
From: Denis Istratov <red__tea@outlook.com>
Date: Fri, 25 Oct 2024 12:00:00 +0300
Subject: [PATCH] update account_schema.go, repository.go

---
 pkg/repository/account_schema.go | 84 ++++++++++++++++++++++++++++++++
 pkg/repository/repository.go     | 27 ++++++++++
 2 files changed, 111 insertions(+)
 create mode 100644 pkg/repository/account_schema.go
 create mode 100644 pkg/repository/repository.go

diff --git a/pkg/repository/account_schema.go b/pkg/repository/account_schema.go
new file mode 100644
index 0000000..5291913
--- /dev/null
+++ b/pkg/repository/account_schema.go
@@ -0,0 +1,84 @@
+package repository
+
+import (
+	"esaccount"
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+type AccountSchema struct {
+	db *gorm.DB
+}
+
+func NewAccountSchema(db *gorm.DB) *AccountSchema {
+	return &AccountSchema{db: db}
+}
+
+func (r AccountSchema) CreateAccount(id int64) (*esaccount.Account, error) {
+	account := &esaccount.Account{
+		Id:          id,
+		Description: "",
+		DateOfBirth: "",
+		Follows:     []int64{},
+		Followers:   []int64{},
+		Confidentials: map[string]esaccount.ConfidentialType{
+			"username":      "ALL",
+			"first_name":    "ALL",
+			"last_name":     "ALL",
+			"email":         "NOBODY",
+			"description":   "ALL",
+			"date_of_birth": "FRIENDS",
+			"follows":       "ALL",
+			"followers":     "ALL",
+		},
+	}
+	err := r.db.Save(account).Error
+	return account, err
+}
+
+func (r AccountSchema) SaveAccount(user *esaccount.Account) (int64, error) {
+	err := r.db.Save(user).Error
+	return user.Id, err
+}
+
+func (r AccountSchema) GetAccount(id int64) (*esaccount.Account, error) {
+	user := esaccount.Account{}
+	err := r.db.Find(&user, "id = ?", id).Error
+	if user.Id == -1 {
+		err = fmt.Errorf("unknown account %d", id)
+	}
+	return &user, err
+}
+
+func (r AccountSchema) FindAccount(query, param string) (esaccount.Account, error) {
+	user := esaccount.Account{}
+	err := r.db.Find(&user, query, param).Error
+	return user, err
+}
+
+func (r AccountSchema) GetAccounts(start, limit int) []int64 {
+	result := []esaccount.Account{}
+	r.db.Select("id").Limit(limit).Offset(start).Find(&result)
+	var ans = make([]int64, len(result))
+	for i, res := range result {
+		ans[i] = res.Id
+	}
+	return ans
+}
+
+func (r AccountSchema) GetAccountsCount() int64 {
+	var count int64 = 0
+	r.db.Model(&esaccount.Account{}).Count(&count)
+	return count
+}
+
+func (r AccountSchema) DeleteAccount(id int64) error {
+	user := esaccount.Account{}
+	err := r.db.Find(&user, "id = ?", id).Error
+	if user.Id == -1 {
+		err = fmt.Errorf("unknown user %d", id)
+	}
+	r.db.Delete(user)
+	return err
+}
diff --git a/pkg/repository/repository.go b/pkg/repository/repository.go
new file mode 100644
index 0000000..dbfd594
--- /dev/null
+++ b/pkg/repository/repository.go
@@ -0,0 +1,27 @@
+package repository
+
+import (
+	"esaccount"
+
+	"gorm.io/gorm"
+)
+
+type Account interface {
+	CreateAccount(int64) (*esaccount.Account, error)
+	SaveAccount(user *esaccount.Account) (int64, error)
+	GetAccount(id int64) (*esaccount.Account, error)
+	FindAccount(query, param string) (esaccount.Account, error)
+	GetAccounts(start, limit int) []int64
+	DeleteAccount(id int64) error
+	GetAccountsCount() int64
+}
+
+type Repository struct {
+	Account
+}
+
+func NewRepository(db *gorm.DB) *Repository {
+	return &Repository{
+		Account: NewAccountSchema(db),
+	}
+}