<?php

class Assignment_model extends MO_Model
{
    public function __construct()
    {
        parent::__construct();
    }

    public function assign_unit($assid, $num, $slotid, $res, $villageid, $userid)
    {
        $this->set_resources($res);

        //check if there is building in that slot
        $sql = "SELECT buildings.* FROM village_buildings
		INNER JOIN buildings ON village_buildings.buildingid=buildings.id
		WHERE slotid='$slotid' AND villageid='$villageid'";

        $q = $this->db->query($sql);

        if (!$q->num_rows()) {
            return 1;
        }

        $building = $q->row_array();

        //check if there is already an assignment in which case we just deassign
        //in this case num will be 0 unless they play with the form,
        //but we shouldn't care about that since the user can write 0 into the form
        $sql = "SELECT * FROM building_assignments
	        WHERE assignmentid='$assid'
		AND slotid='$slotid'
		AND villageid='$villageid'";
        $q = $this->db->query($sql);

        if ($q->num_rows()) {
            //the user wants to just deassign
            $ass = $q->row_array();

            //delete assignment
            $sql = "DELETE FROM building_assignments WHERE id='" . $ass['id'] . "'";
            $this->db->query($sql);

            //giving back units
            
            //getting units
            $sql = "SELECT * FROM village_units 
		    WHERE villageid='$villageid'
	       	    AND userid='$userid'";
            $q = $this->db->query($sql);

            if ($q->num_rows()) {
                $units = $q->result_array();
            } else {
                $units = false;
            }

            $found = false;

            if ($units) {
                foreach ($units as $row) {
                    if ($row['unitid'] == $ass['unitid']) {
                        $found = true;
                        $fdata = $row['id'];
                    }
                }
            }

            if ($found) {
                $sql = "UPDATE village_units
			SET unitcount = unitcount + '". $ass['num_unit'] . "'
			WHERE id='$fdata'";
            } else {
                $sql = "INSERT INTO village_units
			VALUES(default, '$userid', '$villageid', 
			'" . $ass['unitid'] . "', '" . $ass['num_unit'] . "')";
            }

            $this->db->query($sql);

            //substract the bonuses
            $this->substract_modifiers(
                $assid,
                $villageid,
                'assignment',
                $ass['num_bonus']
            );

            $this->write_resources();

            //taking away spells
            $sql = "DELETE FROM building_spells 
				WHERE assignmentid='$assid'
				AND slotid='$slotid'
				AND villageid='$villageid'";

            $this->db->query($sql);

            return 0;
        }

        //check if assignment exists
        $sql = "SELECT * FROM assignments WHERE id='$assid'";
        $q = $this->db->query($sql);

        if (!$q->num_rows()) {
            return 2;
        }

        $ass = $q->row_array();

        //checking technology
        if ($ass['req_tech']) {
            if (!$this->has_req_tech($ass['req_tech'], $villageid, $slotid)) {
                return 3;
            }
        }
        
        //check if building has that assignment
        if (!($building['assignment1'] == $ass['id'] ||
      $building['assignment2'] == $ass['id'] ||
      $building['assignment3'] == $ass['id'] ||
      $building['assignment4'] == $ass['id'] ||
      $building['assignment5'] == $ass['id'])) {
            return 3;
        }

        //getting unit
        $sql = "SELECT * FROM village_units 
			WHERE unitid='" . $ass['unitid'] . "'
			AND userid='$userid'";
        $q = $this->db->query($sql);

        if (!$q->num_rows()) {
            return 4;
        }

        $unit = $q->row_array();

        //it probably shouldn't happen but who knows
        if (!$unit['unitcount']) {
            return 5;
        }

        //if the user just clicks on the assign we should assign max
        if (!$num) {
            $num = $ass['max'];
        }

        //we shouldn't assign more than allowed
        if ($ass['max'] < $num) {
            $num = $ass['max'];
        }

        //determining how much can be assigned
        if ($unit['unitcount'] < $num) {
            $num = $unit['unitcount'];
        }

        //determining number of bonuses granted
        $num_bonus = floor(($num / $ass['bonus_per_assigned']));

        $count = ($unit['unitcount'] - $num);

        //updating unit field in db
        if ($count <= 0) {
            $sql = "DELETE FROM village_units WHERE id='" . $unit['id'] . "'";
        } else {
            $sql = "UPDATE village_units
				SET unitcount=unitcount - '$num'
				WHERE id='" . $unit['id'] . "'";
        }

        $this->db->query($sql);

        //adding assignment
        $sql = "INSERT INTO building_assignments
			VALUES(default, '$villageid', '$slotid', '" . $unit['unitid'] . "', 
				'$num', '" . $assid . "', '$num_bonus')";

        $this->db->query($sql);

        //grant resource bonus
        $this->add_modifiers($assid, $villageid, 'assignment', $num_bonus);
        $this->write_resources();

        //assignment has spell
        if (!$ass['spellid']) {
            return 0;
        }

        /*
        //only give spells if bonus is granted
        if (!$num_bonus)
        return 0;
        */

        //only give spell if max unit assigned
        if (!($ass['max'] == $num)) {
            return 0;
        }

        //granting spell
        $sql = "INSERT INTO building_spells
			VALUES(default, '$villageid', '$slotid', '$assid', 
			'" . $ass['spellid'] . "')";

        $this->db->query($sql);

        return 0;
    }

