From da6dc66fde32ab0c787763f7ab88e923076d4833 Mon Sep 17 00:00:00 2001
From: optclblast <molodoy.p4elovek@mail.ru>
Date: Sun, 5 May 2024 23:37:27 +0300
Subject: [PATCH] tmp users repo

---
 backend/internal/pkg/models/user.go           |  3 ++
 .../usecase/repository/users/repository.go    | 52 +++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/backend/internal/pkg/models/user.go b/backend/internal/pkg/models/user.go
index bb8300f..dd2650c 100644
--- a/backend/internal/pkg/models/user.go
+++ b/backend/internal/pkg/models/user.go
@@ -1,6 +1,8 @@
 package models
 
 import (
+	"time"
+
 	"github.com/google/uuid"
 )
 
@@ -15,6 +17,7 @@ type User struct {
 	Bip32Seed []byte
 	Admin     bool
 	Activated bool
+	CteatedAt time.Time
 }
 
 func NewUser(id uuid.UUID, seed []byte) *User {
diff --git a/backend/internal/usecase/repository/users/repository.go b/backend/internal/usecase/repository/users/repository.go
index 537de4d..39e163b 100644
--- a/backend/internal/usecase/repository/users/repository.go
+++ b/backend/internal/usecase/repository/users/repository.go
@@ -5,6 +5,7 @@ import (
 	"database/sql"
 	"fmt"
 
+	sq "github.com/Masterminds/squirrel"
 	"github.com/emochka2007/block-accounting/internal/pkg/models"
 	sqltools "github.com/emochka2007/block-accounting/internal/pkg/sqlutils"
 	"github.com/google/uuid"
@@ -35,6 +36,14 @@ func NewRepository(db *sql.DB) Repository {
 	}
 }
 
+func (s *repositorySQL) Conn(ctx context.Context) sqltools.DBTX {
+	if tx, ok := ctx.Value(sqltools.TxCtxKey{}).(*sql.Tx); ok {
+		return tx
+	}
+
+	return s.db
+}
+
 func (r *repositorySQL) Get(ctx context.Context, params GetParams) (*models.User, error) {
 	var user *models.User
 
@@ -48,17 +57,60 @@ func (r *repositorySQL) Get(ctx context.Context, params GetParams) (*models.User
 }
 
 func (r *repositorySQL) Create(ctx context.Context, user *models.User) error {
+	if err := sqltools.Transaction(ctx, r.db, func(ctx context.Context) error {
+		query := sq.Insert("users").Columns(
+			"id", "seed", "created_at",
+		).Values(
+			user.ID,
+			user.Bip32Seed,
+			user.CteatedAt,
+		)
 
+		if user.Activated {
+			query = query.Columns("activated_at").Values(user.CteatedAt)
+		}
+
+		if _, err := query.RunWith(r.Conn(ctx)).ExecContext(ctx); err != nil {
+			return fmt.Errorf("error insert new user. %w", err)
+		}
+
+		return nil
+	}); err != nil {
+		return fmt.Errorf("error execute transactional operation. %w", err)
+	}
+
+	return nil
 }
 
 func (r *repositorySQL) Activate(ctx context.Context, id string) error {
+	if err := sqltools.Transaction(ctx, r.db, func(ctx context.Context) error {
 
+		return nil
+	}); err != nil {
+		return fmt.Errorf("error execute transactional operation. %w", err)
+	}
+
+	return nil
 }
 
 func (r *repositorySQL) Update(ctx context.Context, user *models.User) error {
+	if err := sqltools.Transaction(ctx, r.db, func(ctx context.Context) error {
 
+		return nil
+	}); err != nil {
+		return fmt.Errorf("error execute transactional operation. %w", err)
+	}
+
+	return nil
 }
 
 func (r *repositorySQL) Delete(ctx context.Context, id string) error {
+	if err := sqltools.Transaction(ctx, r.db, func(ctx context.Context) error {
 
+		return nil
+	}); err != nil {
+		return fmt.Errorf("error execute transactional operation. %w", err)
+	}
+
+	return nil
 }