<?php

namespace App\Exports;

use Illuminate\Support\Facades\DB;
use Maatwebsite\Excel\Concerns\WithHeadings;
use Maatwebsite\Excel\Concerns\FromCollection;
use Illuminate\Support\Facades\Session;
use App\Models\User;
use App\Models\State;
use App\Models\Country;
use App\Models\Restaurant;

class RestaurantExport implements FromCollection, WithHeadings
{
	/**
	 * @return \Illuminate\Support\Collection
	 */
	public function collection()
	{

		$query = Restaurant::with(["state_name","country_name"])
    ->orderBy('created_at', 'DESC')->get(
			[
				"admin_id",
        "restaurant_name",
        "owner_name",
        "mobile",
        "email",
        "gst",
        "address",
        "city",
        "state",
        "country",
        "zip",
        "manager_name",
        "manager_mobile",
        "manager_email",
        "notes",
        "created_at",
        "updated_at"
      ]);

      foreach($query as $q)
      {
        $s_name = "";
        $c_name = "";
        if($q->state)
        {
          $s_name = $q->state_name ? $q->state_name->state_name : $q->state;
        }
        if($q->country)
        {
          $c_name = $q->country_name ? $q->country_name->name : $q->country;
        }

        $q->state = $s_name;
        $q->country = $c_name;

      }

      return $query;

	}

	public function headings(): array
	{
		return [
			"admin_id",
        "restaurant_name",
        "owner_name",
        "mobile",
        "email",
        "gst",
        "address",
        "city",
        "state",
        "country",
        "zip",
        "manager_name",
        "manager_mobile",
        "manager_email",
        "notes",
        "created_at",
        "updated_at"
		];
	}
}