Common-RDBO-Helper/t/lib/NewDB.pm

   1 package NewDB;
   2
   3 use DBI;
   4
   5 sub new {
   6     my $class = shift;
   7
   8     my $self = { db => "/tmp/common-rdbo-helper.db" };
   9
  10     bless $self, $class;
  11
  12     return $self;
  13 }
  14
  15 sub init {
  16     my $self = shift;
  17
  18     my $db = $self->{ db };
  19
  20     unless ( -f $db ) {
  21         warn 'Creating new db...';
  22
  23         my $dbh = DBI->connect( "dbi:SQLite:$db" ) or die $DBI::errstr;
  24
  25         $dbh->do( <<SQL );
  26 CREATE TABLE `post` (
  27   `id` INTEGER PRIMARY KEY NOT NULL,
  28   `addtime` INTEGER NOT NULL,
  29   `title` VARCHAR(255) NOT NULL,
  30   `content` VARCHAR(255) NOT NULL
  31 );
  32 SQL
  33
  34         $dbh->do( <<SQL );
  35 CREATE TABLE `article` (
  36   `id` INTEGER PRIMARY KEY NOT NULL,
  37   `addtime` INTEGER NOT NULL,
  38   `title` VARCHAR(255) NOT NULL,
  39   `content` VARCHAR(255) NOT NULL
  40 );
  41 SQL
  42
  43         $dbh->do( <<SQL );
  44 CREATE TABLE tag (
  45   id INTEGER PRIMARY KEY NOT NULL,
  46   name VARCHAR NOT NULL,
  47   UNIQUE(name)
  48 );
  49 SQL
  50
  51         $dbh->do( <<SQL );
  52 CREATE TABLE post_tag_map (
  53   post_id INTEGER KEY NOT NULL,
  54   tag_id INTEGER NOT NULL,
  55   PRIMARY KEY(post_id, tag_id)
  56 );
  57 SQL
  58
  59         $dbh->do( <<SQL );
  60 CREATE TABLE comment (
  61   id INTEGER PRIMARY KEY NOT NULL,
  62   master_id INTEGER NOT NULL,
  63   type VARCHAR,
  64   user_id INTEGER,
  65   `addtime` INTEGER NOT NULL,
  66   name VARCHAR,
  67   url VARCHAR,
  68   email VARCHAR,
  69   content TEXT NOT NULL
  70 );
  71 SQL
  72
  73         $dbh->do( <<SQL );
  74 CREATE TABLE pic (
  75   id INTEGER PRIMARY KEY NOT NULL,
  76   master_id INTEGER NOT NULL,
  77   type VARCHAR,
  78   user_id INTEGER,
  79   `addtime` INTEGER NOT NULL,
  80   pic VARCHAR
  81 );
  82 SQL
  83
  84         $dbh->disconnect();
  85     }
  86 }
  87
  88 sub cleanup {
  89     my $self = shift;
  90
  91     unlink $self->{ db };
  92 }
  93
  94 =head1 AUTHOR
  95
  96 vti
  97
  98 =head1 LICENSE
  99
 100 This library is free software, you can redistribute it and/or modify
 101 it under the same terms as Perl itself.
 102
 103 =cut
 104
 105 1;