#!/usr/bin/perl
-# setup_tables: setup MYSQL/PGSQL side of things for blootbot.
+# setup_tables: setup MYSQL/PGSQL side of things for infobot.
# written by the xk.
###
-require "src/core.pl";
require "src/logger.pl";
+require "src/core.pl";
require "src/modules.pl";
require "src/Misc.pl";
-require "src/interface.pl";
+require "src/CLI/Support.pl";
+
+$bot_src_dir = "src/";
-$bot_src_dir = "./src/";
+# read param stuff from infobot.config.
+&loadConfig("files/infobot.config");
-# read param stuff from blootbot.config.
-&loadConfig("files/blootbot.config");
&loadDBModules();
my $dbname = $param{'DBName'};
my $query;
if ($dbname eq "") {
- print "error: appears that teh config file was not loaded properly.\n";
+ print "error: appears that the config file was not loaded properly.\n";
exit 1;
}
exit 1;
}
- &openDB("mysql", $adminuser, $adminpass);
+ &sqlOpenDB("mysql", "mysql", $adminuser, $adminpass);
- &status("Creating db ...");
- &dbRaw("CREATE DATABASE $param{'DBName'}");
+ my $database_exists = 0;
+ foreach $database (&sqlRawReturn("SHOW DATABASES")) {
+ $database_exists++ if $database eq $param{DBName};
+ }
+ if ($database_exists) {
+ &status("Database '$param{DBName}' already exists. Continuing...");
+ } else {
+ &status("Creating db ...");
+ &sqlRaw("create(database)", "CREATE DATABASE $param{DBName}");
+ }
- # Step 1.
- &status("Step 1: Adding user information.");
+ &status("--- Adding user information for user '$param{'SQLUser'}'");
- # Step 2.
- if (!&dbGet("user","user",$param{'SQLUser'},"user")) {
- &status(" Adding user $param{'SQLUser'} $dbname/user table...");
+ if (!&sqlSelect("user", "user", { 'user' => &sqlQuote($param{'SQLUser'}) })) {
+ &status("--- Adding user '$param{'SQLUser'}' $dbname/user table...");
$query = "INSERT INTO user VALUES ".
"('localhost', '$param{'SQLUser'}', ".
"password('$param{'SQLPass'}'), ";
$query .= "'Y','Y','Y','Y','Y','Y','N','N','N','N','N','N','N','N')";
-### $query .= "'Y','Y','Y','Y','N','N','N','N','N','N')";
- &dbRaw("create(user)", $query);
+ &sqlRaw("create(user)", $query);
+ } else {
+ &status("... user information already present.");
}
- # Step 3. what's this for?
- if (!&dbGet("db","db",$param{'SQLUser'},"db")) {
- &status(" Adding 'db' entry");
+ if (!&sqlSelect("db", "db", { 'db' => &sqlQuote($param{'SQLUser'}) })) {
+ &status("--- Adding database information for database '$dbname'.");
$query = "INSERT INTO db VALUES ".
"('localhost', '$dbname', ".
"'$param{'SQLUser'}', ";
$query .= "'Y','Y','Y','Y','Y','Y','Y','N','N','N')";
-### $query .= "'Y','Y','Y','Y','Y','N')";
-
- &dbRaw("create(db)", $query);
- }
- # grant.
- &status(" Granting user access to table.");
- foreach ("factoids","seen","freshmeat") {
- $query = "GRANT SELECT,INSERT,UPDATE,DELETE ON $dbname.$_ TO $param{'SQLUser'}";
- &dbRaw("GRANT", $query);
+ &sqlRaw("create(db)", $query);
+ } else {
+ &status("... db info already present.");
}
# flush.
&status("Flushing privileges...");
- $query = "FLUSH PRIVILEGES"; # DOES NOT WORK on slink?
- &dbRaw("mysql(flush)", $query);
-
-} elsif ($param{'DBType'} =~ /pgsql|postgres/i) {
- if ($param{'DBType'} =~ /pgsql|postgres/i) { use Pg; } # for runtime.
- my $dbh = Pg::connectdb("dbname=$dbname");
-
- if (PGRES_CONNECTION_OK eq $conn->status) {
- print " opened mysql connection to $param{'mysqlHost'}\n";
- } else {
- print " error: cannot connect to $param{'mysqlHost'}.\n";
- print " $conn->errorMessage\n";
- exit 1;
- }
-
- # retrieve a list of db's from the server.
- my %db;
- foreach ($dbh->func('_ListTables')) {
- $db{$_} = 1;
- }
-
- # Step 4.
- print "Step 4: Creating the tables.\n";
-
- # factoid db.
- if (!exists $db{'factoids'}) {
- print " creating new table factoids...\n";
-
- $query = "CREATE TABLE factoids (".
- "factoid_key varying(64) NOT NULL,".
-
- "requested_by varying(64),".
- "requested_time numeric(11,0),".
- "requested_count numeric(5,0),".
- "created_by varying(64),".
- "created_time numeric(11,0),".
-
- "modified_by character varying(192),".
- "modified_time numeric(11,0),".
-
- "locked_by character varying(64),".
- "locked_time numeric(11,0),".
-
- "factoid_value text NOT NULL,".
-
- "PRIMARY KEY (factoid_key)".
- ")";
-
- &dbRaw("create(factoids)", $query);
- }
-
- # freshmeat.
- if (!exists $db{'freshmeat'}) {
- print " creating new table freshmeat...\n";
-
- $query = "CREATE TABLE freshmeat (".
- "name charcter varying(64) NOT NULL,".
- "stable character varying(32),".
- "devel character varying(32),".
- "section character varying(40),".
- "license character varying(32),".
- "homepage character varying(128),".
- "download character varying(128),".
- "changelog character varying(128),".
- "deb character varying(128),".
- "rpm character varying(128),".
- "link character varying(55),".
- "oneliner character varying(96) NOT NULL,".
-
- "PRIMARY KEY (name)".
- ")";
-
- &dbRaw("create(freshmeat)", $query);
- }
-
- # karma.
- if (!exists $db{'karma'}) {
- print " creating new table karma...\n";
-
- $query = "CREATE TABLE karma (".
- "nick character varying(20) NOT NULL,".
- "karma numeric(5,0),".
-
- "PRIMARY KEY (nick)".
- ")";
-
- &dbRaw("create(karma)", $query);
- }
-
- # rootwarn.
- if (!exists $db{'rootwarn'}) {
- print " creating new table rootwarn...\n";
-
- $query = "CREATE TABLE rootwarn (".
- "nick character varying(20) NOT NULL,".
- "attempt numeric(5,0),".
- "time numeric(11,0) NOT NULL,".
- "host character varying(80) NOT NULL,".
- "channel character varying(20) NOT NULL,".
-
- "PRIMARY KEY (nick)".
- ")";
-
- &dbRaw("create(rootwarn)", $query);
- }
-
- # seen.
- if (!exists $db{'seen'}) {
- print " creating new table seen...\n";
-
- $query = "CREATE TABLE seen (".
- "nick character varying(20) NOT NULL,".
- "time numeric(11,0) NOT NULL,".
- "channel character varying(20) NOT NULL,".
- "host character varying(80) NOT NULL,".
- "messagecount numeric(5,0),".
- "hehcount numeric(5,0),".
- "message text NOT NULL,".
-
- "PRIMARY KEY (nick)".
- ")";
-
- &dbRaw("create(seen)", $query);
- }
+ $query = "FLUSH PRIVILEGES";
+ &sqlRaw("mysql(flush)", $query);
}
-print "Done.\n";
+&status("Done.");
+
+&sqlCloseDB();
-&closeDB();
+# vim:ts=4:sw=4:expandtab:tw=80
projects / infobot.git / blobdiff