ragflow/internal/dao/user.go

//
//  Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

package dao

import (
	"ragflow/internal/entity"
)

// UserDAO user data access object
type UserDAO struct{}

// NewUserDAO create user DAO
func NewUserDAO() *UserDAO {
	return &UserDAO{}
}

// Create create user
func (dao *UserDAO) Create(user *entity.User) error {
	return DB.Create(user).Error
}

// GetByID get user by ID
func (dao *UserDAO) GetByID(id uint) (*entity.User, error) {
	var user entity.User
	err := DB.First(&user, id).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (dao *UserDAO) GetByTenantID(tenantID string) (*entity.User, error) {
	var user entity.User
	err := DB.Where("id = ?", tenantID).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

// GetByEmail get user by email
func (dao *UserDAO) GetByEmail(email string) (*entity.User, error) {
	var user entity.User
	query := DB.Where("email = ?", email)
	err := query.First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

// GetByAccessToken get user by access token
func (dao *UserDAO) GetByAccessToken(token string) (*entity.User, error) {
	var user entity.User
	err := DB.Where("access_token = ?", token).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

// Update update user
func (dao *UserDAO) Update(user *entity.User) error {
	return DB.Save(user).Error
}

// UpdateAccessToken update user's access token
func (dao *UserDAO) UpdateAccessToken(user *entity.User, token string) error {
	return DB.Model(user).Update("access_token", token).Error
}

// List list users (only active users with status != "0")
func (dao *UserDAO) List(offset, limit int) ([]*entity.User, int64, error) {
	var users []*entity.User
	var total int64

	// Only count users with status != "0" (not deleted)
	if err := DB.Model(&entity.User{}).Count(&total).Error; err != nil {
		return nil, 0, err
	}

	query := DB.Model(&entity.User{})
	if offset > 0 {
		query = query.Offset(offset)
	}
	if limit > 0 {
		query = query.Limit(limit)
	}
	err := query.Find(&users).Error
	return users, total, err
}

// Delete delete user
func (dao *UserDAO) Delete(id uint) error {
	return DB.Delete(&entity.User{}, id).Error
}

// DeleteByID delete user by string ID (soft delete - set status to 0)
func (dao *UserDAO) DeleteByID(id string) error {
	return DB.Model(&entity.User{}).Where("id = ?", id).Update("status", "0").Error
}

// HardDelete hard delete user by string ID
func (dao *UserDAO) HardDelete(id string) error {
	return DB.Unscoped().Where("id = ?", id).Delete(&entity.User{}).Error
}

// ListByEmail list users by email (only active users with status != "0")
// Returns all users matching the given email address
func (dao *UserDAO) ListByEmail(email string) ([]*entity.User, error) {
	var users []*entity.User
	err := DB.Where("email = ?", email).Find(&users).Error
	return users, err
}