3 // Upgrade YOURLS and DB schema
\r
4 function yourls_upgrade( $step, $oldver, $newver, $oldsql, $newsql ) {
\r
5 // As of now, there's no other upgrading possibility.
\r
6 // In the future this function may contain tests to upgrade
\r
7 // from any release to the latest
\r
8 yourls_upgrade_to_14( $step );
\r
12 // Upgrade DB Schema from 1.3-RC1 or prior to 1.4
\r
13 function yourls_upgrade_to_14( $step ) {
\r
17 // create table log & table options
\r
18 // update table url structure
\r
20 yourls_create_tables_for_14();
\r
21 yourls_alter_url_table_to_14();
\r
22 yourls_clean_htaccess_for_14();
\r
23 yourls_create_htaccess();
\r
24 yourls_redirect_javascript( YOURLS_SITE."/admin/upgrade.php?step=2&oldver=1.3&newver=1.4&oldsql=100&newsql=200" );
\r
28 // convert each link in table url
\r
29 yourls_update_table_to_14();
\r
33 // update table url structure part 2: recreate indexes
\r
34 yourls_alter_url_table_to_14_part_two();
\r
35 yourls_redirect_javascript( YOURLS_SITE."/admin/upgrade.php?step=4&oldver=1.3&newver=1.4&oldsql=100&newsql=200" );
\r
39 // update version & db_version & next_id in the option table
\r
40 // attempt to drop YOURLS_DB_TABLE_NEXTDEC
\r
41 yourls_update_options_to_14();
\r
45 // Update options to reflect new version
\r
46 function yourls_update_options_to_14() {
\r
47 yourls_update_option( 'version', YOURLS_VERSION );
\r
48 yourls_update_option( 'db_version', YOURLS_DB_VERSION );
\r
51 $table = YOURLS_DB_TABLE_NEXTDEC;
\r
52 $next_id = $ydb->get_var("SELECT `next_id` FROM `$table`");
\r
53 yourls_update_option( 'next_id', $next_id );
\r
54 @$ydb->query( "DROP TABLE `$table`" );
\r
57 // Create new tables for YOURLS 1.4: options & log
\r
58 function yourls_create_tables_for_14() {
\r
63 $queries[YOURLS_DB_TABLE_OPTIONS] =
\r
64 'CREATE TABLE IF NOT EXISTS `'.YOURLS_DB_TABLE_OPTIONS.'` ('.
\r
65 '`option_id` int(11) unsigned NOT NULL auto_increment,'.
\r
66 '`option_name` varchar(64) NOT NULL default "",'.
\r
67 '`option_value` longtext NOT NULL,'.
\r
68 'PRIMARY KEY (`option_id`,`option_name`),'.
\r
69 'KEY `option_name` (`option_name`)'.
\r
72 $queries[YOURLS_DB_TABLE_LOG] =
\r
73 'CREATE TABLE IF NOT EXISTS `'.YOURLS_DB_TABLE_LOG.'` ('.
\r
74 '`click_id` int(11) NOT NULL auto_increment,'.
\r
75 '`click_time` datetime NOT NULL,'.
\r
76 '`shorturl` varchar(200) NOT NULL,'.
\r
77 '`referrer` varchar(200) NOT NULL,'.
\r
78 '`user_agent` varchar(255) NOT NULL,'.
\r
79 '`ip_address` varchar(41) NOT NULL,'.
\r
80 '`country_code` char(2) NOT NULL,'.
\r
81 'PRIMARY KEY (`click_id`),'.
\r
82 'KEY `shorturl` (`shorturl`,`referrer`,`country_code`)'.
\r
85 foreach( $queries as $query ) {
\r
86 $ydb->query( $query );
\r
89 echo "<p>New tables created. Please wait...</p>";
\r
93 // Alter table structure, part 1 (change schema, drop index)
\r
94 function yourls_alter_url_table_to_14() {
\r
96 $table = YOURLS_DB_TABLE_URL;
\r
100 $alters[] = "ALTER TABLE `$table` CHANGE `id` `keyword` VARCHAR( 200 ) NOT NULL";
\r
101 $alters[] = "ALTER TABLE `$table` CHANGE `url` `url` TEXT NOT NULL";
\r
102 $alters[] = "ALTER TABLE `$table` DROP PRIMARY KEY";
\r
104 foreach ( $alters as $query ) {
\r
105 $ydb->query( $query );
\r
108 echo "<p>Structure of existing tables updated. Please wait...</p>";
\r
111 // Alter table structure, part 2 (recreate index after the table is up to date)
\r
112 function yourls_alter_url_table_to_14_part_two() {
\r
114 $table = YOURLS_DB_TABLE_URL;
\r
117 $alters[] = "ALTER TABLE `$table` ADD PRIMARY KEY ( `keyword` )";
\r
119 foreach ( $alters as $query ) {
\r
120 $ydb->query( $query );
\r
123 echo "<p>New table index created</p>";
\r
126 // Convert each link from 1.3 (id) to 1.4 (keyword) structure
\r
127 function yourls_update_table_to_14() {
\r
129 $table = YOURLS_DB_TABLE_URL;
\r
131 // Modify each link to reflect new structure
\r
133 $from = isset($_GET['from']) ? intval( $_GET['from'] ) : 0 ;
\r
134 $total = yourls_get_db_stats();
\r
135 $total = $total['total_links'];
\r
137 $sql = "SELECT `keyword`,`url` FROM `$table` WHERE 1=1 ORDER BY `url` ASC LIMIT $from, $chunk ;";
\r
139 $rows = $ydb->get_results($sql);
\r
142 foreach( $rows as $row ) {
\r
143 $keyword = $row->keyword;
\r
145 $newkeyword = yourls_int2string( $keyword );
\r
146 $result = $ydb->query("UPDATE `$table` SET `keyword` = '$newkeyword' WHERE `url` = '$url';");
\r
150 if ( $count == $chunk ) {
\r
151 // there are probably other rows to convert
\r
152 $from = $from + $chunk;
\r
153 $remain = $total - $from;
\r
154 echo "<p>Converted $chunk database rows ($remain remaining). Continuing... Please do not close this window until it's finished!</p>";
\r
155 yourls_redirect_javascript( YOURLS_SITE."/admin/upgrade.php?step=2&oldver=1.3&newver=1.4&oldsql=100&newsql=200&from=$from" );
\r
158 echo '<p>All rows converted! Please wait...</p>';
\r
159 yourls_redirect_javascript( YOURLS_SITE."/admin/upgrade.php?step=3&oldver=1.3&newver=1.4&oldsql=100&newsql=200" );
\r
164 // Clean .htaccess as it existed before 1.4
\r
165 function yourls_clean_htaccess_for_14() {
\r
166 $filename = dirname(dirname(__FILE__)).'/.htaccess';
\r
168 $contents = implode( '', file( $filename );
\r
169 // remove "ShortURL" block
\r
170 $contents = preg_replace( '/# BEGIN ShortURL.*# END ShortURL/', '', $contents );
\r
171 // comment out deprecated RewriteRule
\r
172 $find = 'RewriteRule .* - [E=REMOTE_USER:%{HTTP:Authorization},L]';
\r
173 $replace = "# You can safely remove this 4 lines block -- it's no longer used in YOURLS\n".
\r
175 $contents = str_replace( $find, $replace, $contents );
\r
177 // Write cleaned file
\r
178 $f = fopen( $filename, 'w' );
\r
179 fwrite( $f, $contents );
\r
182 echo "<p>Old .htaccess file cleaned up</p>";
\r