SettlementController.php

<?php

namespace BM2\SiteBundle\Controller;

use BM2\SiteBundle\Entity\Settlement;
use BM2\SiteBundle\Form\SettlementPermissionsSetType;
use BM2\SiteBundle\Form\SoldiersManageType;
use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
use Sensio\Bundle\FrameworkExtraBundle\Configuration\Template;
use Symfony\Bundle\FrameworkBundle\Controller\Controller;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpKernel\Exception\AccessDeniedHttpException;


/**
 * @Route("/settlement")
 */
class SettlementController extends Controller {

	private $slice_size = 500;

	/**
	  * @Route("/{id}", name="bm2_settlement", requirements={"id"="\d+"})
	  * @Template("BM2SiteBundle:Settlement:settlement.html.twig")
	  */
	public function indexAction(Settlement $id) {
		$em = $this->getDoctrine()->getManager();
		$settlement = $id; // we use $id because it's hardcoded in the linkhelper

		// check if we should be able to see details
		$character = $this->get('appstate')->getCharacter(false);
		if ($character) {
			$heralds = $character->getAvailableEntourageOfType('Herald')->count();
		} else {
			$heralds = 0;
		}
		$details = $this->get('interactions')->characterViewDetails($character, $settlement);
		if (isset($details['startme'])) {
			// still in start mode
			$form_map = $this->createFormBuilder()->add('settlement_id', 'hidden', array(
				'constraints' => array() // TODO: constrained to available settlements
			))->getForm();
			$details['startme'] = $form_map->createView();			
		}

		// FIXME: shouldn't this use geodata?
		$query = $em->createQuery('SELECT s.id, s.name, ST_Distance(y.center, x.center) AS distance, ST_Azimuth(y.center, x.center) AS direction 
			FROM BM2SiteBundle:Settlement s JOIN s.geo_data x, BM2SiteBundle:GeoData y WHERE y=:here AND ST_Touches(x.poly, y.poly)=true');
		$query->setParameter('here', $settlement);
		$neighbours = $query->getArrayResult();

		if ($details['spy'] || $settlement->getOwner() == $character) {
			$militia = $settlement->getActiveMilitiaByType();
		} else {
			$militia = null;
		}

		// FIXME: better: some trend analysis
		$query = $em->createQuery('SELECT s.population as pop FROM BM2SiteBundle:StatisticSettlement s WHERE s.settlement = :here ORDER BY s.cycle DESC');
		$query->setParameter('here', $settlement);
		$query->setMaxResults(3);
		$data = $query->getArrayResult();
		if (isset($data[2])) {
			$popchange = $data[0]['pop'] - $data[2]['pop'];
		} else {
			$popchange = 0;
		}

		$corruption = $this->get('economy')->calculateCorruption($settlement);
		if ($character != $settlement->getOwner()) {
			// rounding this to full percents to fuzz it a bit, to prevent people from understanding which characters belong to the same player by corruption values
			$corruption = round($corruption*100)/100;
		}

		$economy = array();
		$all = $em->getRepository('BM2SiteBundle:ResourceType')->findAll();
		foreach ($all as $resource) {
			$local = $settlement->findResource($resource);
			if ($local) {
				$base = $local->getAmount();
				$storage = $local->getStorage();
			} else {
				$base = 0;
				$storage = 0;
			}

			if ($details['spot'] && ($details['prospector'] || $settlement->getOwner() == $character)) {
				// TODO: we should fuzz corruption a bit to prevent people spotting same users by comparing corruption
				$full_demand = $this->get('economy')->ResourceDemand($settlement, $resource, true);
				$demand = 0;
				foreach ($full_demand as $key=>$value) {
					$demand += $value;
				}
				$production = $this->get('economy')->ResourceProduction($settlement, $resource);
				$base_production = $this->get('economy')->ResourceProduction($settlement, $resource, true);
			} else {
				$full_demand = array('base'=>0, 'corruption'=>0, 'operation'=>0, 'construction'=>0);
				$demand = $production = $base_production = 0;
			}

			$economy[] = array(
				'name' => $resource->getName(),
				'base' => $base,
				'storage' => $storage,
				'base_production' => $base_production,
				'total_production' => $production,
				'tradebalance' => $this->get('economy')->TradeBalance($settlement, $resource),
				'base_demand' => $full_demand['base'],
				'corruption' => $full_demand['corruption'],
				'total_demand' => $demand,
				'building_prod' => $production - $base_production,
				'building_demand' => $full_demand['operation'],
				'building_construction' => $full_demand['construction']
			);
			if ($resource->getName()=="food") {
				if ($demand>0) {
					$FoodSupply = $this->get('economy')->ResourceAvailable($settlement, $resource) / $demand;
				} else {
					$FoodSupply = 1.0;
				}
			} 
		}

		return array(
			'settlement' => $settlement,
			'familiarity' => $character?$this->get('geography')->findRegionFamiliarityLevel($character, $settlement->getGeoData()):false,
			'details' => $details,
			'popchange' => $popchange,
			'foodsupply' => $FoodSupply,
			'economy' => $economy,
			'corruption' => $corruption,
			'area' => $this->get('geography')->calculateArea($settlement->getGeoData()),
			'density' => $this->get('geography')->calculatePopulationDensity($settlement),
			'regionpoly'=> $this->get('geography')->findRegionPolygon($settlement),
			'neighbours' => $neighbours,
			'militia' => $militia,
			'recruits' => $settlement->getRecruits()->count(),
			'security' => round(($this->get('economy')->EconomicSecurity($settlement)-1.0)*16),
			'heralds' => $heralds
		);
	}


	/**
	  * @Route("/{id}/soldiers/{start}", requirements={"id"="\d+", "start"="\d+"}, defaults={"start"=0})
	  * @Template
	  */
	public function soldiersAction($id, $start, Request $request) {
		$em = $this->getDoctrine()->getManager();

		list($character, $settlement) = $this->get('dispatcher')->gateway('personalMilitiaTest', true);
		if (!$settlement) {
			throw $this->createNotFoundException('error.notfound.settlement');
		}
		if ($settlement->getId() != $id) {
			throw new AccessDeniedHttpException('error.noaccess.settlement');
		}
		if ($character->getPrisonerOf()) {
			$others = array($character->getPrisonerOf());
		} else {
			$others = $this->get('dispatcher')->getActionableCharacters(true);
		}

		$resupply = $this->get('military')->findAvailableEquipment($settlement, false);
		$training = $this->get('military')->findAvailableEquipment($settlement, true);

		// this duplicates functionality from the Soldier Entity (e.g. determine who is militia and who recruit),
		// but I think we need it for memory reasons, since loading all and then slicing is a ton less efficient
		$query = $em->createQuery("SELECT count(s) FROM BM2SiteBundle:Soldier s WHERE s.base = :here AND s.training_required <= 0")			
			->setParameter('here', $settlement);
		$total_soldiers = $query->getSingleScalarResult();
		$query = $em->createQuery("SELECT s FROM BM2SiteBundle:Soldier s WHERE s.base = :here AND s.training_required <= 0 ORDER BY s.name ASC")
			->setParameter('here', $settlement)
			->setFirstResult($start)
			->setMaxResults($this->slice_size);
		$militia_slice = array();
		foreach ($query->getResult() as $soldier) {
			$militia_slice[$soldier->getId()] = $soldier;
		}


		$form = $this->createForm(new SoldiersManageType($em, $militia_slice, $resupply, $training, $others, $settlement));
		$form->handleRequest($request);
		if ($form->isValid()) {
			$data = $form->getData();

			$this->get('military')->manage($militia_slice, $data, $settlement, $character);
			$em->flush();
			$this->get('appstate')->setSessionData($character); // update, because maybe we changed our soldiers count
			return $this->redirect($request->getUri());
		}

		return array(
			'settlement' => $settlement,
			'soldiers' => $militia_slice,
			'recruits' => $settlement->getRecruits(),
			'resupply' => $resupply,
			'training' => $training,
			'total_soldiers' => $total_soldiers,
			'start' => $start,
			'slice' => $this->slice_size,
			'form' => $form->createView(),
			'limit' => $this->get('appstate')->getGlobal('pagerlimit', 100)
		);
	}


	/**
	* @Route("/canceltraining")
	*/
	public function cancelTrainingAction(Request $request) {
		if ($request->isMethod('POST') && $request->request->has("settlement") && $request->request->has("recruit")) {
			list($character, $settlement) = $this->get('dispatcher')->gateway('personalMilitiaTest', true);

			$em = $this->getDoctrine()->getManager();
			$recruit = $em->getRepository('BM2SiteBundle:Soldier')->find($request->request->get('recruit'));
			if (!$recruit || !$recruit->isRecruit() || $recruit->getBase()!=$settlement) {
				throw $this->createNotFoundException('error.notfound.recruit');
			}

			// return his equipment to the stockpile:
			$this->get('military')->returnItem($settlement, $recruit->getWeapon());
			$this->get('military')->returnItem($settlement, $recruit->getArmour());
			$this->get('military')->returnItem($settlement, $recruit->getEquipment());

			if ($recruit->getOldWeapon() || $recruit->getOldArmour() || $recruit->getOldEquipment()) {
				// old soldier - return to militia with his old stuff
				$recruit->setWeapon($recruit->getOldWeapon());
				$recruit->setArmour($recruit->getOldArmour());
				$recruit->setEquipment($recruit->getOldEquipment());
				$recruit->setTraining(0)->setTrainingRequired(0);
				$this->get('history')->addToSoldierLog($recruit, 'traincancel');
			} else {
				// fresh recruit - return to workforce
				$settlement->setPopulation($settlement->getPopulation()+1);
				$em->remove($recruit);
			}
			$em->flush();
		}
		return new Response();
	}


	/**
	  * @Route("/{id}/permissions", requirements={"id"="\d+"})
	  * @Template
	  */
	public function permissionsAction($id, Request $request) {
		$character = $this->get('dispatcher')->gateway();
		$em = $this->getDoctrine()->getManager();
		$settlement = $em->getRepository('BM2SiteBundle:Settlement')->find($id);
		if (!$settlement) {
			throw $this->createNotFoundException('error.notfound.settlement');
		}

		$original_permissions = clone $settlement->getPermissions();

		$form = $this->createForm(new SettlementPermissionsSetType($character, $this->getDoctrine()->getManager()), $settlement);

		// FIXME: right now, nothing happens if we disallow thralls while having some
		//			 something should happen - set them free? most should vanish, but some stay as peasants?
		//			 but do we want large numbers of people to simply disappear? where will they go?

		$form->handleRequest($request);
		if ($form->isValid()) {
			$data = $form->getData();

			foreach ($settlement->getPermissions() as $permission) {
				$permission->setValueRemaining($permission->getValue());
				if (!$permission->getId()) {
					$em->persist($permission);
				}
			}
			foreach ($original_permissions as $orig) {
				if (!$settlement->getPermissions()->contains($orig)) {
					$em->remove($orig);
				}
			}
			$em->flush();
			$this->addFlash('notice', $this->get('translator')->trans('control.permissions.success', array(), 'actions'));
			return $this->redirect($request->getUri());
		}

		return array(
			'settlement' => $settlement,
			'permissions' => $em->getRepository('BM2SiteBundle:Permission')->findByClass('settlement'),
			'form' => $form->createView()
		);
	}

	/**
	  * @Route("/{id}/quests", requirements={"id"="\d+"})
	  * @Template
	  */
	public function questsAction($id, Request $request) {
		$character = $this->get('dispatcher')->gateway();
		$em = $this->getDoctrine()->getManager();
		$settlement = $em->getRepository('BM2SiteBundle:Settlement')->find($id);
		if (!$settlement) {
			throw $this->createNotFoundException('error.notfound.settlement');
		}

		return array('settlement'=>$settlement, 'quests'=>$settlement->getQuests());
	}

}