common/repository/yclients/yclients.go

package yclients

import (
	"context"
	"database/sql"
	"gitea.pena/SQuiz/common/dal/sqlcgen"
	"gitea.pena/SQuiz/common/model"
)

type YclientsRepository struct {
	queries *sqlcgen.Queries
	pool    *sql.DB
}

type Deps struct {
	Queries *sqlcgen.Queries
	Pool    *sql.DB
}

func NewYclientsRepository(deps Deps) *YclientsRepository {
	return &YclientsRepository{
		queries: deps.Queries,
		pool:    deps.Pool,
	}
}

func (r *YclientsRepository) GetCurrentAccount(ctx context.Context, accountID string) (*model.YclientsAccount, error) {
	row, err := r.queries.GetCurrentYclientsCompany(ctx, accountID)
	if err != nil {
		return nil, err
	}

	user := model.YclientsAccount{
		ID:         row.ID,
		AccountID:  row.Accountid,
		YclientsID: row.Yclientsid,
		Name:       row.Name,
		Country:    row.Country,
		Deleted:    row.Deleted,
		CreatedAt:  row.Createdat,
		Subdomain:  row.Subdomain,
	}

	//_, err = r.queries.CheckExpiredYclientsToken(ctx, accountID)
	//if err != nil {
	//	if err == sql.ErrNoRows {
	//		user.Stale = false
	//		return &user, nil
	//	}
	//	return nil, err
	//}

	user.Stale = true

	return &user, nil
}

//func (r *YclientsRepository) SoftDeleteAccount(ctx context.Context, accountID string) error {
//	err := r.queries.SoftDeleteYclientsAccount(ctx, accountID)
//	if err != nil {
//		return err
//	}
//	return nil
//}

func (r *YclientsRepository) GettingUserWithPagination(ctx context.Context, req *model.PaginationReq, accountID string) (*model.UserListYclientsResp, error) {
	rows, err := r.queries.GetUsersYclientsWithPagination(ctx, sqlcgen.GetUsersYclientsWithPaginationParams{
		Accountid: accountID,
		Column2:   req.Page,
		Limit:     req.Size,
	})
	if err != nil {
		return nil, err
	}
	var users []model.YclientsAccountUser
	var count int64
	for _, row := range rows {
		users = append(users, model.YclientsAccountUser{
			ID:             row.ID,
			YclientsID:     row.Yclientsid,
			YclientsUserID: row.Yclientsuserid,
			Name:           row.Name,
			Role:           row.Role,
			Deleted:        row.Deleted,
			CreatedAt:      row.Createdat,
		})
		count = row.TotalCount
	}

	resp := model.UserListYclientsResp{
		Count: count,
		Items: users,
	}

	return &resp, nil
}

// company

func (r *YclientsRepository) GettingCompanyWithPagination(ctx context.Context, req *model.PaginationReq, accountID string) (*model.CompanyListYclientsResp, error) {
	rows, err := r.queries.GetCompanyYclientsWithPagination(ctx, sqlcgen.GetCompanyYclientsWithPaginationParams{
		Accountid: accountID,
		Column2:   req.Page,
		Limit:     req.Size,
	})
	if err != nil {
		return nil, err
	}
	var users []model.YclientsCompany
	var count int64
	for _, row := range rows {
		users = append(users, model.YclientsCompany{
			ID:                row.ID,
			YclientsID:        row.Yclientsid,
			YclientsCompanyID: row.Yclientscompanyid,
			Title:             row.Title,
			ShortDecription:   row.Shortdecription,
			Active:            row.Active,
			Country:           row.Country,
			GroupPriority:     row.Grouppriority,
			Deleted:           row.Deleted,
			CreatedAt:         row.Createdat,
		})
		count = row.TotalCount
	}

	resp := model.CompanyListYclientsResp{
		Count: count,
		Items: users,
	}

	return &resp, nil
}

// todo
func (r *YclientsRepository) UpdateCompany() {

}

// services

func (r *YclientsRepository) GettingServicesWithPagination(ctx context.Context, req *model.PaginationReq, accountID string) (*model.ServicesListYclientsResp, error) {
	rows, err := r.queries.GetServicesYclientsWithPagination(ctx, sqlcgen.GetServicesYclientsWithPaginationParams{
		Accountid: accountID,
		Column2:   req.Page,
		Limit:     req.Size,
	})
	if err != nil {
		return nil, err
	}
	var users []model.YclientsServices
	var count int64
	for _, row := range rows {
		users = append(users, model.YclientsServices{
			ID:         row.ID,
			YclientsID: row.Yclientsid,
			ServiceID:  row.Serviceid,
			Title:      row.Title,
			CategoryID: row.Categoryid,
			PriceMin:   row.Pricemin,
			PriceMax:   row.Pricemax,
			Discount:   row.Discount,
			Comment:    row.Comment,
			Active:     row.Active,
			ApiID:      row.Apiid,
			Deleted:    row.Deleted,
			CreatedAt:  row.Createdat,
		})
		count = row.TotalCount
	}

	resp := model.ServicesListYclientsResp{
		Count: count,
		Items: users,
	}

	return &resp, nil
}

// todo
func (r *YclientsRepository) UpdateServices() {

}