make pagination more robust. add undelete method for posts, if you know the id.

2 files changed, 10 insertions, 3 deletions

diff --git a/lib/database.php b/lib/database.php
index 0ffdb3a..5774d95 100644
--- a/lib/database.php
+++ b/lib/database.php
@@ -3,6 +3,9 @@
 //connect or create the database
 try {
 	$db = new PDO('sqlite:'.ROOT.DS.'posts.db');
+	$db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+	$db->setAttribute(PDO::ATTR_EMULATE_PREPARES, false);
+
 	$config['db_version'] = $db->query("PRAGMA user_version")->fetch(PDO::FETCH_ASSOC)['user_version'];
 } catch(PDOException $e) {
 	print 'Exception : '.$e->getMessage();
diff --git a/lib/functions.php b/lib/functions.php
index 268808c..7046eb5 100644
--- a/lib/functions.php
+++ b/lib/functions.php
@@ -40,7 +40,7 @@ function db_insert($content, $timestamp=NOW) {
 	return $db->lastInsertId();
 }
 
-function db_delete($post_id) {
+function db_delete($post_id, $undelete=false) {
 	global $db;
 	if(empty($db)) return false;
 	if(!is_numeric($post_id) || $post_id <= 0) return false;
@@ -50,10 +50,14 @@ function db_delete($post_id) {
 	$statement->bindParam(':id', $post_id, PDO::PARAM_INT);
 	*/
 
+	// delete or undelete/restore
+	$post_deleted = !$undelete ? time() : null;
+	$type = !$undelete ? PDO::PARAM_INT : PDO::PARAM_NULL;
+
 	// mark as deleted instead (for undo?!)
 	$statement = $db->prepare('UPDATE posts SET post_deleted = :post_deleted WHERE id = :id');
 	$statement->bindValue(':id', $post_id, PDO::PARAM_INT);
-	$statement->bindValue(':post_deleted', time(), PDO::PARAM_INT);
+	$statement->bindValue(':post_deleted', $post_deleted, $type);
 
 	$statement->execute();
 
@@ -115,7 +119,7 @@ function db_posts_count() {
 	global $db;
 	if(empty($db)) return false;
 
-	$statement = $db->prepare('SELECT COUNT(*) AS posts_count FROM posts');
+	$statement = $db->prepare('SELECT COUNT(*) AS posts_count FROM posts WHERE post_deleted IS NULL');
 	$statement->execute();
 	$row = $statement->fetch(PDO::FETCH_ASSOC);