Changeset 219 for trunk/lib/Vote/DB/Poll.pm

Timestamp:

04/19/09 20:06:03 (15 years ago)

Author:

nanardon

Message:

• add specific object to manage ballot data

File:

• trunk/lib/Vote/DB/Poll.pm (modified) (3 diffs)

trunk/lib/Vote/DB/Poll.pm

@@ -9 +9 @@
 use Crypt::RSA::Key::Public::SSH;
 use Crypt::RSA::Key::Private::SSH;
-use Crypt::CBC;
-use XML::Simple;
 use MIME::Base64;
 use Vote::DB::Ballot;
 use Vote::DB::Voting;
 use Vote::DB::Choice;
+use Vote::DB::Poll::Bdata;
 
 =head1 NAME
@@ -371 +370 @@
 }
 
-sub _register_ballot {
-    my ($self, $choice, $fchoice) = @_;
-
-    my $uid = ($self->is_crypted
-        ? $self->_register_ballot_crypted($choice, $fchoice)
-        : $self->_register_ballot_clear($choice, $fchoice))
-        or do {
-            $self->rollback;
-            return;
-        };
-
-    $uid
-}
-
-sub _register_ballot_clear {
-    my ($self, $choice, $fchoice, $uid) = @_;
-
-    my $addb = $self->db->prepare_cached(
-        q{
-        insert into ballot (id, poll, invalid) values (?,?,?)
-        }
-    );
-    $uid ||= Vote::DB::common::gen_uid();
-    $addb->execute($uid, $self->voteid, scalar(@{$fchoice || []}) ? undef : 'f') or do {
+sub find_choice_key {
+    my ($self, $value) = @_;
+
+    my $sth = $self->db->prepare_cached(
+        q{select key from choice where lower(label) = ? and poll = ?}
+    );
+    $sth->execute(lc($value), $self->voteid);
+    my $res = $sth->fetchrow_hashref;
+    $sth->finish;
+    $res->{key}
+}
+
+sub decrypted_ballots {
+    my ($self, $password) = @_;
+    my $privkey = $self->private_key($password) or return;
+    foreach ($self->list_ballot_need_dec) {
+        my $bdata = $self->bdata;
+        $bdata->decrypt($_, $privkey);
+    }
+    1
+}
+
+sub register_ballot {
+    my ($self, $vmail, $choice, $referal) = @_;
+
+    my $bdata = $self->bdata;
+    foreach (@{ $choice || []}) {
+        $bdata->add_item($_);
+    }
+    $bdata->submit($vmail, $referal) or do {
         $self->rollback;
         return;
     };
 
-    $self->_register_ballot_items($uid, $choice, $fchoice) or do {
-        $self->rollback;
-        return;
-    };
-
-    $uid
-}
-
-sub find_choice_key {
-    my ($self, $value) = @_;
-
-    my $sth = $self->db->prepare_cached(
-        q{select key from choice where lower(label) = ? and poll = ?}
-    );
-    $sth->execute(lc($value), $self->voteid);
-    my $res = $sth->fetchrow_hashref;
-    $sth->finish;
-    $res->{key}
-}
-
-sub _register_ballot_items {
-    my ($self, $uid, $choice, $fchoice) = @_;
-
-    my $addbc = $self->db->prepare_cached(
-        q{
-        insert into ballot_item (id, value, fromlist) values (?,?,?)
-        }
-    );
-    foreach (@{ $choice || []}) {
-        $addbc->execute($uid, $_, 't') or do {
-            $self->rollback;
-            return;
-        };
-    }
-    foreach (@{ $fchoice || []}) {
-        $_ or next;
-        my $chkey = $self->find_choice_key($_);
-        $addbc->execute($uid, $_, $chkey ? 't' : 'f') or do {
-            $self->rollback;
-            return;
-        };
-    }
-
-    $uid;
-}
-
-sub _register_ballot_crypted {
-    my ($self, $choice, $fchoice) = @_;
-    my $xml = XML::Simple->new(ForceArray => 1, RootName => 'ballot');
-    my $symkey = map{ chr(rand(256)) } (1 .. (256 / 8));
-    my $cipher = new Crypt::CBC($symkey, 'DES');
-    my $ballotuid = Vote::DB::common::gen_uid();
-    my $encryptedballot = $cipher->encrypt_hex(
-        $xml->XMLout({
-            id => $ballotuid,
-            sbal => $choice,
-            fsbal => $fchoice
-        })
-    );
-    my $encsymkey = $self->rsa->encrypt (
-        Message    => $symkey,
-        Key        => $self->public_key,
-        Armour     => 1,
-    ) || die $self->rsa->errstr();
-
-    my $addenc = $self->db->prepare_cached(
-        q{insert into ballot_enc (id, data, enckey, poll) values (?,?,?,?)}
-    );
-
-    my $uid = Vote::DB::common::gen_uid();
-    $addenc->execute($uid, $encryptedballot, $encsymkey, $self->voteid);
-    $ballotuid;
-}
-
-sub _decrypted_ballot {
-    my ($self, $ballotid, $privkey) = @_;
-    my $sth = $self->db->prepare_cached(
-        q{select * from ballot_enc where id = ? for update}
-    );
-    $sth->execute($ballotid);
-    my $ballot = $sth->fetchrow_hashref;
-    $sth->finish;
-    my $encsymkey = $ballot->{enckey};
-    my $data = $ballot->{data};
-    my $symkey = $self->rsa->decrypt (
-        Cyphertext => $encsymkey,
-        Key        => $privkey,
-        Armour     => 1,
-    ) || die $self->rsa->errstr();
-    my $cipher = new Crypt::CBC($symkey, 'DES');
-    my $xmldata = XMLin($cipher->decrypt_hex($data), ForceArray => 1);
-    $self->_register_ballot_clear($xmldata->{sbal}, $xmldata->{fsbal}, $xmldata->{id});
-    my $upd = $self->db->prepare_cached(q{update ballot_enc set decrypted = true where id = ?});
-    if ($upd->execute($ballotid)) {
-        $self->commit;
-        return;
-    } else {
-        $self->rollback;
-        return 1;
-    }
-}    
-
-sub decrypted_ballots {
-    my ($self, $password) = @_;
-    my $privkey = $self->private_key($password);
-    foreach ($self->list_ballot_need_dec) {
-        $self->_decrypted_ballot($_, $privkey);
-    }
-}
-
-sub register_ballot {
-    my ($self, $vmail, $choice, $fchoice, $referal) = @_;
-
-    my $uid;
-    for (0..2) { # 3 try
-    # First we register voting has voted
-    $self->_register_signing($vmail, $referal) or return; # TODO error ?
-
-    # registring choices
-    $uid = $self->_register_ballot($choice, $fchoice);
-    defined($uid) and last;
-
-    }
     # everything went fine, saving!
     $self->commit;
 
-    $uid
+    $bdata->uid
 }
 
@@ -589 +471 @@
 
     return 1;
+}
+
+sub bdata {
+    my ($self) = @_;
+    Vote::DB::Poll::Bdata->new($self->{dbstring}, $self);
 }