Move the "dbObject"s to their own package.

[gigi.git] / src / org / cacert / gigi / util / Notary.java

diff --git a/src/org/cacert/gigi/util/Notary.java b/src/org/cacert/gigi/util/Notary.java
index f7327eb9b3e1a1fce672354888d196e8d1aee62d..46a2c3ade06881a8d8acf32e663f7b15d774dd9f 100644 (file)
--- a/src/org/cacert/gigi/util/Notary.java
+++ b/src/org/cacert/gigi/util/Notary.java
@@ -1,91 +1,84 @@
 package org.cacert.gigi.util;
 
-import java.io.PrintWriter;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
 
-import org.cacert.gigi.User;
 import org.cacert.gigi.database.DatabaseConnection;
+import org.cacert.gigi.dbObjects.User;
 
 public class Notary {
-       public static void writeUserAgreement(int memid, String document,
-                       String method, String comment, boolean active, int secmemid)
-                       throws SQLException {
-               PreparedStatement q = DatabaseConnection
-                               .getInstance()
-                               .prepare(
-                                               "insert into `user_agreements` set `memid`=?, `secmemid`=?,"
-                                                               + " `document`=?,`date`=NOW(), `active`=?,`method`=?,`comment`=?");
-               q.setInt(1, memid);
-               q.setInt(2, secmemid);
-               q.setString(3, document);
-               q.setInt(4, active ? 1 : 0);
-               q.setString(5, method);
-               q.setString(6, comment);
-               q.execute();
-       }
 
-       public static boolean checkAssuranceIsPossible(User assurer, User target,
-                       PrintWriter errOut) {
-               if (assurer.getId() == target.getId()) {
-                       if (errOut != null) {
-                               errOut.println("Cannot assure myself.");
-                       }
-                       return false;
-               }
-               try {
-                       PreparedStatement ps = DatabaseConnection
-                                       .getInstance()
-                                       .prepare(
-                                                       "SELECT 1 FROM `notary` where `to`=? and `from`=? AND `deleted`=0");
-                       ps.setInt(1, target.getId());
-                       ps.setInt(2, assurer.getId());
-                       ResultSet rs = ps.executeQuery();
-                       if (rs.next()) {
-                               if (errOut != null) {
-                                       errOut.println("You already assured this person.");
-                               }
-                               rs.close();
-                               return false;
-                       }
-                       rs.close();
-                       if (!assurer.canAssure()) {
-                               if (errOut != null) {
-                                       errOut.println("You cannot assure.");
-                               }
-                               return false;
-                       }
-               } catch (SQLException e) {
-                       e.printStackTrace();
-               }
-               return true;
-       }
+    public static void writeUserAgreement(int memid, String document, String method, String comment, boolean active, int secmemid) throws SQLException {
+        PreparedStatement q = DatabaseConnection.getInstance().prepare("insert into `user_agreements` set `memid`=?, `secmemid`=?," + " `document`=?,`date`=NOW(), `active`=?,`method`=?,`comment`=?");
+        q.setInt(1, memid);
+        q.setInt(2, secmemid);
+        q.setString(3, document);
+        q.setInt(4, active ? 1 : 0);
+        q.setString(5, method);
+        q.setString(6, comment);
+        q.execute();
+    }
 
-       public synchronized static boolean assure(User assurer, User target,
-                       int awarded, String location, String date) throws SQLException {
-               if (!checkAssuranceIsPossible(assurer, target, null)) {
-                       return false;
-               }
-               User u = new User(target.getId());
-               if (!u.equals(target)) {
-                       return false;
-               }
-               System.out.println("Would now assure.");
-               if (awarded > assurer.getMaxAssurePoints() || awarded < 0) {
-                       return false;
-               }
+    public static AssuranceResult checkAssuranceIsPossible(User assurer, User target) {
+        if (assurer.getId() == target.getId()) {
+            return AssuranceResult.CANNOT_ASSURE_SELF;
+        }
+        try {
+            PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT 1 FROM `notary` where `to`=? and `from`=? AND `deleted`=0");
+            ps.setInt(1, target.getId());
+            ps.setInt(2, assurer.getId());
+            ResultSet rs = ps.executeQuery();
+            if (rs.next()) {
+                rs.close();
+                return AssuranceResult.ALREADY_ASSUREED;
+            }
+            rs.close();
+            if ( !assurer.canAssure()) {
+                return AssuranceResult.NO_ASSURER;
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return AssuranceResult.ASSURANCE_SUCCEDED;
+    }
 
-               PreparedStatement ps = DatabaseConnection
-                               .getInstance()
-                               .prepare(
-                                               "INSERT INTO `notary` SET `from`=?, `to`=?, `points`=?, `location`=?, `date`=?");
-               ps.setInt(1, assurer.getId());
-               ps.setInt(2, target.getId());
-               ps.setInt(3, awarded);
-               ps.setString(4, location);
-               ps.setString(5, date);
-               ps.execute();
-               return true;
-       }
+    public enum AssuranceResult {
+        NO_ASSURER("You are not an assurer."), ALREADY_ASSUREED("You already assured this person."), CANNOT_ASSURE_SELF("Cannot assure myself."), ASSURANCE_SUCCEDED(""), ASSUREE_CHANGED("Person details changed. Please start over again."), POINTS_OUT_OF_RANGE("Points out of range.");
+
+        private final String message;
+
+        private AssuranceResult(String message) {
+            this.message = message;
+        }
+
+        public String getMessage() {
+            return message;
+        }
+    }
+
+    public synchronized static AssuranceResult assure(User assurer, User target, int awarded, String location, String date) throws SQLException {
+        AssuranceResult can = checkAssuranceIsPossible(assurer, target);
+        if (can != AssuranceResult.ASSURANCE_SUCCEDED) {
+            return can;
+        }
+        User u = new User(target.getId());
+        if ( !u.equals(target)) {
+            return AssuranceResult.ASSUREE_CHANGED;
+        }
+        if (awarded > assurer.getMaxAssurePoints() || awarded < 0) {
+            return AssuranceResult.POINTS_OUT_OF_RANGE;
+        }
+
+        PreparedStatement ps = DatabaseConnection.getInstance().prepare("INSERT INTO `notary` SET `from`=?, `to`=?, `points`=?, `location`=?, `date`=?");
+        ps.setInt(1, assurer.getId());
+        ps.setInt(2, target.getId());
+        ps.setInt(3, awarded);
+        ps.setString(4, location);
+        ps.setString(5, date);
+        ps.execute();
+        assurer.invalidateMadeAssurances();
+        target.invalidateReceivedAssurances();
+        return AssuranceResult.ASSURANCE_SUCCEDED;
+    }
 }