util-testing/org/cacert/gigi/pages/Manager.java

   1 package org.cacert.gigi.pages;
   2
   3 import java.io.IOException;
   4 import java.io.PrintWriter;
   5 import java.lang.reflect.Field;
   6 import java.sql.Date;
   7 import java.util.Calendar;
   8 import java.util.GregorianCalendar;
   9 import java.util.HashMap;
  10 import java.util.Iterator;
  11 import java.util.LinkedList;
  12 import java.util.Locale;
  13 import java.util.Map;
  14 import java.util.Properties;
  15
  16 import javax.servlet.http.HttpServletRequest;
  17 import javax.servlet.http.HttpServletResponse;
  18
  19 import org.cacert.gigi.GigiApiException;
  20 import org.cacert.gigi.database.DatabaseConnection;
  21 import org.cacert.gigi.database.GigiPreparedStatement;
  22 import org.cacert.gigi.dbObjects.EmailAddress;
  23 import org.cacert.gigi.dbObjects.Group;
  24 import org.cacert.gigi.dbObjects.Name;
  25 import org.cacert.gigi.dbObjects.User;
  26 import org.cacert.gigi.email.EmailProvider;
  27 import org.cacert.gigi.localisation.Language;
  28 import org.cacert.gigi.output.template.Form;
  29 import org.cacert.gigi.output.template.IterableDataset;
  30 import org.cacert.gigi.output.template.Template;
  31 import org.cacert.gigi.util.Notary;
  32
  33 public class Manager extends Page {
  34
  35     public static final String PATH = "/manager";
  36
  37     Field f;
  38
  39     private Manager() {
  40         super("Test Manager");
  41         try {
  42             f = EmailAddress.class.getDeclaredField("hash");
  43             f.setAccessible(true);
  44         } catch (ReflectiveOperationException e) {
  45             throw new Error(e);
  46         }
  47     }
  48
  49     public User[] getAssurers() {
  50         if (assurers != null) {
  51             return assurers;
  52         }
  53         assurers = new User[10];
  54         try {
  55             GigiPreparedStatement ps = DatabaseConnection.getInstance().prepare("INSERT INTO `notary` SET `from`=?, `to`=?, `points`=?, `location`=?, `date`=?");
  56             for (int i = 0; i < assurers.length; i++) {
  57                 String mail = "test-assurer" + i + "@example.com";
  58                 User u = User.getByEmail(mail);
  59                 if (u == null) {
  60                     createUser(mail);
  61                     u = User.getByEmail(mail);
  62                     passCATS(u);
  63                     ps.setInt(1, u.getId());
  64                     ps.setInt(2, u.getId());
  65                     ps.setInt(3, 100);
  66                     ps.setString(4, "Manager init code");
  67                     ps.setString(5, "1990-01-01");
  68                     ps.execute();
  69                 }
  70                 assurers[i] = u;
  71
  72             }
  73         } catch (ReflectiveOperationException | GigiApiException e) {
  74             e.printStackTrace();
  75         }
  76         return assurers;
  77     }
  78
  79     private void passCATS(User u) {
  80         GigiPreparedStatement ps = DatabaseConnection.getInstance().prepare("INSERT INTO cats_passed SET user_id=?, variant_id=3");
  81         ps.setInt(1, u.getId());
  82         ps.execute();
  83     }
  84
  85     private static Manager instance;
  86
  87     Template t = new Template(Manager.class.getResource("ManagerMails.templ"));
  88
  89     HashMap<String, LinkedList<String>> emails = new HashMap<>();
  90
  91     public static Manager getInstance() {
  92         if (instance == null) {
  93             instance = new Manager();
  94         }
  95         return instance;
  96     }
  97
  98     public static class MailFetcher extends EmailProvider {
  99
 100         public MailFetcher(Properties p) {}
 101
 102         @Override
 103         public String checkEmailServer(int forUid, String address) throws IOException {
 104             return OK;
 105         }
 106
 107         @Override
 108         public synchronized void sendmail(String to, String subject, String message, String from, String replyto, String toname, String fromname, String errorsto, boolean extra) throws IOException {
 109             HashMap<String, LinkedList<String>> mails = Manager.getInstance().emails;
 110             LinkedList<String> hismails = mails.get(to);
 111             if (hismails == null) {
 112                 mails.put(to, hismails = new LinkedList<>());
 113             }
 114             hismails.addFirst(subject + "\n" + message);
 115         }
 116
 117     }
 118
 119     public class ManagementForm extends Form {
 120
 121         public ManagementForm(HttpServletRequest hsr) {
 122             super(hsr);
 123         }
 124
 125         @Override
 126         public boolean submit(PrintWriter out, HttpServletRequest req) throws GigiApiException {
 127             return false;
 128         }
 129
 130         @Override
 131         protected void outputContent(PrintWriter out, Language l, Map<String, Object> vars) {
 132             getDefaultTemplate().output(out, l, vars);
 133         }
 134
 135     }
 136
 137     public void batchCreateUsers(String mailPrefix, String domain, int amount, PrintWriter out) {
 138
 139         try {
 140             if (amount > 100) {
 141                 out.print("100 at most, please.");
 142                 return;
 143             }
 144             for (int i = 0; i < amount; i++) {
 145                 String email = mailPrefix + i + "@" + domain;
 146                 createUser(email);
 147             }
 148         } catch (ReflectiveOperationException e) {
 149             out.println("failed");
 150             e.printStackTrace();
 151         } catch (GigiApiException e) {
 152             out.println("failed: " + e.getMessage());
 153             e.printStackTrace();
 154         }
 155     }
 156
 157     private void createUser(String email) throws GigiApiException, IllegalAccessException {
 158         User u = new User();
 159         u.setName(new Name("Först", "Läst", "Müddle", "Süffix"));
 160         u.setEmail(email);
 161         Calendar gc = GregorianCalendar.getInstance();
 162         gc.set(1990, 0, 1);
 163         u.setDoB(new Date(gc.getTime().getTime()));
 164         u.setPreferredLocale(Locale.ENGLISH);
 165         u.insert("xvXV12°§");
 166         EmailAddress ea = new EmailAddress(u, email);
 167         ea.insert(Language.getInstance(Locale.ENGLISH));
 168         String hash = (String) f.get(ea);
 169
 170         ea.verify(hash);
 171     }
 172
 173     User[] assurers;
 174
 175     @Override
 176     public void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
 177         if (req.getParameter("create") != null) {
 178             batchCreateUsers(req.getParameter("prefix"), req.getParameter("suffix"), Integer.parseInt(req.getParameter("amount")), resp.getWriter());
 179             resp.getWriter().println("User batch created.");
 180         } else if (req.getParameter("addpriv") != null || req.getParameter("delpriv") != null) {
 181             User u = User.getByEmail(req.getParameter("email"));
 182             if (u == null) {
 183                 resp.getWriter().println("User not found.");
 184                 return;
 185             }
 186             if (req.getParameter("addpriv") != null) {
 187                 u.grantGroup(u, Group.getByString(req.getParameter("priv")));
 188                 resp.getWriter().println("Privilege granted");
 189             } else {
 190                 u.revokeGroup(u, Group.getByString(req.getParameter("priv")));
 191                 resp.getWriter().println("Privilege revoked");
 192             }
 193         } else if (req.getParameter("fetch") != null) {
 194             String mail = req.getParameter("femail");
 195             fetchMails(req, resp, mail);
 196         } else if (req.getParameter("cats") != null) {
 197             String mail = req.getParameter("catsEmail");
 198             User byEmail = User.getByEmail(mail);
 199             if (byEmail == null) {
 200                 resp.getWriter().println("User not found.");
 201                 return;
 202             }
 203             passCATS(byEmail);
 204             resp.getWriter().println("User has been passed CATS");
 205         } else if (req.getParameter("assure") != null) {
 206             String mail = req.getParameter("assureEmail");
 207             User byEmail = User.getByEmail(mail);
 208             if (byEmail == null) {
 209                 resp.getWriter().println("User not found.");
 210                 return;
 211             }
 212             try {
 213                 for (int i = 0; i < getAssurers().length; i++) {
 214                     Notary.assure(getAssurers()[i], byEmail, byEmail.getName(), byEmail.getDoB(), 10, "Testmanager Assure up code", "2014-11-06");
 215                 }
 216             } catch (GigiApiException e) {
 217                 throw new Error(e);
 218             }
 219             resp.getWriter().println("User has been assured.");
 220         }
 221     }
 222
 223     private void fetchMails(HttpServletRequest req, HttpServletResponse resp, String mail) throws IOException {
 224         final LinkedList<String> mails = emails.get(mail);
 225         HashMap<String, Object> vars = new HashMap<>();
 226         vars.put("mail", mail);
 227         if (mails != null) {
 228             vars.put("mails", new IterableDataset() {
 229
 230                 Iterator<String> s = mails.iterator();
 231
 232                 @Override
 233                 public boolean next(Language l, Map<String, Object> vars) {
 234                     if ( !s.hasNext()) {
 235                         return false;
 236                     }
 237                     vars.put("body", s.next().replaceAll("(https?://\\S+)", "<a href=\"$1\">$1</a>"));
 238                     return true;
 239                 }
 240             });
 241         }
 242         t.output(resp.getWriter(), getLanguage(req), vars);
 243         if (mails == null) {
 244             resp.getWriter().println("No mails");
 245
 246         }
 247     }
 248
 249     @Override
 250     public boolean needsLogin() {
 251         return false;
 252     }
 253
 254     @Override
 255     public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
 256         getAssurers();
 257         String pi = req.getPathInfo().substring(PATH.length());
 258         if (pi.length() > 1 && pi.startsWith("/fetch-")) {
 259             String mail = pi.substring(pi.indexOf('-', 2) + 1);
 260             fetchMails(req, resp, mail);
 261             return;
 262         }
 263
 264         new ManagementForm(req).output(resp.getWriter(), getLanguage(req), new HashMap<String, Object>());
 265     }
 266 }