From 0a4874ce13c82f1d12e05bd05bbe49d363d96410 Mon Sep 17 00:00:00 2001
From: Jacek Kowalski <Jacek@jacekk.info>
Date: Sat, 11 Jun 2022 19:09:50 +0000
Subject: [PATCH] Add KrakTransRem buses (341-349, 596-598)

---
 lib/Database.php |   42 ++++++++++++++++++++++++++++++------------
 1 files changed, 30 insertions(+), 12 deletions(-)

diff --git a/lib/Database.php b/lib/Database.php
index 22cc51c..e83f4ab 100644
--- a/lib/Database.php
+++ b/lib/Database.php
@@ -10,13 +10,23 @@
 		$this->pdo = new PDO('sqlite:'.$file);
 		$this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
 		
-		$this->pdo->query('CREATE TABLE IF NOT EXISTS vehicles (
+		$this->pdo->query('CREATE TABLE IF NOT EXISTS vehicles2 (
 			id INT PRIMARY KEY,
 			num INT UNIQUE,
-			weight INT
+			weight INT,
+			line VARCHAR,
+			date INT
 		)');
+		try {
+			$this->beginTransaction();
+			$this->pdo->query('INSERT INTO vehicles2 SELECT id, num, weight, \'?\', \''.time().'\' FROM vehicles');
+			$this->commit();
+			$this->pdo->query('DROP TABLE vehicles');
+		} catch(PDOException $e) {
+			$this->rollback();
+		}
 		
-		$this->addStatement = $this->pdo->prepare('INSERT OR REPLACE INTO vehicles (id, num, weight) VALUES (:id, :num, :weight)');
+		$this->addStatement = $this->pdo->prepare('INSERT OR REPLACE INTO vehicles2 (id, num, weight, line, date) VALUES (:id, :num, :weight, :line, :date)');
 		
 		$this->_cacheClear();
 	}
@@ -35,7 +45,7 @@
 	
 	protected function _cachePopulate() {
 		if($this->cacheId === NULL) {
-			$st = $this->pdo->prepare('SELECT * FROM vehicles');
+			$st = $this->pdo->prepare('SELECT * FROM vehicles2');
 			$st->execute();
 			$result = $st->fetchAll(PDO::FETCH_ASSOC);
 			
@@ -58,9 +68,14 @@
 		$this->cacheNum = NULL;
 	}
 	
-	public function getAll() {
+	public function getAllById() {
 		$this->_cachePopulate();
 		return $this->cacheId;
+	}
+	
+	public function getAllByNum() {
+		$this->_cachePopulate();
+		return $this->cacheNum;
 	}
 	
 	public function getById($id) {
@@ -74,25 +89,28 @@
 	}
 	
 	public function clear() {
-		$this->pdo->query('DELETE FROM vehicles');
+		$this->pdo->query('DELETE FROM vehicles2');
 		$this->_cacheClear();
 	}
 	
-	public function add($id, $num, $weight) {
+	public function add($id, $num, $weight, $line = NULL, $date = NULL) {
 		$vehicle = [
 			'id' => (string)$id,
 			'num' => (string)$num,
-			'weight' => (string)$weight
+			'weight' => (string)$weight,
+			'line' => (string)($line ?? ''),
+			'date' => (string)($date ?? time()),
 		];
 		$this->addStatement->execute($vehicle);
 		$this->_cacheAdd($vehicle);
 	}
 	
-	public function addMapping($mapping) {
+	public function addMapping($vehiclesMapping, Mapper $mapper) {
 		$this->beginTransaction();
-		$weight = count($mapping);
-		foreach($mapping as $id => $num) {
-			$this->add($id, $num, $weight);
+		$weight = count($vehiclesMapping);
+		foreach($vehiclesMapping as $id => $num) {
+			$trip = $mapper->getTTSSVehicleTrip($id);
+			$this->add($id, $num, $weight, ($trip['line'] ?? '?') . ' ' . ($trip['direction'] ?? '?'));
 		}
 		$this->commit();
 	}

--
Gitblit v1.9.1