#!/usr/bin/php
<?php
require_once(dirname(__FILE__).'/shared/include/includes.php'); // USE __DIR__ in 5.3.0
require_once(SHARED.'/config.php');
require_once(SHARED.'/include/dbinit.php');
function prompt_bool($prompt, $default=true) {
	$yes=$default?'Y':'y';
	$no=$default?'n':'N';
	while (true) {
		echo trim($prompt)." [$yes/$no] ";
		$answer=trim(fgets(STDIN));
		if (strlen($answer) == 0) {
			return $default;
		}
		if (strtolower(substr($answer, 0, 1)) == 'y') {
			return true;
		} elseif (strtolower(substr($answer, 0, 1)) == 'n') {
			return false;
		} else {
			echo "Please respond yes or no.\n";
		}
	}
}
function prompt_string($prompt, $default=null) {
	echo $prompt.' '.($default?"[$default] ":'');
	$answer=trim(fgets(STDIN));
	if (strlen($answer) == 0) {
		if ($default) {
			return $default;
		}
	}
	return $answer;
}
$pdo=&$S['pdo'];
$r=$pdo->query('SHOW TABLES');
$tables=$r->fetchAll(PDO::FETCH_COLUMN);
foreach ($tables as $i => $table) {
	if (sql_row_obj::table_to_class($table)) {
		unset($tables[$i]);
	}
}
if ($tables) {
	echo 'Found table(s) without corresponding class: '.implode(', ', $tables)."\n";
	if (!prompt_bool('Generate classes for these tables?')) {
		unset($tables);
	}
} else {
	echo "No tables found without corresponding classes.\n";
}
if (!$tables) {
	echo 'Which tables should we generate classes for? (space-separated) ';
	$tables=explode(' ', trim(fgets(STDIN)));
	if ($tables[0] == '') {
		echo "Exiting.\n";
		exit;
	}
}
echo 'Generating classes for table(s): '.implode(', ', $tables)."\n";
foreach ($tables as $table) {
	$default='sql_';
	if (substr($table, -1) == 's') {
		$default.=substr($table, 0, strlen($table)-1);
	} else {
		$default.=$table;
	}
	$class=prompt_string('Class name for table '.$table.':', $default);
	if (substr($class, 0, 4) == 'sql_') {
		$default=substr($class, 4);
	} else {
		$default=$class;
	}
	$default.='.php';
	$file=SHARED.'/classes/'.prompt_string('Filename for class '.$class.': '.SHARED.'/classes/', $default);
	if (is_file($file)) {
		if (!prompt_bool($file.' exists. Overwrite?', false)) {
			echo 'Skipping table '.$table."\n";
			continue;
		}
	}
	echo 'Writing '.$file.'...';
	$file=fopen($file, 'w');
	fputs($file, "<?php\n");
	eval ("class fake_class_for_newclass extends sql_row_obj {\nprotected \$table='$table';\t}");
	$obj=new fake_class_for_newclass();
	fputs($file, str_replace('fake_class_for_newclass', $class, $obj->to_php()));
	fputs($file, "?>\n");
	fclose($file);
	echo "done\n";
}
?>