<?php

namespace Forest\Repositories\Imports;

use Forest\Models\ImportationColumn;

use Auth, DB;

use Forest\Repositories\BaseRepository;

class ImportationColumnsRepository extends BaseRepository{

    protected $entity = ImportationColumn::class;

    /*
    * Pegar as colunas de uma importação
    */
    public function getImportationColumns($data)
    {
    	$importation_column = ImportationColumn::
    		where('importation_id', '=', $data['importation_id'])
    		->get();

        return $importation_column;
    }

    /*
    * Criar as colunas da importação
    */
    public function store($data)
    {
    	$importation_column = new ImportationColumn;
        $importation_column->column_name = $data['column_name'];
        $importation_column->importation_id = $data['importation_id'];
        $importation_column->save();

        return $importation_column;
    }
}