    public function get_assignments($slotid, $villageid, $userid)
    {
        if ($slotid > parent::TOTAL_BUILDINGS) {
            return 1;
        } //that shouldn't happen

        $sql = "SELECT buildings.* FROM village_buildings
			INNER JOIN buildings ON village_buildings.buildingid = buildings.id
			WHERE village_buildings.slotid='$slotid' 
			AND village_buildings.villageid='$villageid'";

        $q = $this->db->query($sql);

        if (!$q->num_rows()) {
            return 2;
        } //nothing in that slot

        $res = $q->row_array();

        //getting assignment data
        $sql = "SELECT assignments.*,units.name 
			FROM assignments
			INNER JOIN units ON assignments.unitid=units.id
			WHERE assignments.id='" . $res['assignment1'] . "'
			OR assignments.id='" . $res['assignment2'] . "'
			OR assignments.id='" . $res['assignment3'] . "'
			OR assignments.id='" . $res['assignment4'] . "'
			OR assignments.id='" . $res['assignment5'] . "'";

        $q = $this->db->query($sql);


        if (!$q->num_rows()) {
            $data['assigndata'] = false; //building doesn't have assignments
            return $data;
        }

        $adata = $q->result_array();

        $tech = $this->get_village_technologies($villageid, $slotid);

        foreach ($adata as $row) {
            if ($this->have_technology($tech, $row['req_tech'])) {
                $data['assigndata'][] = $row;
            }
        }

        if (!isset($data['assigndata'])) {
            $data['assigndata'] = false;
            return $data;
        }

        //getting assigned data
        $sql = "SELECT * FROM building_assignments
			WHERE slotid='$slotid' 
			AND villageid='$villageid'";

        $q = $this->db->query($sql);

        if (!$q->num_rows()) {
            $data['assigned'] = false;
        } else {
            $data['assigned'] = $q->result_array();
        }

        $data['building'] = $res;

        //getting units
        $sql = "SELECT * FROM village_units 
			WHERE villageid='$villageid' 
			AND userid='$userid'";
        $q = $this->db->query($sql);

        if ($q->num_rows) {
            $data['units'] = $q->result_array();
        } else {
            $data['units'] = false;
        }

        return $data;
    }

    public function get_assignment_list_drop_admin()
    {
        $sql = "SELECT * FROM assignments";
        $q = $this->db->query($sql);

        $res = $q->result_array();

        $data[0] = 'Nothing';

        foreach ($res as $row) {
            $data[$row['id']] = $row['description'];
        }

        return $data;
    }

    public function get_assignments_admin()
    {
        $sql = "SELECT * FROM assignments";

        $q = $this->db->query($sql);

        return $q->result_array();
    }

    public function add_assignment_admin($data)
    {
        $sql = "INSERT INTO assignments
			VALUES (default,
			'" . $data['unitid'] . "',
			'" . $data['max'] . "',
			'" . $data['bonus_per_assigned'] . "',
			'" . $data['spellid'] . "',
			'" . $data['req_tech'] . "',
			'" . $data['mod_max_food'] . "',
			'" . $data['mod_max_wood'] . "',
			'" . $data['mod_max_stone'] . "',
			'" . $data['mod_max_iron'] . "',
			'" . $data['mod_max_mana'] . "',
			'" . $data['mod_rate_food'] . "',
			'" . $data['mod_rate_wood'] . "',
			'" . $data['mod_rate_stone'] . "',
			'" . $data['mod_rate_iron'] . "',
			'" . $data['mod_rate_mana'] . "',
			'" . $data['mod_percent_food'] . "',
			'" . $data['mod_percent_wood'] . "',
			'" . $data['mod_percent_stone'] . "',
			'" . $data['mod_percent_iron'] . "',
			'" . $data['mod_percent_mana'] . "',
			'" . $data['description'] . "')";

        $this->db->query($sql);

        $this->_create_sql($sql);
    }

    public function edit_assignment_admin($data)
    {
        $sql = "UPDATE assignments
			SET unitid='" . $data['unitid'] . "',
			max='" . $data['max'] . "',
			bonus_per_assigned='" . $data['bonus_per_assigned'] . "',
			spellid='" . $data['spellid'] . "',
			req_tech='" . $data['req_tech'] . "',
			mod_max_food='" . $data['mod_max_food'] . "',
			mod_max_wood='" . $data['mod_max_wood'] . "',
			mod_max_stone='" . $data['mod_max_stone'] . "',
			mod_max_iron='" . $data['mod_max_iron'] . "',
			mod_max_mana='" . $data['mod_max_mana'] . "',
			mod_rate_food='" . $data['mod_rate_food'] . "',
			mod_rate_wood='" . $data['mod_rate_wood'] . "',
			mod_rate_stone='" . $data['mod_rate_stone'] . "',
			mod_rate_iron='" . $data['mod_rate_iron'] . "',
			mod_rate_mana='" . $data['mod_rate_mana'] . "',
			mod_percent_food='" . $data['mod_percent_food'] . "',
			mod_percent_wood='" . $data['mod_percent_wood'] . "',
			mod_percent_stone='" . $data['mod_percent_stone'] . "',
			mod_percent_iron='" . $data['mod_percent_iron'] . "',
			mod_percent_mana='" . $data['mod_percent_mana'] . "',
			description='" . $data['description'] . "'
			WHERE id='" . $data['id'] . "'";

        $this->db->query($sql);

        $this->_create_sql($sql);
    }

    public function get_assignment_admin($id)
    {
        $sql = "SELECT * FROM assignments WHERE id='$id'";

        $q = $this->db->query($sql);

        return $q->row_array();
    }
}
//nowhitesp