X-Git-Url: https://code.wpia.club/?a=blobdiff_plain;f=src%2Forg%2Fcacert%2Fgigi%2Futil%2FNotary.java;h=cda4ce5a1ac222f925057d761d617cfd35706c71;hb=aa382fb02c1a8674c461e0f51b75a77056e24f80;hp=657264a6e7c90b8dbf589aee579c0136373e6f49;hpb=7f4a02fc53f56a5507d1d059c5c0636ed977ae62;p=gigi.git

diff --git a/src/org/cacert/gigi/util/Notary.java b/src/org/cacert/gigi/util/Notary.java
index 657264a6..cda4ce5a 100644
--- a/src/org/cacert/gigi/util/Notary.java
+++ b/src/org/cacert/gigi/util/Notary.java
@@ -1,6 +1,5 @@
 package org.cacert.gigi.util;
 
-import java.io.PrintWriter;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
@@ -9,83 +8,77 @@ import org.cacert.gigi.User;
 import org.cacert.gigi.database.DatabaseConnection;
 
 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()) {
-			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;
+    }
 }