package repository

import (
	"context"
	"database/sql"
	"fmt"

	"github.com/muyuqingfeng/iloom/auth-service/internal/model"
)

type ProfileRepo struct {
	db *sql.DB
}

func NewProfileRepo(db *sql.DB) *ProfileRepo {
	return &ProfileRepo{db: db}
}

func (r *ProfileRepo) GetByID(ctx context.Context, userID string) (*model.Profile, error) {
	query := `SELECT user_id, username, real_name, phone, email, avatar, status, created_at
		FROM sys_user WHERE user_id = ? AND deleted_at IS NULL`
	p := &model.Profile{}
	err := r.db.QueryRowContext(ctx, query, userID).Scan(
		&p.ID, &p.Username, &p.RealName, &p.Phone, &p.Email,
		&p.Avatar, &p.Status, &p.CreatedAt,
	)
	if err != nil {
		if err == sql.ErrNoRows {
			return nil, nil
		}
		return nil, fmt.Errorf("get profile by id: %w", err)
	}
	return p, nil
}

func (r *ProfileRepo) GetByUsername(ctx context.Context, username string) (*model.Profile, error) {
	query := `SELECT user_id, username, real_name, phone, email, avatar, status, created_at
		FROM sys_user WHERE username = ? AND deleted_at IS NULL`
	p := &model.Profile{}
	err := r.db.QueryRowContext(ctx, query, username).Scan(
		&p.ID, &p.Username, &p.RealName, &p.Phone, &p.Email,
		&p.Avatar, &p.Status, &p.CreatedAt,
	)
	if err != nil {
		if err == sql.ErrNoRows {
			return nil, nil
		}
		return nil, fmt.Errorf("get profile by username: %w", err)
	}
	return p, nil
}

func (r *ProfileRepo) GetWithCompany(ctx context.Context, userID string) (*model.Profile, error) {
	query := `SELECT su.user_id, su.username, su.real_name, su.phone, su.email, su.avatar, su.status, su.created_at,
		COALESCE(cm.company_id, ''), COALESCE(c.name, ''), COALESCE(c.type, ''), COALESCE(cm.is_master, 0)
		FROM sys_user su
		LEFT JOIN ilm_company_member cm ON cm.user_id = su.user_id
		LEFT JOIN ilm_company c ON c.id = cm.company_id
		WHERE su.user_id = ? AND su.deleted_at IS NULL
		LIMIT 1`
	p := &model.Profile{}
	err := r.db.QueryRowContext(ctx, query, userID).Scan(
		&p.ID, &p.Username, &p.RealName, &p.Phone, &p.Email,
		&p.Avatar, &p.Status, &p.CreatedAt,
		&p.CompanyID, &p.CompanyName, &p.Role, &p.IsMaster,
	)
	if err != nil {
		if err == sql.ErrNoRows {
			return nil, nil
		}
		return nil, fmt.Errorf("get profile with company: %w", err)
	}
	return p, nil
}