query( $sql ); echo "

Updating table structure. Please wait...

"; } /************************** 1.4.3 -> 1.5 **************************/ /** * Main func for upgrade from 1.4.3 to 1.5 * */ function yourls_upgrade_to_15( ) { // Create empty 'active_plugins' entry in the option if needed if( yourls_get_option( 'active_plugins' ) === false ) yourls_add_option( 'active_plugins', array() ); echo "

Enabling the plugin API. Please wait...

"; // Alter URL table to store titles global $ydb; $table_url = YOURLS_DB_TABLE_URL; $sql = "ALTER TABLE `$table_url` ADD `title` TEXT AFTER `url`;"; $ydb->query( $sql ); echo "

Updating table structure. Please wait...

"; // Update .htaccess yourls_create_htaccess(); echo "

Updating .htaccess file. Please wait...

"; } /************************** 1.4.1 -> 1.4.3 **************************/ /** * Main func for upgrade from 1.4.1 to 1.4.3 * */ function yourls_upgrade_to_143( ) { // Check if we have 'keyword' (borked install) or 'shorturl' (ok install) global $ydb; $table_log = YOURLS_DB_TABLE_LOG; $sql = "SHOW COLUMNS FROM `$table_log`"; $cols = $ydb->get_results( $sql ); if ( $cols[2]->Field == 'keyword' ) { $sql = "ALTER TABLE `$table_log` CHANGE `keyword` `shorturl` VARCHAR( 200 ) BINARY;"; $ydb->query( $sql ); } echo "

Structure of existing tables updated. Please wait...

"; } /************************** 1.4 -> 1.4.1 **************************/ /** * Main func for upgrade from 1.4 to 1.4.1 * */ function yourls_upgrade_to_141( ) { // Kill old cookies from 1.3 and prior setcookie('yourls_username', null, time() - 3600 ); setcookie('yourls_password', null, time() - 3600 ); // alter table URL yourls_alter_url_table_to_141(); // recreate the htaccess file if needed yourls_create_htaccess(); } /** * Alter table URL to 1.4.1 * */ function yourls_alter_url_table_to_141() { global $ydb; $table_url = YOURLS_DB_TABLE_URL; $alter = "ALTER TABLE `$table_url` CHANGE `keyword` `keyword` VARCHAR( 200 ) BINARY, CHANGE `url` `url` TEXT BINARY "; $ydb->query( $alter ); echo "

Structure of existing tables updated. Please wait...

"; } /************************** 1.3 -> 1.4 **************************/ /** * Main func for upgrade from 1.3-RC1 to 1.4 * */ function yourls_upgrade_to_14( $step ) { switch( $step ) { case 1: // create table log & table options // update table url structure // update .htaccess yourls_create_tables_for_14(); // no value returned, assuming it went OK yourls_alter_url_table_to_14(); // no value returned, assuming it went OK $clean = yourls_clean_htaccess_for_14(); // returns bool $create = yourls_create_htaccess(); // returns bool if ( !$create ) echo "

Please create your .htaccess file (I could not do it for you). Please refer to http://yourls.org/htaccess."; yourls_redirect_javascript( yourls_admin_url( "upgrade.php?step=2&oldver=1.3&newver=1.4&oldsql=100&newsql=200" ), $create ); break; case 2: // convert each link in table url yourls_update_table_to_14(); break; case 3: // update table url structure part 2: recreate indexes yourls_alter_url_table_to_14_part_two(); // update version & db_version & next_id in the option table // attempt to drop YOURLS_DB_TABLE_NEXTDEC yourls_update_options_to_14(); // Now upgrade to 1.4.1 yourls_redirect_javascript( yourls_admin_url( "upgrade.php?step=1&oldver=1.4&newver=1.4.1&oldsql=200&newsql=210" ) ); break; } } /** * Update options to reflect new version * */ function yourls_update_options_to_14() { yourls_update_option( 'version', '1.4' ); yourls_update_option( 'db_version', '200' ); if( defined('YOURLS_DB_TABLE_NEXTDEC') ) { global $ydb; $table = YOURLS_DB_TABLE_NEXTDEC; $next_id = $ydb->get_var("SELECT `next_id` FROM `$table`"); yourls_update_option( 'next_id', $next_id ); @$ydb->query( "DROP TABLE `$table`" ); } else { yourls_update_option( 'next_id', 1 ); // In case someone mistakenly deleted the next_id constant or table too early } } /** * Create new tables for YOURLS 1.4: options & log * */ function yourls_create_tables_for_14() { global $ydb; $queries = array(); $queries[YOURLS_DB_TABLE_OPTIONS] = 'CREATE TABLE IF NOT EXISTS `'.YOURLS_DB_TABLE_OPTIONS.'` ('. '`option_id` int(11) unsigned NOT NULL auto_increment,'. '`option_name` varchar(64) NOT NULL default "",'. '`option_value` longtext NOT NULL,'. 'PRIMARY KEY (`option_id`,`option_name`),'. 'KEY `option_name` (`option_name`)'. ');'; $queries[YOURLS_DB_TABLE_LOG] = 'CREATE TABLE IF NOT EXISTS `'.YOURLS_DB_TABLE_LOG.'` ('. '`click_id` int(11) NOT NULL auto_increment,'. '`click_time` datetime NOT NULL,'. '`shorturl` varchar(200) NOT NULL,'. '`referrer` varchar(200) NOT NULL,'. '`user_agent` varchar(255) NOT NULL,'. '`ip_address` varchar(41) NOT NULL,'. '`country_code` char(2) NOT NULL,'. 'PRIMARY KEY (`click_id`),'. 'KEY `shorturl` (`shorturl`)'. ');'; foreach( $queries as $query ) { $ydb->query( $query ); // There's no result to be returned to check if table was created (except making another query to check table existence, which we'll avoid) } echo "

