<?php

namespace App\Exports;

use App\Models\Auth\Role;
use Maatwebsite\Excel\Concerns\FromCollection;
use Maatwebsite\Excel\Concerns\WithHeadings;
use Maatwebsite\Excel\Concerns\WithMapping;
use Maatwebsite\Excel\Concerns\ShouldAutoSize;

class RoleExport implements FromCollection, WithHeadings, WithMapping, ShouldAutoSize
{
    protected $roleIds;

    public function __construct(array $roleIds = [])
    {
        $this->roleIds = $roleIds;
    }

    /**
     * 获取数据集合
     */
    public function collection()
    {
        $query = Role::with(['permissions']);

        if (!empty($this->roleIds)) {
            $query->whereIn('id', $this->roleIds);
        }

        return $query->get();
    }

    /**
     * 设置表头
     */
    public function headings(): array
    {
        return [
            'ID',
            '角色名称',
            '角色编码',
            '描述',
            '权限',
            '排序',
            '状态',
            '创建时间',
        ];
    }

    /**
     * 映射数据
     */
    public function map($role): array
    {
        return [
            $role->id,
            $role->name,
            $role->code,
            $role->description,
            $role->permissions->pluck('title')->implode(','),
            (int)$role->sort,
            $role->status == 1 ? '启用' : '禁用',
            $role->created_at ? (string)$role->created_at : '',
        ];
    }
}