// REPLACE THIS METHOD in app/Http/Controllers/Api/CompanyController.php
// Replace the index method (around lines 34-64)

    public function index(Request $request)
    {
        $query = Company::query();

        // Filter by status
        if ($request->has('status')) {
            switch ($request->status) {
                case 'approved':
                    $query->where('is_approved', true);
                    break;
                case 'pending':
                    $query->pending();
                    break;
                case 'custom':
                    $query->custom();
                    break;
            }
        }

        // Search
        if ($request->has('search')) {
            $query->where('name', 'like', '%' . $request->search . '%');
        }

        // Select only necessary columns
        $companies = $query->select('id', 'name', 'industry', 'is_approved', 'is_custom', 'user_id', 'created_at')
            ->orderBy('created_at', 'desc')
            ->paginate(20);

        // Only load user relationship if needed (when showing custom companies)
        if ($request->status === 'custom' || $request->status === 'pending') {
            $companies->load('user:id,name,email');
        }

        return response()->json([
            'success' => true,
            'data' => $companies,
        ]);
    }