New tables created. Please wait...

"; } /** * Alter table structure, part 1 (change schema, drop index) * */ function yourls_alter_url_table_to_14() { global $ydb; $table = YOURLS_DB_TABLE_URL; $alters = array(); $results = array(); $alters[] = "ALTER TABLE `$table` CHANGE `id` `keyword` VARCHAR( 200 ) NOT NULL"; $alters[] = "ALTER TABLE `$table` CHANGE `url` `url` TEXT NOT NULL"; $alters[] = "ALTER TABLE `$table` DROP PRIMARY KEY"; foreach ( $alters as $query ) { $ydb->query( $query ); } echo "

Structure of existing tables updated. Please wait...

"; } /** * Alter table structure, part 2 (recreate indexes after the table is up to date) * */ function yourls_alter_url_table_to_14_part_two() { global $ydb; $table = YOURLS_DB_TABLE_URL; $alters = array(); $alters[] = "ALTER TABLE `$table` ADD PRIMARY KEY ( `keyword` )"; $alters[] = "ALTER TABLE `$table` ADD INDEX ( `ip` )"; $alters[] = "ALTER TABLE `$table` ADD INDEX ( `timestamp` )"; foreach ( $alters as $query ) { $ydb->query( $query ); } echo "

New table index created

"; } /** * Convert each link from 1.3 (id) to 1.4 (keyword) structure * */ function yourls_update_table_to_14() { global $ydb; $table = YOURLS_DB_TABLE_URL; // Modify each link to reflect new structure $chunk = 45; $from = isset($_GET['from']) ? intval( $_GET['from'] ) : 0 ; $total = yourls_get_db_stats(); $total = $total['total_links']; $sql = "SELECT `keyword`,`url` FROM `$table` WHERE 1=1 ORDER BY `url` ASC LIMIT $from, $chunk ;"; $rows = $ydb->get_results($sql); $count = 0; $queries = 0; foreach( $rows as $row ) { $keyword = $row->keyword; $url = $row->url; $newkeyword = yourls_int2string( $keyword ); $ydb->query("UPDATE `$table` SET `keyword` = '$newkeyword' WHERE `url` = '$url';"); if( $ydb->result === true ) { $queries++; } else { echo "

Huho... Could not update rown with url='$url', from keyword '$keyword' to keyword '$newkeyword'

"; // Find what went wrong :/ } $count++; } // All done for this chunk of queries, did it all go as expected? $success = true; if( $count != $queries ) { $success = false; $num = $count - $queries; echo "

$num error(s) occured while updating the URL table :(

"; } if ( $count == $chunk ) { // there are probably other rows to convert $from = $from + $chunk; $remain = $total - $from; echo "

Converted $chunk database rows ($remain remaining). Continuing... Please do not close this window until it's finished!

"; yourls_redirect_javascript( yourls_admin_url( "upgrade.php?step=2&oldver=1.3&newver=1.4&oldsql=100&newsql=200&from=$from" ), $success ); } else { // All done echo '

All rows converted! Please wait...

'; yourls_redirect_javascript( yourls_admin_url( "upgrade.php?step=3&oldver=1.3&newver=1.4&oldsql=100&newsql=200" ), $success ); } } /** * Clean .htaccess as it existed before 1.4. Returns boolean * */ function yourls_clean_htaccess_for_14() { $filename = YOURLS_ABSPATH.'/.htaccess'; $result = false; if( is_writeable( $filename ) ) { $contents = implode( '', file( $filename ) ); // remove "ShortURL" block $contents = preg_replace( '/# BEGIN ShortURL.*# END ShortURL/s', '', $contents ); // comment out deprecated RewriteRule $find = 'RewriteRule .* - [E=REMOTE_USER:%{HTTP:Authorization},L]'; $replace = "# You can safely remove this 5 lines block -- it's no longer used in YOURLS\n". "# $find"; $contents = str_replace( $find, $replace, $contents ); // Write cleaned file $f = fopen( $filename, 'w' ); fwrite( $f, $contents ); fclose( $f ); $result = true; } return $result; }