lychee/php/modules/db.php

<?php

/**
 * @name        DB Module
 * @author      Philipp Maurer
 * @author      Tobias Reich
 * @copyright   2014 by Philipp Maurer, Tobias Reich
 */

if (!defined('LYCHEE')) exit('Error: Direct access is not allowed!');

function dbConnect() {

    global $dbUser, $dbPassword, $dbHost, $dbName;

    $database = new mysqli($dbHost, $dbUser, $dbPassword);

    if (mysqli_connect_errno()) {
	    echo mysqli_connect_errno().': '.mysqli_connect_error();
        return false;
	}

	if (!$database->select_db($dbName))
		if (!createDatabase($dbName, $database)) exit('Error: Could not create database!');
    if (!$database->query("SELECT * FROM lychee_photos, lychee_albums, lychee_settings;"))
    	if (!createTables($database)) exit('Error: Could not create tables!');

    // Avoid sql injection on older MySQL versions
	if ($database->server_version<50500) $database->set_charset('GBK');

    return $database;

}

function dbClose() {

	global $database;

    if (!$database->close()) exit("Error: Closing the connection failed!");

    return true;

}

function createConfig($dbHost = 'localhost', $dbUser, $dbPassword, $dbName = 'lychee') {

	$dbPassword = urldecode($dbPassword);
	$database = new mysqli($dbHost, $dbUser, $dbPassword);

	if (mysqli_connect_errno()||$dbUser=="") return "Warning: Connection failed!";
	else {

$config = "<?php

/**
 * @name        Config
 * @author      Philipp Maurer
 * @author      Tobias Reich
 * @copyright   2014 by Philipp Maurer, Tobias Reich
*/

if(!defined('LYCHEE')) exit('Error: Direct access is not allowed!');

// Database configurations
\$dbHost = '$dbHost'; //Host of the Database
\$dbUser = '$dbUser'; //Username of the database
\$dbPassword = '$dbPassword'; //Password of the Database
\$dbName = '$dbName'; //Database name

?>";

		if (file_put_contents("config.php", $config)===false) return "Warning: Could not create file!";
		else {

			$_SESSION['login'] = true;
			return true;

		}

	}

}

function createDatabase($dbName, $database) {

	$result = $database->query("CREATE DATABASE IF NOT EXISTS $dbName;");
	$database->select_db($dbName);

	if (!$database->select_db($dbName)||!$result) return false;
	return true;

}

function createTables($database) {

	if (!$database->query("SELECT * FROM lychee_settings;")) {

		$query = "

			CREATE TABLE `lychee_settings` (
				`key` varchar(50) NOT NULL DEFAULT '',
				`value` varchar(50) DEFAULT ''
			) ENGINE=MyISAM DEFAULT CHARSET=latin1;

		";

		if (!$database->query($query)) return false;

		$query = "

			INSERT INTO `lychee_settings` (`key`, `value`)
			VALUES
			('username',''),
			('password',''),
			('thumbQuality','90'),
			('checkForUpdates','1'),
			('sorting','ORDER BY id DESC');

		";

		if (!$database->query($query)) return false;

	}

	if (!$database->query("SELECT * FROM lychee_albums;")) {

		$query = "

			CREATE TABLE `lychee_albums` (
				`id` int(11) NOT NULL AUTO_INCREMENT,
				`title` varchar(50) NOT NULL,
				`description` varchar(1000) DEFAULT '',
				`sysdate` varchar(10) NOT NULL,
				`public` tinyint(1) DEFAULT '0',
				`password` varchar(100) DEFAULT '',
				PRIMARY KEY (`id`)
			) ENGINE=MyISAM DEFAULT CHARSET=latin1;

		";

	    if (!$database->query($query)) return false;

    }

    if (!$database->query("SELECT * FROM lychee_photos;")) {

		$query = "

			CREATE TABLE `lychee_photos` (
				`id` bigint(14) NOT NULL,
				`title` varchar(50) NOT NULL,
				`description` varchar(1000) NOT NULL DEFAULT '',
				`url` varchar(100) NOT NULL,
				`public` tinyint(1) NOT NULL,
				`type` varchar(10) NOT NULL,
				`width` int(11) NOT NULL,
				`height` int(11) NOT NULL,
				`size` varchar(20) NOT NULL,
				`sysdate` varchar(10) NOT NULL,
				`systime` varchar(8) NOT NULL,
				`iso` varchar(15) NOT NULL,
				`aperture` varchar(20) NOT NULL,
				`make` varchar(20) NOT NULL,
				`model` varchar(50) NOT NULL,
				`shutter` varchar(30) NOT NULL,
				`focal` varchar(20) NOT NULL,
				`takedate` varchar(20) NOT NULL,
				`taketime` varchar(8) NOT NULL,
				`star` tinyint(1) NOT NULL,
				`thumbUrl` varchar(50) NOT NULL,
				`album` varchar(30) NOT NULL DEFAULT '0',
				`import_name` varchar(100) DEFAULT '',
				PRIMARY KEY (`id`)
			) ENGINE=MyISAM DEFAULT CHARSET=latin1;

		";

	    if (!$database->query($query)) return false;

    }

    return true;

}

?>