use more widely compatible SQL; CONCAT is not supported across all RDBMS
[bdrem.git] / src / bdrem / Source / Sql.php
1 <?php
2 namespace bdrem;
3
4 /**
5  * Fetch data from an SQL database
6  */
7 class Source_Sql
8 {
9     protected $dsn;
10     protected $user;
11     protected $password;
12     protected $table;
13     protected $fields ;
14
15     public function __construct($config)
16     {
17         $this->dsn      = $config['dsn'];
18         $this->user     = $config['user'];
19         $this->password = $config['password'];
20         $this->table    = $config['table'];
21         $this->fields   = $config['fields'];
22     }
23
24     /**
25      * @param string $strDate Date the events shall be found for, YYYY-MM-DD
26      */
27     public function getEvents($strDate, $nDaysPrevious, $nDaysNext)
28     {
29         $dbh = new \PDO($this->dsn, $this->user, $this->password);
30         $arDays = $this->getDates($strDate, $nDaysPrevious, $nDaysNext);
31         $arEvents = array();
32
33         foreach ($this->fields['date'] as $field => $typeName) {
34             $sqlMonth = 'EXTRACT(MONTH FROM ' . $field . ')';
35             $sqlDay = 'EXTRACT(DAY FROM ' . $field . ')';
36
37             $parts = array();
38             foreach ($arDays as $month => $days) {
39                 $parts[] = '('
40                     . $sqlMonth . ' = ' . $dbh->quote($month, \PDO::PARAM_INT)
41                     . ' AND ' . $sqlDay . ' >= '
42                     . $dbh->quote(min($days), \PDO::PARAM_INT)
43                     . ' AND ' . $sqlDay . ' <= '
44                     . $dbh->quote(max($days), \PDO::PARAM_INT)
45                     . ')';
46             }
47             $sql = 'SELECT ' . $field . ' AS e_date'
48                 . ', ' . $this->fields['name'] . ' AS e_name'
49                 . ' FROM ' . $this->table
50                 . ' WHERE '
51                 . implode(' OR ', $parts);
52
53             $res = $dbh->query($sql);
54             if ($res === false) {
55                 $errorInfo = $dbh->errorInfo();
56                 throw new \Exception(
57                     'SQL error #' . $errorInfo[0]
58                     . ': ' . $errorInfo[1]
59                     . ': ' . $errorInfo[2],
60                     (int) $errorInfo[1]
61                 );
62             }
63             while ($row = $res->fetchObject()) {
64                 $event = new Event(
65                     $row->e_name, $typeName, 
66                     str_replace('0000', '????', $row->e_date)
67                 );
68                 if ($event->isWithin($strDate, $nDaysPrevious, $nDaysNext)) {
69                     $arEvents[] = $event;
70                 }
71             }
72         }
73         return $arEvents;
74     }
75     
76     /**
77      * @return array Key is the month, value an array of days
78      */
79     protected function getDates($strDate, $nDaysPrevious, $nDaysNext)
80     {
81         $ts = strtotime($strDate) - 86400 * $nDaysPrevious;
82         $numDays = $nDaysPrevious + $nDaysNext;
83
84         $arDays = array();
85         do {
86             $arDays[(int) date('n', $ts)][] = (int) date('j', $ts);
87             $ts += 86400;
88         } while (--$numDays >= 0);
89         return $arDays;
90     }
91 }
92 ?>