[SPIP] ~maj 3.0.11
[ptitvelo/web/www.git] / www / ecrire / req / pg.php
1 <?php
2
3 /* *************************************************************************\
4 * SPIP, Systeme de publication pour l'internet *
5 * *
6 * Copyright (c) 2001-2012 *
7 * Arnaud Martin, Antoine Pitrou, Philippe Riviere, Emmanuel Saint-James *
8 * *
9 * Ce programme est un logiciel libre distribue sous licence GNU/GPL. *
10 * Pour plus de details voir le fichier COPYING.txt ou l'aide en ligne. *
11 \***************************************************************************/
12
13 /**
14 * Ce fichier contient les fonctions gerant
15 * les instructions SQL pour PostgreSQL
16 *
17 * @package SPIP\SQL\PostgreSQL
18 */
19
20 if (!defined('_ECRIRE_INC_VERSION')) return;
21
22 define('_DEFAULT_DB', 'spip');
23
24 // Se connecte et retourne le nom de la fonction a connexion persistante
25 // A la premiere connexion de l'installation (BD pas precisee)
26 // si on ne peut se connecter sans la preciser
27 // on reessaye avec le login comme nom de BD
28 // et si ca marche toujours pas, avec "spip" (constante ci-dessus)
29 // si ca ne marche toujours pas, echec.
30
31 // http://doc.spip.org/@req_pg_dist
32 function req_pg_dist($addr, $port, $login, $pass, $db='', $prefixe='') {
33 static $last_connect = array();
34 if (!charger_php_extension('pgsql')) return false;
35
36 // si provient de selectdb
37 if (empty($addr) && empty($port) && empty($login) && empty($pass)){
38 foreach (array('addr','port','login','pass','prefixe') as $a){
39 $$a = $last_connect[$a];
40 }
41 }
42 @list($host, $p) = explode(';', $addr);
43 if ($p >0) $port = " port=$p" ; else $port = '';
44 $erreurs = array();
45 if ($db) {
46 @$link = pg_connect("host=$host$port dbname=$db user=$login password=$pass", PGSQL_CONNECT_FORCE_NEW);
47 } elseif (!@$link = pg_connect("host=$host$port user=$login password=$pass", PGSQL_CONNECT_FORCE_NEW)) {
48 $erreurs[] = pg_last_error();
49 if (@$link = pg_connect("host=$host$port dbname=$login user=$login password=$pass", PGSQL_CONNECT_FORCE_NEW)) {
50 $db = $login;
51 } else {
52 $erreurs[] = pg_last_error();
53 $db = _DEFAULT_DB;
54 $link = pg_connect("host=$host$port dbname=$db user=$login password=$pass", PGSQL_CONNECT_FORCE_NEW);
55 }
56 }
57 if (!$link) {
58 $erreurs[] = pg_last_error();
59 foreach($erreurs as $e)
60 spip_log('Echec pg_connect. Erreur : ' . $e,'pg.'._LOG_HS);
61 return false;
62 }
63
64 if ($link)
65 $last_connect = array (
66 'addr' => $addr,
67 'port' => $port,
68 'login' => $login,
69 'pass' => $pass,
70 'db' => $db,
71 'prefixe' => $prefixe,
72 );
73
74 spip_log("Connexion vers $host, base $db, prefixe $prefixe " . ($link ? 'operationnelle' : 'impossible'),'pg.'._LOG_DEBUG);
75
76 return !$link ? false : array(
77 'db' => $db,
78 'prefixe' => $prefixe ? $prefixe : $db,
79 'link' => $link,
80 );
81 }
82
83 $GLOBALS['spip_pg_functions_1'] = array(
84 'alter' => 'spip_pg_alter',
85 'count' => 'spip_pg_count',
86 'countsel' => 'spip_pg_countsel',
87 'create' => 'spip_pg_create',
88 'create_base' => 'spip_pg_create_base',
89 'create_view' => 'spip_pg_create_view',
90 'date_proche' => 'spip_pg_date_proche',
91 'delete' => 'spip_pg_delete',
92 'drop_table' => 'spip_pg_drop_table',
93 'drop_view' => 'spip_pg_drop_view',
94 'errno' => 'spip_pg_errno',
95 'error' => 'spip_pg_error',
96 'explain' => 'spip_pg_explain',
97 'fetch' => 'spip_pg_fetch',
98 'seek' => 'spip_pg_seek',
99 'free' => 'spip_pg_free',
100 'hex' => 'spip_pg_hex',
101 'in' => 'spip_pg_in',
102 'insert' => 'spip_pg_insert',
103 'insertq' => 'spip_pg_insertq',
104 'insertq_multi' => 'spip_pg_insertq_multi',
105 'listdbs' => 'spip_pg_listdbs',
106 'multi' => 'spip_pg_multi',
107 'optimize' => 'spip_pg_optimize',
108 'query' => 'spip_pg_query',
109 'quote' => 'spip_pg_quote',
110 'replace' => 'spip_pg_replace',
111 'replace_multi' => 'spip_pg_replace_multi',
112 'select' => 'spip_pg_select',
113 'selectdb' => 'spip_pg_selectdb',
114 'set_connect_charset' => 'spip_pg_set_connect_charset',
115 'showbase' => 'spip_pg_showbase',
116 'showtable' => 'spip_pg_showtable',
117 'update' => 'spip_pg_update',
118 'updateq' => 'spip_pg_updateq',
119 );
120
121 // Par ou ca passe une fois les traductions faites
122 // http://doc.spip.org/@spip_pg_trace_query
123 function spip_pg_trace_query($query, $serveur='')
124 {
125 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
126 $prefixe = $connexion['prefixe'];
127 $link = $connexion['link'];
128 $db = $connexion['db'];
129
130 if (isset($_GET['var_profile'])) {
131 include_spip('public/tracer');
132 $t = trace_query_start();
133 } else $t = 0 ;
134
135 $connexion['last'] = $query;
136 $r = spip_pg_query_simple($link, $query);
137
138 if ($e = spip_pg_errno($serveur)) // Log de l'erreur eventuelle
139 $e .= spip_pg_error($query, $serveur); // et du fautif
140 return $t ? trace_query_end($query, $t, $r, $e, $serveur) : $r;
141 }
142
143 // Fonction de requete generale quand on est sur que c'est SQL standard.
144 // Elle change juste le noms des tables ($table_prefix) dans le FROM etc
145
146 // http://doc.spip.org/@spip_pg_query
147 function spip_pg_query($query, $serveur='',$requeter=true)
148 {
149 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
150 $prefixe = $connexion['prefixe'];
151 $link = $connexion['link'];
152 $db = $connexion['db'];
153
154 if (preg_match('/\s(SET|VALUES|WHERE|DATABASE)\s/i', $query, $regs)) {
155 $suite = strstr($query, $regs[0]);
156 $query = substr($query, 0, -strlen($suite));
157 } else $suite ='';
158 $query = preg_replace('/([,\s])spip_/', '\1'.$prefixe.'_', $query) . $suite;
159
160 // renvoyer la requete inerte si demandee
161 if (!$requeter) return $query;
162
163 return spip_pg_trace_query($query, $serveur);
164 }
165
166 function spip_pg_query_simple($link, $query){
167 #spip_log(var_export($query,true), 'pg.'._LOG_DEBUG);
168 return pg_query($link, $query);
169 }
170
171 /*
172 * Retrouver les champs 'timestamp'
173 * pour les ajouter aux 'insert' ou 'replace'
174 * afin de simuler le fonctionnement de mysql
175 *
176 * stocke le resultat pour ne pas faire
177 * de requetes showtable intempestives
178 */
179 function spip_pg_ajouter_champs_timestamp($table, $couples, $desc='', $serveur=''){
180 static $tables = array();
181
182 if (!isset($tables[$table])){
183
184 if (!$desc){
185 $trouver_table = charger_fonction('trouver_table', 'base');
186 $desc = $trouver_table($table, $serveur);
187 // si pas de description, on ne fait rien, ou on die() ?
188 if (!$desc) return $couples;
189 }
190
191 // recherche des champs avec simplement 'TIMESTAMP'
192 // cependant, il faudra peut etre etendre
193 // avec la gestion de DEFAULT et ON UPDATE
194 // mais ceux-ci ne sont pas utilises dans le core
195 $tables[$table] = array();
196 foreach ($desc['field'] as $k=>$v){
197 $v = strtolower(ltrim($v));
198 // ne pas ajouter de timestamp now() si un default est specifie
199 if (strpos($v, 'timestamp')===0 AND strpos($v, 'default')===false)
200 $tables[$table][] = $k;
201 }
202 }
203
204 // ajout des champs type 'timestamp' absents
205 foreach ($tables[$table] as $maj){
206 if (!array_key_exists($maj, $couples))
207 $couples[$maj] = "NOW()";
208 }
209 return $couples;
210 }
211
212
213 // Alter en PG ne traite pas les index
214 // http://doc.spip.org/@spip_pg_alter
215 function spip_pg_alter($query, $serveur='',$requeter=true) {
216 // il faudrait une regexp pour eviter de spliter ADD PRIMARY KEY (colA, colB)
217 // tout en cassant en deux alter distincts "ADD PRIMARY KEY (colA, colB), ADD INDEX (chose)"...
218 // ou revoir l'api de sql_alter en creant un
219 // sql_alter_table($table,array($actions));
220 if (!preg_match("/\s*((\s*IGNORE)?\s*TABLE\s*([^\s]*))\s*(.*)?/is", $query, $regs)){
221 spip_log("$query mal comprise", 'pg.'._LOG_ERREUR);
222 return false;
223 }
224 $debut = $regs[1];
225 $table = $regs[3];
226 $suite = $regs[4];
227 $todo = explode(',', $suite);
228 // on remet les morceaux dechires ensembles... que c'est laid !
229 $todo2 = array(); $i=0;
230 $ouverte = false;
231 while ($do = array_shift($todo)) {
232 $todo2[$i] = isset($todo2[$i]) ? $todo2[$i] . "," . $do : $do;
233 $o=(false!==strpos($do,"("));
234 $f=(false!==strpos($do,")"));
235 if ($o AND !$f) $ouverte=true;
236 elseif ($f) $ouverte=false;
237 if (!$ouverte) $i++;
238 }
239 $todo=$todo2;
240 $query = $debut.' '.array_shift($todo);
241
242 if (!preg_match('/^\s*(IGNORE\s*)?TABLE\s+(\w+)\s+(ADD|DROP|CHANGE|MODIFY|RENAME)\s*(.*)$/is', $query, $r)) {
243 spip_log("$query incompris", 'pg.'._LOG_ERREUR);
244 } else {
245 if ($r[1]) spip_log("j'ignore IGNORE dans $query", 'pg.'._LOG_AVERTISSEMENT);
246 $f = 'spip_pg_alter_' . strtolower($r[3]);
247 if (function_exists($f))
248 $f($r[2], $r[4], $serveur, $requeter);
249 else spip_log("$query non prevu", 'pg.'._LOG_ERREUR);
250 }
251 // Alter a plusieurs args. Faudrait optimiser.
252 if ($todo)
253 spip_pg_alter("TABLE $table " . join(',',$todo));
254
255 }
256
257 // http://doc.spip.org/@spip_pg_alter_change
258 function spip_pg_alter_change($table, $arg, $serveur='',$requeter=true)
259 {
260 if (!preg_match('/^`?(\w+)`?\s+`?(\w+)`?\s+(.*?)\s*(DEFAULT .*?)?(NOT\s+NULL)?\s*(DEFAULT .*?)?$/i',$arg, $r)) {
261 spip_log("alter change: $arg incompris", 'pg.'._LOG_ERREUR);
262 } else {
263 list(,$old, $new, $type, $default, $null, $def2) = $r;
264 $actions = array("ALTER $old TYPE " . mysql2pg_type($type));
265 if ($null)
266 $actions[]= "ALTER $old SET NOT NULL";
267 else
268 $actions[]= "ALTER $old DROP NOT NULL";
269
270 if ($d = ($default ? $default : $def2))
271 $actions[]= "ALTER $old SET $d";
272 else
273 $actions[]= "ALTER $old DROP DEFAULT";
274
275 spip_pg_query("ALTER TABLE $table " . join(', ', $actions));
276
277 if ($old != $new)
278 spip_pg_query("ALTER TABLE $table RENAME $old TO $new", $serveur);
279 }
280 }
281
282 // http://doc.spip.org/@spip_pg_alter_add
283 function spip_pg_alter_add($table, $arg, $serveur='',$requeter=true) {
284 if (!preg_match('/^(COLUMN|INDEX|KEY|PRIMARY\s+KEY|)\s*(.*)$/', $arg, $r)) {
285 spip_log("alter add $arg incompris", 'pg.'._LOG_ERREUR);
286 return NULL;
287 }
288 if (!$r[1] OR $r[1]=='COLUMN') {
289 preg_match('/`?(\w+)`?(.*)/',$r[2], $m);
290 if (preg_match('/^(.*)(BEFORE|AFTER|FIRST)(.*)$/is', $m[2], $n)) {
291 $m[2]=$n[1];
292 }
293 return spip_pg_query("ALTER TABLE $table ADD " . $m[1] . ' ' . mysql2pg_type($m[2]), $serveur, $requeter);
294 } elseif ($r[1][0] == 'P') {
295 // la primary peut etre sur plusieurs champs
296 $r[2] = trim(str_replace('`','',$r[2]));
297 $m = ($r[2][0]=='(') ? substr($r[2],1,-1) : $r[2];
298 return spip_pg_query("ALTER TABLE $table ADD CONSTRAINT $table" .'_pkey PRIMARY KEY (' . $m . ')', $serveur, $requeter);
299 } else {
300 preg_match('/([^\s,]*)\s*(.*)?/',$r[2], $m);
301 // peut etre "(colonne)" ou "nom_index (colonnes)"
302 // bug potentiel si qqn met "(colonne, colonne)"
303 //
304 // nom_index (colonnes)
305 if ($m[2]) {
306 $colonnes = substr($m[2],1,-1);
307 $nom_index = $m[1];
308 }
309 else {
310 // (colonne)
311 if ($m[1][0] == "(") {
312 $colonnes = substr($m[1],1,-1);
313 if (false!==strpos(",",$colonnes)) {
314 spip_log(_LOG_GRAVITE_ERREUR,"PG : Erreur, impossible de creer un index sur plusieurs colonnes"
315 ." sans qu'il ait de nom ($table, ($colonnes))", 'pg');
316 } else {
317 $nom_index = $colonnes;
318 }
319 }
320 // nom_index
321 else {
322 $nom_index = $colonnes = $m[1];
323 }
324 }
325 return spip_pg_create_index($nom_index, $table, $colonnes, $serveur, $requeter);
326 }
327 }
328
329 // http://doc.spip.org/@spip_pg_alter_drop
330 function spip_pg_alter_drop($table, $arg, $serveur='',$requeter=true) {
331 if (!preg_match('/^(COLUMN|INDEX|KEY|PRIMARY\s+KEY|)\s*`?(\w*)`?/', $arg, $r))
332 spip_log("alter drop: $arg incompris", 'pg.'._LOG_ERREUR);
333 else {
334 if (!$r[1] OR $r[1]=='COLUMN')
335 return spip_pg_query("ALTER TABLE $table DROP " . $r[2], $serveur);
336 elseif ($r[1][0] == 'P')
337 return spip_pg_query("ALTER TABLE $table DROP CONSTRAINT $table" . '_pkey', $serveur);
338 else {
339 return spip_pg_query("DROP INDEX " . $table . '_' . $r[2], $serveur);
340 }
341 }
342 }
343
344 function spip_pg_alter_modify($table, $arg, $serveur='',$requeter=true) {
345 if (!preg_match('/^`?(\w+)`?\s+(.*)$/',$arg, $r)) {
346 spip_log("alter modify: $arg incompris", 'pg.'._LOG_ERREUR);
347 } else {
348 return spip_pg_alter_change($table, $r[1].' '.$arg, $serveur='',$requeter=true);
349 }
350 }
351
352 // attention (en pg) :
353 // - alter table A rename to X = changer le nom de la table
354 // - alter table A rename X to Y = changer le nom de la colonne X en Y
355 // pour l'instant, traiter simplement RENAME TO X
356 function spip_pg_alter_rename($table, $arg, $serveur='',$requeter=true) {
357 $rename="";
358 // si TO, mais pas au debut
359 if (!stripos($arg,'TO ')){
360 $rename=$arg;
361 }
362 elseif (preg_match('/^(TO)\s*`?(\w*)`?/', $arg, $r)) {
363 $rename=$r[2];
364 } else {
365 spip_log("alter rename: $arg incompris", 'pg.'._LOG_ERREUR);
366 }
367 return $rename?spip_pg_query("ALTER TABLE $table RENAME TO $rename"):false;
368 }
369
370
371 /**
372 * Fonction de creation d'un INDEX
373 *
374 * @param string $nom : nom de l'index
375 * @param string $table : table sql de l'index
376 * @param string/array $champs : liste de champs sur lesquels s'applique l'index
377 * @param string $serveur : nom de la connexion sql utilisee
378 * @param bool $requeter : true pour executer la requete ou false pour retourner le texte de la requete
379 *
380 * @return bool ou requete
381 */
382 function spip_pg_create_index($nom, $table, $champs, $serveur='', $requeter=true) {
383 if (!($nom OR $table OR $champs)) {
384 spip_log("Champ manquant pour creer un index pg ($nom, $table, (".@join(',',$champs)."))",'pg.'._LOG_ERREUR);
385 return false;
386 }
387
388 $nom = str_replace("`","",$nom);
389 $champs = str_replace("`","",$champs);
390
391 // PG ne differentie pas noms des index en fonction des tables
392 // il faut donc creer des noms uniques d'index pour une base pg
393 $nom = $table.'_'.$nom;
394 // enlever d'eventuelles parentheses deja presentes sur champs
395 if (!is_array($champs)){
396 if ($champs[0]=="(") $champs = substr($champs,1,-1);
397 $champs = array($champs);
398 }
399 $query = "CREATE INDEX $nom ON $table (" . join(',',$champs) . ")";
400 if (!$requeter) return $query;
401 $res = spip_pg_query($query, $serveur, $requeter);
402 return $res;
403 }
404
405
406 // http://doc.spip.org/@spip_pg_explain
407 function spip_pg_explain($query, $serveur='',$requeter=true){
408 if (strpos(ltrim($query), 'SELECT') !== 0) return array();
409 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
410 $prefixe = $connexion['prefixe'];
411 $link = $connexion['link'];
412 if (preg_match('/\s(SET|VALUES|WHERE)\s/i', $query, $regs)) {
413 $suite = strstr($query, $regs[0]);
414 $query = substr($query, 0, -strlen($suite));
415 } else $suite ='';
416 $query = 'EXPLAIN ' . preg_replace('/([,\s])spip_/', '\1'.$prefixe.'_', $query) . $suite;
417
418 if (!$requeter) return $query;
419 $r = spip_pg_query_simple($link,$query);
420 return spip_pg_fetch($r, NULL, $serveur);
421 }
422
423
424 /**
425 * Selectionne une base de donnees
426 *
427 * @param string $nom
428 * Nom de la base a utiliser
429 * @param string $serveur
430 * Nom du connecteur
431 * @param bool $requeter
432 * Inutilise
433 *
434 * @return bool|string
435 * Nom de la base en cas de success.
436 * False en cas d'erreur.
437 **/
438 function spip_pg_selectdb($db, $serveur='',$requeter=true) {
439 // se connecter a la base indiquee
440 // avec les identifiants connus
441 $index = $serveur ? strtolower($serveur) : 0;
442
443 if ($link = spip_connect_db('', '', '', '', $db, 'pg', '', '')){
444 if (($db==$link['db']) && $GLOBALS['connexions'][$index] = $link)
445 return $db;
446 } else
447 return false;
448 }
449
450 // Qu'une seule base pour le moment
451
452 // http://doc.spip.org/@spip_pg_listdbs
453 function spip_pg_listdbs($serveur) {
454 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
455 $link = $connexion['link'];
456 $dbs = array();
457 $res = spip_pg_query_simple($link, "select * From pg_database");
458 while ($row = pg_fetch_array($res, NULL, PGSQL_NUM))
459 $dbs[] = reset($row);
460
461 return $dbs;
462 }
463
464 // http://doc.spip.org/@spip_pg_select
465 function spip_pg_select($select, $from, $where='',
466 $groupby=array(), $orderby='', $limit='',
467 $having='', $serveur='',$requeter=true){
468
469 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
470 $prefixe = $connexion['prefixe'];
471 $link = $connexion['link'];
472 $db = $connexion['db'];
473
474 $limit = preg_match("/^\s*(([0-9]+),)?\s*([0-9]+)\s*$/", $limit,$limatch);
475 if ($limit) {
476 $offset = $limatch[2];
477 $count = $limatch[3];
478 }
479
480 $select = spip_pg_frommysql($select);
481
482 // si pas de tri explicitement demande, le GROUP BY ne
483 // contient que la clef primaire.
484 // lui ajouter alors le champ de tri par defaut
485 if (preg_match("/FIELD\(([a-z]+\.[a-z]+),/i", $orderby[0], $groupbyplus)) {
486 $groupby[] = $groupbyplus[1];
487 }
488
489 $orderby = spip_pg_orderby($orderby, $select);
490
491 if ($having) {
492 if (is_array($having))
493 $having = join("\n\tAND ", array_map('calculer_pg_where', $having));
494 }
495 $from = spip_pg_from($from, $prefixe);
496 $query = "SELECT ". $select
497 . (!$from ? '' : "\nFROM $from")
498 . (!$where ? '' : ("\nWHERE " . (!is_array($where) ? calculer_pg_where($where) : (join("\n\tAND ", array_map('calculer_pg_where', $where))))))
499 . spip_pg_groupby($groupby, $from, $select)
500 . (!$having ? '' : "\nHAVING $having")
501 . ($orderby ? ("\nORDER BY $orderby") :'')
502 . (!$limit ? '' : (" LIMIT $count" . (!$offset ? '' : " OFFSET $offset")));
503
504 // renvoyer la requete inerte si demandee
505 if ($requeter === false) return $query;
506
507 $r = spip_pg_trace_query($query, $serveur);
508 return $r ? $r : $query;;
509 }
510
511 // Le traitement des prefixes de table dans un Select se limite au FROM
512 // car le reste de la requete utilise les alias (AS) systematiquement
513
514 // http://doc.spip.org/@spip_pg_from
515 function spip_pg_from($from, $prefixe)
516 {
517 if (is_array($from)) $from = spip_pg_select_as($from);
518 return !$prefixe ? $from : preg_replace('/(\b)spip_/','\1'.$prefixe.'_', $from);
519 }
520
521 // http://doc.spip.org/@spip_pg_orderby
522 function spip_pg_orderby($order, $select)
523 {
524 $res = array();
525 $arg = (is_array($order) ? $order : preg_split('/\s*,\s*/',$order));
526
527 foreach($arg as $v) {
528 if (preg_match('/(case\s+.*?else\s+0\s+end)\s*AS\s+' . $v .'/', $select, $m)) {
529
530 $res[] = $m[1];
531 } else $res[]=$v;
532 }
533 return spip_pg_frommysql(join(',',$res));
534 }
535
536 // Conversion a l'arrach' des jointures MySQL en jointures PG
537 // A refaire pour tirer parti des possibilites de PG et de MySQL5
538 // et pour enlever les repetitions (sans incidence de perf, mais ca fait sale)
539
540 // http://doc.spip.org/@spip_pg_groupby
541 function spip_pg_groupby($groupby, $from, $select)
542 {
543 $join = strpos($from, ",");
544 // ismplifier avant de decouper
545 if (is_string($select))
546 // fct SQL sur colonne et constante apostrophee ==> la colonne
547 $select = preg_replace('/\w+\(\s*([^(),\']*),\s*\'[^\']*\'[^)]*\)/','\\1', $select);
548
549 if ($join OR $groupby) $join = is_array($select) ? $select : explode(", ", $select);
550 if ($join) {
551 // enlever les 0 as points, '', ...
552 foreach($join as $k=>$v){
553 $v = str_replace('DISTINCT ','',$v);
554 // fct SQL sur colonne et constante apostrophee ==> la colonne
555 $v = preg_replace('/\w+\(\s*([^(),\']*),\s*\'[^\']*\'[^)]*\)/','\\1', $v);
556 $v = preg_replace('/CAST\(\s*([^(),\' ]*\s+)as\s*\w+\)/','\\1', $v);
557 // resultat d'agregat ne sont pas a mettre dans le groupby
558 $v = preg_replace('/(SUM|COUNT|MAX|MIN|UPPER)\([^)]+\)(\s*AS\s+\w+)\s*,?/i','', $v);
559 // idem sans AS (fetch numerique)
560 $v = preg_replace('/(SUM|COUNT|MAX|MIN|UPPER)\([^)]+\)\s*,?/i','', $v);
561 // des AS simples : on garde le cote droit du AS
562 $v = preg_replace('/^.*\sAS\s+(\w+)\s*$/i','\\1', $v);
563 // ne reste plus que les vrais colonnes, ou des constantes a virer
564 if (preg_match(',^[\'"],',$v) OR is_numeric($v))
565 unset($join[$k]);
566 else
567 $join[$k] = trim($v);
568 }
569 $join = array_diff($join,array(''));
570 $join = implode(',',$join);
571 }
572 if (is_array($groupby)) $groupby = join(',',$groupby);
573 if ($join) $groupby = $groupby ? "$groupby, $join" : $join;
574 if (!$groupby) return '';
575
576 $groupby = spip_pg_frommysql($groupby);
577 $groupby = preg_replace('/\s+AS\s+\w+\s*/i','', $groupby);
578
579 return "\nGROUP BY $groupby";
580 }
581
582 // Conversion des operateurs MySQL en PG
583 // IMPORTANT: "0+X" est vu comme conversion numerique du debut de X
584 // Les expressions de date ne sont pas gerees au-dela de 3 ()
585 // Le 'as' du 'CAST' est en minuscule pour echapper au dernier preg_replace
586 // de spip_pg_groupby.
587 // A ameliorer.
588
589 // http://doc.spip.org/@spip_pg_frommysql
590 function spip_pg_frommysql($arg)
591 {
592 if (is_array($arg)) $arg = join(", ", $arg);
593
594 $res = spip_pg_fromfield($arg);
595
596 $res = preg_replace('/\brand[(][)]/i','random()', $res);
597
598 $res = preg_replace('/\b0\.0[+]([a-zA-Z0-9_.]+)\s*/',
599 'CAST(substring(\1, \'^ *[0-9.]+\') as float)',
600 $res);
601 $res = preg_replace('/\b0[+]([a-zA-Z0-9_.]+)\s*/',
602 'CAST(substring(\1, \'^ *[0-9]+\') as int)',
603 $res);
604 $res = preg_replace('/\bconv[(]([^,]*)[^)]*[)]/i',
605 'CAST(substring(\1, \'^ *[0-9]+\') as int)',
606 $res);
607
608 $res = preg_replace('/UNIX_TIMESTAMP\s*[(]\s*[)]/',
609 ' EXTRACT(epoch FROM NOW())', $res);
610
611 // la fonction md5(integer) n'est pas connu en pg
612 // il faut donc forcer les types en text (cas de md5(id_article))
613 $res = preg_replace('/md5\s*[(]([^)]*)[)]/i',
614 'MD5(CAST(\1 AS text))', $res);
615
616 $res = preg_replace('/UNIX_TIMESTAMP\s*[(]([^)]*)[)]/',
617 ' EXTRACT(epoch FROM \1)', $res);
618
619 $res = preg_replace('/\bDAYOFMONTH\s*[(]([^()]*([(][^()]*[)][^()]*)*[^)]*)[)]/',
620 ' EXTRACT(day FROM \1)',
621 $res);
622
623 $res = preg_replace('/\bMONTH\s*[(]([^()]*([(][^)]*[)][^()]*)*[^)]*)[)]/',
624 ' EXTRACT(month FROM \1)',
625 $res);
626
627 $res = preg_replace('/\bYEAR\s*[(]([^()]*([(][^)]*[)][^()]*)*[^)]*)[)]/',
628 ' EXTRACT(year FROM \1)',
629 $res);
630
631 $res = preg_replace('/TO_DAYS\s*[(]([^()]*([(][^)]*[)][()]*)*)[)]/',
632 ' EXTRACT(day FROM \1 - \'0001-01-01\')',
633 $res);
634
635 $res = preg_replace("/(EXTRACT[(][^ ]* FROM *)\"([^\"]*)\"/", '\1\'\2\'', $res);
636
637 $res = preg_replace('/DATE_FORMAT\s*[(]([^,]*),\s*\'%Y%m%d\'[)]/', 'to_char(\1, \'YYYYMMDD\')', $res);
638
639 $res = preg_replace('/DATE_FORMAT\s*[(]([^,]*),\s*\'%Y%m\'[)]/', 'to_char(\1, \'YYYYMM\')', $res);
640
641 $res = preg_replace('/DATE_SUB\s*[(]([^,]*),/', '(\1 -', $res);
642 $res = preg_replace('/DATE_ADD\s*[(]([^,]*),/', '(\1 +', $res);
643 $res = preg_replace('/INTERVAL\s+(\d+\s+\w+)/', 'INTERVAL \'\1\'', $res);
644 $res = preg_replace('/([+<>-]=?)\s*(\'\d+-\d+-\d+\s+\d+:\d+(:\d+)\')/', '\1 timestamp \2', $res);
645 $res = preg_replace('/(\'\d+-\d+-\d+\s+\d+:\d+:\d+\')\s*([+<>-]=?)/', 'timestamp \1 \2', $res);
646
647 $res = preg_replace('/([+<>-]=?)\s*(\'\d+-\d+-\d+\')/', '\1 timestamp \2', $res);
648 $res = preg_replace('/(\'\d+-\d+-\d+\')\s*([+<>-]=?)/', 'timestamp \1 \2', $res);
649
650 $res = preg_replace('/(timestamp .\d+)-00-/','\1-01-', $res);
651 $res = preg_replace('/(timestamp .\d+-\d+)-00/','\1-01',$res);
652 # correct en theorie mais produit des debordements arithmetiques
653 # $res = preg_replace("/(EXTRACT[(][^ ]* FROM *)(timestamp *'[^']*' *[+-] *timestamp *'[^']*') *[)]/", '\2', $res);
654 $res = preg_replace("/(EXTRACT[(][^ ]* FROM *)('[^']*')/", '\1 timestamp \2', $res);
655 $res = preg_replace("/\sLIKE\s+/", ' ILIKE ', $res);
656 return str_replace('REGEXP', '~', $res);
657 }
658
659 // http://doc.spip.org/@spip_pg_fromfield
660 function spip_pg_fromfield($arg)
661 {
662 while(preg_match('/^(.*?)FIELD\s*\(([^,]*)((,[^,)]*)*)\)/', $arg, $m)) {
663
664 preg_match_all('/,([^,]*)/', $m[3], $r, PREG_PATTERN_ORDER);
665 $res = '';
666 $n=0;
667 $index = $m[2];
668 foreach($r[1] as $v) {
669 $n++;
670 $res .= "\nwhen $index=$v then $n";
671 }
672 $arg = $m[1] . "case $res else 0 end "
673 . substr($arg,strlen($m[0]));
674 }
675 return $arg;
676 }
677
678 // http://doc.spip.org/@calculer_pg_where
679 function calculer_pg_where($v)
680 {
681 if (!is_array($v))
682 return spip_pg_frommysql($v);
683
684 $op = str_replace('REGEXP', '~', array_shift($v));
685 if (!($n=count($v)))
686 return $op;
687 else {
688 $arg = calculer_pg_where(array_shift($v));
689 if ($n==1) {
690 return "$op($arg)";
691 } else {
692 $arg2 = calculer_pg_where(array_shift($v));
693 if ($n==2) {
694 return "($arg $op $arg2)";
695 } else return "($arg $op ($arg2) : $v[0])";
696 }
697 }
698 }
699
700
701 // http://doc.spip.org/@calculer_pg_expression
702 function calculer_pg_expression($expression, $v, $join = 'AND'){
703 if (empty($v))
704 return '';
705
706 $exp = "\n$expression ";
707
708 if (!is_array($v)) $v = array($v);
709
710 if (strtoupper($join) === 'AND')
711 return $exp . join("\n\t$join ", array_map('calculer_pg_where', $v));
712 else
713 return $exp . join($join, $v);
714 }
715
716 // http://doc.spip.org/@spip_pg_select_as
717 function spip_pg_select_as($args)
718 {
719 $argsas = "";
720 foreach($args as $k => $v) {
721 if (substr($k,-1)=='@') {
722 // c'est une jointure qui se refere au from precedent
723 // pas de virgule
724 $argsas .= ' ' . $v ;
725 }
726 else {
727 $as = '';
728 // spip_log("$k : $v", _LOG_DEBUG);
729 if (!is_numeric($k)) {
730 if (preg_match('/\.(.*)$/', $k, $r))
731 $v = $k;
732 elseif ($v != $k) {
733 $p = strpos($v, " ");
734 if ($p)
735 $v = substr($v,0,$p) . " AS $k" . substr($v,$p);
736 else $as = " AS $k";
737 }
738 }
739 // spip_log("subs $k : $v avec $as", _LOG_DEBUG);
740 // if (strpos($v, 'JOIN') === false) $argsas .= ', ';
741 $argsas .= ', '. $v . $as;
742 }
743 }
744 return substr($argsas,2);
745 }
746
747 // http://doc.spip.org/@spip_pg_fetch
748 function spip_pg_fetch($res, $t='', $serveur='',$requeter=true) {
749
750 if ($res) $res = pg_fetch_array($res, NULL, PGSQL_ASSOC);
751 return $res;
752 }
753
754 function spip_pg_seek($r, $row_number, $serveur='',$requeter=true) {
755 if ($r) return pg_result_seek($r,$row_number);
756 }
757
758
759 // http://doc.spip.org/@spip_pg_countsel
760 function spip_pg_countsel($from = array(), $where = array(), $groupby=array(),
761 $having = array(), $serveur='',$requeter=true)
762 {
763 $c = !$groupby ? '*' : ('DISTINCT ' . (is_string($groupby) ? $groupby : join(',', $groupby)));
764 $r = spip_pg_select("COUNT($c)", $from, $where,'', '', '', $having, $serveur, $requeter);
765 if (!$requeter) return $r;
766 if (!is_resource($r)) return 0;
767 list($c) = pg_fetch_array($r, NULL, PGSQL_NUM);
768 return $c;
769 }
770
771 // http://doc.spip.org/@spip_pg_count
772 function spip_pg_count($res, $serveur='',$requeter=true) {
773 return !$res ? 0 : pg_numrows($res);
774 }
775
776 // http://doc.spip.org/@spip_pg_free
777 function spip_pg_free($res, $serveur='',$requeter=true) {
778 // rien a faire en postgres
779 }
780
781 // http://doc.spip.org/@spip_pg_delete
782 function spip_pg_delete($table, $where='', $serveur='',$requeter=true) {
783
784 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
785 $prefixe = $connexion['prefixe'];
786 $link = $connexion['link'];
787 $db = $connexion['db'];
788 if ($prefixe) $table = preg_replace('/^spip/', $prefixe, $table);
789
790 $query = calculer_pg_expression('DELETE FROM', $table, ',')
791 . calculer_pg_expression('WHERE', $where, 'AND');
792
793 // renvoyer la requete inerte si demandee
794 if (!$requeter) return $query;
795
796 $res = spip_pg_trace_query($query, $serveur);
797 if ($res)
798 return pg_affected_rows($res);
799 else
800 return false;
801 }
802
803 // http://doc.spip.org/@spip_pg_insert
804 function spip_pg_insert($table, $champs, $valeurs, $desc=array(), $serveur='',$requeter=true) {
805 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
806 $prefixe = $connexion['prefixe'];
807 $link = $connexion['link'];
808 $db = $connexion['db'];
809
810 if (!$desc) $desc = description_table($table, $serveur);
811 $seq = spip_pg_sequence($table,true);
812 // si pas de cle primaire dans l'insertion, renvoyer curval
813 if (!preg_match(",\b$seq\b,",$champs)){
814 $seq = spip_pg_sequence($table);
815 if ($prefixe)
816 $seq = preg_replace('/^spip/', $prefixe, $seq);
817 $seq = "currval('$seq')";
818 }
819
820
821 if ($prefixe) {
822 $table = preg_replace('/^spip/', $prefixe, $table);
823 }
824 $ret = !$seq ? '' : (" RETURNING $seq");
825 $ins = (strlen($champs)<3)
826 ? " DEFAULT VALUES"
827 : "$champs VALUES $valeurs";
828 $q ="INSERT INTO $table $ins $ret";
829 if (!$requeter) return $q;
830 $connexion['last'] = $q;
831 $r = spip_pg_query_simple($link, $q);
832 # spip_log($q,'pg.'._LOG_DEBUG);
833 if ($r) {
834 if (!$ret) return 0;
835 if ($r2 = pg_fetch_array($r, NULL, PGSQL_NUM))
836 return $r2[0];
837 }
838 return false;
839 }
840
841 // http://doc.spip.org/@spip_pg_insertq
842 function spip_pg_insertq($table, $couples=array(), $desc=array(), $serveur='',$requeter=true) {
843
844 if (!$desc) $desc = description_table($table, $serveur);
845 if (!$desc) die("$table insertion sans description");
846 $fields = $desc['field'];
847
848 foreach ($couples as $champ => $val) {
849 $couples[$champ]= spip_pg_cite($val, $fields[$champ]);
850 }
851
852 // recherche de champs 'timestamp' pour mise a jour auto de ceux-ci
853 $couples = spip_pg_ajouter_champs_timestamp($table, $couples, $desc, $serveur);
854
855 return spip_pg_insert($table, "(".join(',',array_keys($couples)).")", "(".join(',', $couples).")", $desc, $serveur, $requeter);
856 }
857
858
859
860 // http://doc.spip.org/@spip_pg_insertq_multi
861 function spip_pg_insertq_multi($table, $tab_couples=array(), $desc=array(), $serveur='',$requeter=true) {
862
863 if (!$desc) $desc = description_table($table, $serveur);
864 if (!$desc) die("$table insertion sans description");
865 $fields = isset($desc['field'])?$desc['field']:array();
866
867 // recherche de champs 'timestamp' pour mise a jour auto de ceux-ci
868 // une premiere fois pour ajouter maj dans les cles
869 $c = isset($tab_couples[0]) ? $tab_couples[0] : array();
870 $les_cles = spip_pg_ajouter_champs_timestamp($table, $c, $desc, $serveur);
871
872 $cles = "(" . join(',',array_keys($les_cles)). ')';
873 $valeurs = array();
874 foreach ($tab_couples as $couples) {
875 foreach ($couples as $champ => $val){
876 $couples[$champ]= spip_pg_cite($val, $fields[$champ]);
877 }
878 // recherche de champs 'timestamp' pour mise a jour auto de ceux-ci
879 $couples = spip_pg_ajouter_champs_timestamp($table, $couples, $desc, $serveur);
880
881 $valeurs[] = '(' .join(',', $couples) . ')';
882 }
883 $valeurs = implode(', ',$valeurs);
884
885 return spip_pg_insert($table, $cles, $valeurs, $desc, $serveur, $requeter);
886 }
887
888
889 // http://doc.spip.org/@spip_pg_update
890 function spip_pg_update($table, $couples, $where='', $desc='', $serveur='',$requeter=true) {
891
892 if (!$couples) return;
893 $connexion = $GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
894 $prefixe = $connexion['prefixe'];
895 $link = $connexion['link'];
896 $db = $connexion['db'];
897 if ($prefixe) $table = preg_replace('/^spip/', $prefixe, $table);
898
899 // recherche de champs 'timestamp' pour mise a jour auto de ceux-ci
900 $couples = spip_pg_ajouter_champs_timestamp($table, $couples, $desc, $serveur);
901
902 $set = array();
903 foreach ($couples as $champ => $val) {
904 $set[] = $champ . '=' . $val;
905 }
906
907 $query = calculer_pg_expression('UPDATE', $table, ',')
908 . calculer_pg_expression('SET', $set, ',')
909 . calculer_pg_expression('WHERE', $where, 'AND');
910
911 // renvoyer la requete inerte si demandee
912 if (!$requeter) return $query;
913
914 return spip_pg_trace_query($query, $serveur);
915 }
916
917 // idem, mais les valeurs sont des constantes a mettre entre apostrophes
918 // sauf les expressions de date lorsqu'il s'agit de fonctions SQL (NOW etc)
919 // http://doc.spip.org/@spip_pg_updateq
920 function spip_pg_updateq($table, $couples, $where='', $desc=array(), $serveur='',$requeter=true) {
921 if (!$couples) return;
922 if (!$desc) $desc = description_table($table, $serveur);
923 $fields = $desc['field'];
924 foreach ($couples as $k => $val) {
925 $couples[$k] = spip_pg_cite($val, $fields[$k]);
926 }
927
928 return spip_pg_update($table, $couples, $where, $desc, $serveur, $requeter);
929 }
930
931
932 // http://doc.spip.org/@spip_pg_replace
933 function spip_pg_replace($table, $values, $desc, $serveur='',$requeter=true) {
934 if (!$values) {spip_log("replace vide $table",'pg.'._LOG_AVERTISSEMENT); return 0;}
935 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
936 $prefixe = $connexion['prefixe'];
937 $link = $connexion['link'];
938 $db = $connexion['db'];
939
940 if (!$desc) $desc = description_table($table, $serveur);
941 if (!$desc) die("$table insertion sans description");
942 $prim = $desc['key']['PRIMARY KEY'];
943 $ids = preg_split('/,\s*/', $prim);
944 $noprims = $prims = array();
945 foreach($values as $k=>$v) {
946 $values[$k] = $v = spip_pg_cite($v, $desc['field'][$k]);
947
948 if (!in_array($k, $ids))
949 $noprims[$k]= "$k=$v";
950 else $prims[$k]= "$k=$v";
951 }
952
953 // recherche de champs 'timestamp' pour mise a jour auto de ceux-ci
954 $values = spip_pg_ajouter_champs_timestamp($table, $values, $desc, $serveur);
955
956 $where = join(' AND ', $prims);
957 if (!$where) {
958 return spip_pg_insert($table, "(".join(',',array_keys($values)).")", "(".join(',', $values).")", $desc, $serveur);
959 }
960 $couples = join(',', $noprims);
961
962 $seq = spip_pg_sequence($table);
963 if ($prefixe) {
964 $table = preg_replace('/^spip/', $prefixe, $table);
965 $seq = preg_replace('/^spip/', $prefixe, $seq);
966 }
967
968 $connexion['last'] = $q = "UPDATE $table SET $couples WHERE $where";
969 if ($couples) {
970 $couples = spip_pg_query_simple($link, $q);
971 # spip_log($q,'pg.'._LOG_DEBUG);
972 if (!$couples) return false;
973 $couples = pg_affected_rows($couples);
974 }
975 if (!$couples) {
976 $ret = !$seq ? '' :
977 (" RETURNING nextval('$seq') < $prim");
978 $connexion['last'] = $q = "INSERT INTO $table (" . join(',',array_keys($values)) . ') VALUES (' .join(',', $values) . ")$ret";
979 $couples = spip_pg_query_simple($link, $q);
980 if (!$couples) {
981 return false;
982 } elseif ($ret) {
983 $r = pg_fetch_array($couples, NULL, PGSQL_NUM);
984 if ($r[0]) {
985 $connexion['last'] = $q = "SELECT setval('$seq', $prim) from $table";
986 // Le code de SPIP met parfois la sequence a 0 (dans l'import)
987 // MySQL n'en dit rien, on fait pareil pour PG
988 $r = @pg_query($link, $q);
989 }
990 }
991 }
992
993 return $couples;
994 }
995
996
997 // http://doc.spip.org/@spip_pg_replace_multi
998 function spip_pg_replace_multi($table, $tab_couples, $desc=array(), $serveur='',$requeter=true) {
999 // boucler pour traiter chaque requete independemment
1000 foreach ($tab_couples as $couples){
1001 $retour = spip_pg_replace($table, $couples, $desc, $serveur,$requeter);
1002 }
1003 // renvoie le dernier id
1004 return $retour;
1005 }
1006
1007
1008
1009 // Donne la sequence eventuelle associee a une table
1010 // Pas extensible pour le moment,
1011
1012 // http://doc.spip.org/@spip_pg_sequence
1013 function spip_pg_sequence($table,$raw=false)
1014 {
1015 global $tables_principales;
1016 include_spip('base/serial');
1017 if (!isset($tables_principales[$table])) return false;
1018 $desc = $tables_principales[$table];
1019 $prim = @$desc['key']['PRIMARY KEY'];
1020 if (!preg_match('/^\w+$/', $prim)
1021 OR strpos($desc['field'][$prim], 'int') === false)
1022 return '';
1023 else { return $raw?$prim:$table . '_' . $prim . "_seq";}
1024 }
1025
1026 // Explicite les conversions de Mysql d'une valeur $v de type $t
1027 // Dans le cas d'un champ date, pas d'apostrophe, c'est une syntaxe ad hoc
1028
1029 // http://doc.spip.org/@spip_pg_cite
1030 function spip_pg_cite($v, $t){
1031 if(is_null($v)) return 'NULL'; // null php se traduit en NULL SQL
1032
1033 if (sql_test_date($t)) {
1034 if (strpos("0123456789", $v[0]) === false)
1035 return spip_pg_frommysql($v);
1036 else {
1037 if (strncmp($v,'0000',4)==0)
1038 $v = "0001" . substr($v,4);
1039 if (strpos($v, "-00-00") === 4)
1040 $v = substr($v,0,4)."-01-01".substr($v,10);
1041 return "timestamp '$v'";
1042 }
1043 }
1044 elseif (!sql_test_int($t))
1045 return ("'" . pg_escape_string($v) . "'");
1046 elseif (is_numeric($v) OR (strpos($v, 'CAST(') === 0))
1047 return $v;
1048 elseif ($v[0]== '0' AND $v[1]!=='x' AND ctype_xdigit(substr($v,1)))
1049 return substr($v,1);
1050 else {
1051 spip_log("Warning: '$v' n'est pas de type $t", 'pg.'._LOG_AVERTISSEMENT);
1052 return intval($v);
1053 }
1054 }
1055
1056 // http://doc.spip.org/@spip_pg_hex
1057 function spip_pg_hex($v)
1058 {
1059 return "CAST(x'" . $v . "' as bigint)";
1060 }
1061
1062 function spip_pg_quote($v, $type='')
1063 {
1064 if (!is_array($v))
1065 return spip_pg_cite($v,$type);
1066 // si c'est un tableau, le parcourir en propageant le type
1067 foreach($v as $k=>$r)
1068 $v[$k] = spip_pg_quote($r, $type);
1069 return join(",", $v);
1070 }
1071
1072 function spip_pg_date_proche($champ, $interval, $unite)
1073 {
1074 return '('
1075 . $champ
1076 . (($interval <= 0) ? '>' : '<')
1077 . (($interval <= 0) ? 'DATE_SUB' : 'DATE_ADD')
1078 . '('
1079 . sql_quote(date('Y-m-d H:i:s'))
1080 . ', INTERVAL '
1081 . (($interval > 0) ? $interval : (0-$interval))
1082 . ' '
1083 . $unite
1084 . '))';
1085 }
1086
1087 // http://doc.spip.org/@spip_pg_in
1088 function spip_pg_in($val, $valeurs, $not='', $serveur) {
1089 //
1090 // IN (...) souvent limite a 255 elements, d'ou cette fonction assistante
1091 //
1092 // s'il n'y a pas de valeur, eviter de produire un IN vide: PG rale.
1093 if (!$valeurs) return $not ? '0=0' : '0=1';
1094 if (strpos($valeurs, "CAST(x'") !== false)
1095 return "($val=" . join("OR $val=", explode(',',$valeurs)).')';
1096 $n = $i = 0;
1097 $in_sql ="";
1098 while ($n = strpos($valeurs, ',', $n+1)) {
1099 if ((++$i) >= 255) {
1100 $in_sql .= "($val $not IN (" .
1101 substr($valeurs, 0, $n) .
1102 "))\n" .
1103 ($not ? "AND\t" : "OR\t");
1104 $valeurs = substr($valeurs, $n+1);
1105 $i = $n = 0;
1106 }
1107 }
1108 $in_sql .= "($val $not IN ($valeurs))";
1109
1110 return "($in_sql)";
1111 }
1112
1113 // http://doc.spip.org/@spip_pg_error
1114 function spip_pg_error($query='', $serveur, $requeter=true) {
1115 $link = $GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0]['link'];
1116 $s = $link ? pg_last_error($link) : pg_last_error();
1117 if ($s) {
1118 $s = str_replace('ERROR', 'errcode: 1000 ', $s);
1119 spip_log("$s - $query", 'pg.'._LOG_ERREUR);
1120 }
1121 return $s;
1122 }
1123
1124 // http://doc.spip.org/@spip_pg_errno
1125 function spip_pg_errno($serveur='') {
1126 // il faudrait avoir la derniere ressource retournee et utiliser
1127 // http://fr2.php.net/manual/fr/function.pg-result-error.php
1128 return 0;
1129 }
1130
1131 // http://doc.spip.org/@spip_pg_drop_table
1132 function spip_pg_drop_table($table, $exist='', $serveur='',$requeter=true)
1133 {
1134 if ($exist) $exist =" IF EXISTS";
1135 if (spip_pg_query("DROP TABLE$exist $table", $serveur, $requeter))
1136 return true;
1137 else return false;
1138 }
1139
1140 // supprime une vue
1141 // http://doc.spip.org/@spip_pg_drop_view
1142 function spip_pg_drop_view($view, $exist='', $serveur='',$requeter=true) {
1143 if ($exist) $exist =" IF EXISTS";
1144 return spip_pg_query("DROP VIEW$exist $view", $serveur, $requeter);
1145 }
1146
1147 /**
1148 * Retourne une ressource de la liste des tables de la base de données
1149 *
1150 * @param string $match
1151 * Filtre sur tables à récupérer
1152 * @param string $serveur
1153 * Connecteur de la base
1154 * @param bool $requeter
1155 * true pour éxecuter la requête
1156 * false pour retourner le texte de la requête.
1157 * @return ressource
1158 * Ressource à utiliser avec sql_fetch()
1159 **/
1160 function spip_pg_showbase($match, $serveur='',$requeter=true)
1161 {
1162 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
1163 $link = $connexion['link'];
1164 $connexion['last'] = $q = "SELECT tablename FROM pg_tables WHERE tablename ILIKE "._q($match);
1165 return spip_pg_query_simple($link, $q);
1166 }
1167
1168 // http://doc.spip.org/@spip_pg_showtable
1169 function spip_pg_showtable($nom_table, $serveur='',$requeter=true)
1170 {
1171 $connexion = &$GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
1172 $link = $connexion['link'];
1173 $connexion['last'] = $q = "SELECT column_name, column_default, data_type FROM information_schema.columns WHERE table_name ILIKE " . _q($nom_table);
1174
1175 $res = spip_pg_query_simple($link, $q);
1176 if (!$res) return false;
1177
1178 // etrangement, $res peut ne rien contenir, mais arriver ici...
1179 // il faut en tenir compte dans le return
1180 $fields = array();
1181 while($field = pg_fetch_array($res, NULL, PGSQL_NUM)) {
1182 $fields[$field[0]] = $field[2] . (!$field[1] ? '' : (" DEFAULT " . $field[1]));
1183 }
1184 $connexion['last'] = $q = "SELECT indexdef FROM pg_indexes WHERE tablename ILIKE " . _q($nom_table);
1185 $res = spip_pg_query_simple($link, $q);
1186 $keys = array();
1187 while($index = pg_fetch_array($res, NULL, PGSQL_NUM)) {
1188 if (preg_match('/CREATE\s+(UNIQUE\s+)?INDEX\s([^\s]+).*\((.*)\)$/', $index[0],$r)) {
1189 $nom = str_replace($nom_table.'_','',$r[2]);
1190 $keys[($r[1] ? "PRIMARY KEY" : ("KEY " . $nom))] = $r[3];
1191 }
1192 }
1193
1194 return count($fields) ? array('field' => $fields, 'key' => $keys) : false;
1195 }
1196
1197 // Fonction de creation d'une table SQL nommee $nom
1198 // a partir de 2 tableaux PHP :
1199 // champs: champ => type
1200 // cles: type-de-cle => champ(s)
1201 // si $autoinc, c'est une auto-increment (i.e. serial) sur la Primary Key
1202 // Le nom des index est prefixe par celui de la table pour eviter les conflits
1203 // http://doc.spip.org/@spip_pg_create
1204 function spip_pg_create($nom, $champs, $cles, $autoinc=false, $temporary=false, $serveur='',$requeter=true) {
1205
1206 $connexion = $GLOBALS['connexions'][$serveur ? strtolower($serveur) : 0];
1207 $prefixe = $connexion['prefixe'];
1208 $link = $connexion['link'];
1209 $db = $connexion['db'];
1210 if ($prefixe) $nom = preg_replace('/^spip/', $prefixe, $nom);
1211 $query = $prim = $prim_name = $v = $s = $p='';
1212 $keys = array();
1213
1214 // certains plugins declarent les tables (permet leur inclusion dans le dump)
1215 // sans les renseigner (laisse le compilo recuperer la description)
1216 if (!is_array($champs) || !is_array($cles))
1217 return;
1218
1219 foreach($cles as $k => $v) {
1220 if (strpos($k, "KEY ") === 0) {
1221 $n = str_replace('`','',$k);
1222 $v = str_replace('`','"',$v);
1223 $i = $nom . preg_replace("/KEY +/", '_',$n);
1224 if ($k != $n) $i = "\"$i\"";
1225 $keys[] = "CREATE INDEX $i ON $nom ($v);";
1226 }
1227 elseif (strpos($k, "UNIQUE ") === 0) {
1228 $k = preg_replace("/^UNIQUE +/", '',$k);
1229 $prim .= "$s\n\t\tCONSTRAINT " . str_replace('`','"',$k) ." UNIQUE ($v)";
1230 }
1231 else {
1232 $prim .= "$s\n\t\t" . str_replace('`','"',$k) ." ($v)";
1233 }
1234 if ($k == "PRIMARY KEY")
1235 $prim_name = $v;
1236 $s = ",";
1237 }
1238 $s = '';
1239
1240 $character_set = "";
1241 if (@$GLOBALS['meta']['charset_sql_base'])
1242 $character_set .= " CHARACTER SET ".$GLOBALS['meta']['charset_sql_base'];
1243 if (@$GLOBALS['meta']['charset_collation_sql_base'])
1244 $character_set .= " COLLATE ".$GLOBALS['meta']['charset_collation_sql_base'];
1245
1246 foreach($champs as $k => $v) {
1247 $k = str_replace('`','"',$k);
1248 if (preg_match(',([a-z]*\s*(\(\s*[0-9]*\s*\))?(\s*binary)?),i',$v,$defs)){
1249 if (preg_match(',(char|text),i',$defs[1]) AND !preg_match(',binary,i',$defs[1]) ){
1250 $v = $defs[1] . $character_set . ' ' . substr($v,strlen($defs[1]));
1251 }
1252 }
1253
1254 $query .= "$s\n\t\t$k "
1255 . (($autoinc && ($prim_name == $k) && preg_match(',\b(big|small|medium|tiny)?int\b,i', $v))
1256 ? " bigserial"
1257 : mysql2pg_type($v)
1258 );
1259 $s = ",";
1260 }
1261 $temporary = $temporary ? 'TEMPORARY':'';
1262
1263 // En l'absence de "if not exists" en PG, on neutralise les erreurs
1264
1265 $q = "CREATE $temporary TABLE $nom ($query" . ($prim ? ",$prim" : '') . ")".
1266 ($character_set?" DEFAULT $character_set":"")
1267 ."\n";
1268
1269 if (!$requeter) return $q;
1270 $connexion['last'] = $q;
1271 $r = @pg_query($link, $q);
1272
1273 if (!$r)
1274 spip_log("Impossible de creer cette table: $q",'pg.'._LOG_ERREUR);
1275 else {
1276 foreach($keys as $index) {pg_query($link, $index);}
1277 }
1278 return $r;
1279 }
1280
1281
1282 function spip_pg_create_base($nom, $serveur='',$requeter=true) {
1283 return spip_pg_query("CREATE DATABASE $nom", $serveur, $requeter);
1284 }
1285
1286 // Fonction de creation d'une vue SQL nommee $nom
1287 // http://doc.spip.org/@spip_pg_create_view
1288 function spip_pg_create_view($nom, $query_select, $serveur='',$requeter=true) {
1289 if (!$query_select) return false;
1290 // vue deja presente
1291 if (sql_showtable($nom, false, $serveur)) {
1292 if ($requeter) spip_log("Echec creation d'une vue sql ($nom) car celle-ci existe deja (serveur:$serveur)",'pg.'._LOG_ERREUR);
1293 return false;
1294 }
1295
1296 $query = "CREATE VIEW $nom AS ". $query_select;
1297 return spip_pg_query($query, $serveur, $requeter);
1298 }
1299
1300
1301 // http://doc.spip.org/@spip_pg_set_connect_charset
1302 function spip_pg_set_connect_charset($charset, $serveur='',$requeter=true){
1303 spip_log("changement de charset sql a ecrire en PG",'pg.'._LOG_ERREUR);
1304 }
1305
1306
1307 /**
1308 * Optimise une table SQL
1309 *
1310 * @param $table nom de la table a optimiser
1311 * @param $serveur nom de la connexion
1312 * @param $requeter effectuer la requete ? sinon retourner son code
1313 * @return bool|string true / false / requete
1314 **/
1315 // http://doc.spip.org/@spip_sqlite_optimize
1316 function spip_pg_optimize($table, $serveur='',$requeter=true){
1317 return spip_pg_query("VACUUM ". $table, $serveur, $requeter);
1318 }
1319
1320 // Selectionner la sous-chaine dans $objet
1321 // correspondant a $lang. Cf balise Multi de Spip
1322
1323 // http://doc.spip.org/@spip_pg_multi
1324 function spip_pg_multi ($objet, $lang) {
1325 $r = "regexp_replace("
1326 . $objet
1327 . ",'<multi>.*[[]"
1328 . $lang
1329 . "[]]([^[]*).*</multi>', E'\\\\1') AS multi";
1330 return $r;
1331 }
1332
1333 // Palanquee d'idiosyncrasies MySQL dans les creations de table
1334 // A completer par les autres, mais essayer de reduire en amont.
1335
1336 // http://doc.spip.org/@mysql2pg_type
1337 function mysql2pg_type($v){
1338 $remplace = array(
1339 '/auto_increment/i' => '', // non reconnu
1340 '/bigint/i' => 'bigint',
1341 '/mediumint/i' => 'mediumint',
1342 '/smallint/i'=> 'smallint',
1343 "/tinyint/i" => 'int',
1344 '/int\s*[(]\s*\d+\s*[)]/i' => 'int',
1345 "/longtext/i" => 'text',
1346 "/mediumtext/i" => 'text',
1347 "/tinytext/i" => 'text',
1348 "/longblob/i" => 'text',
1349 "/0000-00-00/" =>'0001-01-01',
1350 "/datetime/i" => 'timestamp',
1351 "/unsigned/i" => '',
1352 "/double/i" => 'double precision',
1353 '/VARCHAR\((\d+)\)\s+BINARY/i' => 'varchar(\1)',
1354 "/ENUM *[(][^)]*[)]/i" => "varchar(255)",
1355 '/(timestamp .* )ON .*$/is' => '\\1',
1356 );
1357
1358 return preg_replace(array_keys($remplace),array_values($remplace),$v);
1359 }
1360
1361 // Renvoie false si on n'a pas les fonctions pg (pour l'install)
1362 // http://doc.spip.org/@spip_versions_pg
1363 function spip_versions_pg(){
1364 charger_php_extension('pgsql');
1365 return function_exists('pg_connect');
1366 }
1367
1368 ?>