<?php

if (!defined('BASEPATH'))
  exit('No direct script access allowed');

class Address_model extends CI_Model {
  protected $locationType='id';
  function __construct($config=array()) {
    parent::__construct();
  }

  function locationType($type){
    $this->locationType=$type;
    return $this;
  }

  function get($addressID){
    $addressID*=1;
    switch ($this->locationType){
      case 'text':
        $data=$this->db->get_where('address',array('address_id'=>$addressID),1)->row_array();
      break;
      default:
        $data=$this->db->select('a.*,l.location_id,s.state_id,c.country_id')
          ->join('location AS l','l.location_id=a.location_id','inner')
          ->join('state AS s','s.state_id=l.state_id','inner')
          ->join('country AS c','c.country_id=s.country_id','inner')
          ->get_where('address AS a',array('a.address_id'=>$addressID),1)->row_array();
        if ($data){
          $data['countries']=$this->db->get('country')->result_array();
          $data['states']=$this->db->get_where('state',array('country_id'=>$data['country_id']))->result_array();
          $data['locations']=$this->db->get_where('location',array('state_id'=>$data['state_id']))->result_array();
        }
      break;
    }

    return $data;
  }

  function save($data,$required=false){
    $CI=& get_instance();

    $addressData=$CI->input->_fetch_from_array($data,NULL,TRUE);
    $addressID=$addressData['address_id']*1;

    $dataSet=array();
    $errors=array();

    switch ($this->locationType){
      case 'text':
        $fields=array('line_1','country','location','postcode');
      break;
      default:
        $fields=array('line_1','location_id','postcode');
      break;
    }
    foreach($fields AS $f){
      if (empty($addressData[$f])){
        $errors[]=$CI->translate($addressData['type'].'_'.$f.'_should_be_set');
      }
      else {
        $dataSet[$f]=$addressData[$f];
      }
    }
    if (!empty($addressData['line_2'])){
      $dataSet['line_2']=$addressData['line_2'];
    }
    if (!empty($addressData['state'])){
      $dataSet['state']=$addressData['state'];
    }

    if ($required || $dataSet!==array()){
      if ($errors!==array()){
        foreach($errors AS $e){
          $CI->error($e);
        }
      }
    }

    if (!$CI->hasErrors() && $dataSet!==array()){
      if ($addressID==0){
        $this->db->insert('address',$dataSet);
        return $this->db->insert_id();
      }
      else {
        $this->db->where('address_id',$addressID)->update('address',$dataSet);
        return $addressID;
      }
    }
    return false;
  }

}

?>