Coverage Report

 package org.paneris.bibliomania;
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 //import java.io.PrintWriter;
 
 import javax.servlet.http.HttpSession;
 
 import org.melati.Melati;
 import org.melati.servlet.Form;
 import org.melati.login.Login;
 import org.melati.login.LoginHandler;
 import org.melati.poem.AccessToken;
 import org.melati.poem.BaseFieldAttributes;
 import org.melati.poem.Field;
 import org.melati.poem.FieldAttributes;
 import org.melati.poem.PoemTask;
 import org.melati.poem.PoemThread;
 import org.melati.servlet.TemplateServlet;
 import org.melati.template.ServletTemplateContext;
 import org.melati.util.Email;
 import org.melati.util.UnexpectedExceptionException;
 import org.webmacro.Context;
 import org.webmacro.FastWriter;
 import org.webmacro.WebMacroException;
 import org.webmacro.servlet.WebContext;
 
 //import sun.net.smtp.SmtpClient;
 
 public class BibliomaniaLogin extends BibliomaniaServlet {
 
   /**
    * 
    */
   private static final long serialVersionUID = 1L;
 
   protected void fillContext(Melati melati, BibContext it) {
     it.setLogicalDatabase("bibliomania");
   }
 
   public String loginTemplate(String name) {
     return bibliomaniaTemplate("bibliomanialogin/" + name);
   }
 
   public static class Handler extends LoginHandler {
 
     public Handler(TemplateServlet s) {
       super(s);
     }
 
     public String getLogin(ServletTemplateContext context) {
       String it = context.getFormField("field_login");
       return it == null ? context.getFormField("field_email") : it;
     }
   
     protected String loginTemplate(String name) {
       return bibliomaniaTemplate("bibliomanialogin/" + name);
     }
 
     protected String usernameUnknownTemplate() {
       return loginTemplate("Login");
     }
 
     protected String passwordIncorrectTemplate() {
       return loginTemplate("Login");
     }
   }
 
   Handler loginHandler = null;
 
   protected Handler getLoginHandler() {
     if (loginHandler == null)
       loginHandler = new Handler(this);
 
     return loginHandler;
   }
 
   protected String handleRemind(Melati melati, WebContext context)
       throws Exception {
 
     getLoginHandler().setupContext(melati.getServletTemplateContext());
 
     final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
 
     UserTable users = (UserTable)db.getUserTable();
     String login = context.getForm("field_login");
     if (login == null)
       login = context.getForm("field_email");
 
     User u = (User)users.getLoginColumn().firstWhereEq(login);
     if (u == null)
       u = (User)users.getEmailColumn().firstWhereEq(login);
     final User user = u;
 
     if (user == null)
       context.put("loginUnknown", Boolean.TRUE);
     else {
       final String[] email = new String[1];
       ByteArrayOutputStream buf = new ByteArrayOutputStream();
       final FastWriter bufp = new FastWriter(db.getWebMacro().getBroker(),
                                              buf, "UTF8");
 
       PoemThread.withAccessToken(
           AccessToken.root,
           new PoemTask() {
             public void run() {
               email[0] = user.getEmail();
               if (email[0] == null)
                 return;
 
               email[0] = email[0].replace("\n\r", "  ");
               final org.webmacro.Template messageTemplate =
                   db.getPasswordReminderTemplate();
               final Context msgContext = db.getWebMacro().getContext();
 
               msgContext.put("db", db);
               msgContext.put("user", user);
 
               try {
                 messageTemplate.write(bufp.getOutputStream(), msgContext);
               }
               catch (WebMacroException e) {
                 throw new UnexpectedExceptionException(
                     "Error expanding password reminder message", e);
               }
               catch (IOException e) {
                 throw new UnexpectedExceptionException(
                     "Error expanding password reminder message", e);
               }
             }
           });
 
       if (email[0] == null)
         context.put("reminded", "(no address)");
       else {
         bufp.flush();
 
         String from = db.getPasswordReminderFrom().replace("\n\r", "  ");
 /*
         SmtpClient smtp = new SmtpClient(db.getSmtpServer());
         try {
           smtp.from(from);
           smtp.to(email[0]);
           PrintWriter data = new PrintWriter(smtp.startMessage());
           data.println(buf.toString().trim());
           data.flush();
         }
         finally {
           smtp.closeServer();
         }
 */
         Email.send(db.getSmtpServer(), from, email[0], null, "Your bibliomania.com password", buf.toString().trim());
         context.put("reminded", email[0]);
       }
     }
 
     return loginTemplate("Login");
   }
 
   protected String handleRegister(Melati melati, WebContext context,
                                   User template)
       throws Exception {
 
     final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
     UserTable userTable = (UserTable)db.getUserTable();
     ServletTemplateContext tc = melati.getServletTemplateContext();
     String email = context.getForm("field_email");
     if (email == null) email = context.getForm("field_login");
 
     context.put("email", new Field(email,
                 userTable.getEmailColumn()).withNullable(false));
     context.put("name", new Field(Form.getFieldNulled(tc,"field_name"),
                 new BaseFieldAttributes((FieldAttributes)userTable.getNameColumn(), true)));
     context.put("password", 
                 new Field(Form.getFieldNulled(tc,"field_password"),
                 userTable.getPasswordColumn()));
     context.put("age", new Field(Form.getIntegerField(tc,"field_age"),
                                  userTable.getAgeColumn()));
     context.put("sex", new Field(Form.getIntegerField(tc,"field_sex"),
                                  userTable.getSexColumn()));
     context.put("country",
                 new Field(
                     Form.getIntegerField(tc,"field_country"),
                     userTable.getCountryColumn()));
     context.put("fulltimeeducation",
                 new Field(Form.getBooleanField(
                               tc, "field_fulltimeeducation"),
                               userTable.getFulltimeeducationColumn()));
     context.put("wantemailalerts",
                 new Field(Form.getBooleanField(tc,
                                                     "field_wantemailalerts"),
                           userTable.getWantemailalertsColumn()));
     context.put("wantspam",
                 new Field(Form.getBooleanField(tc, "field_wantspam"),
                           userTable.getWantspamColumn()));
 
     if (template == null) {
       final User[] t = new User[1];
       PoemThread.withAccessToken(
           AccessToken.root,
           new PoemTask() {
             public void run() {
               t[0] = (User)db.getTemplateRegisterUser().duplicated();
               t[0].setName_unsafe(null);
             }
           });
       template = t[0];
     }
     
     return loginTemplate("Register");
   }
 
   protected String handleRegisterFinish(final Melati melati,
                                         final WebContext context)
       throws Exception {
     final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
     final UserTable userTable = (UserTable)db.getUserTable();
 
     String login = context.getForm("field_login");
     if (login == null)
       login = context.getForm("field_email");
 
     final User existing =
         (User)userTable.getLoginColumn().firstWhereEq(login);
 
     if (existing == null) {
       final User fromForm = (User)userTable.newPersistent();
       Form.extractFields(melati.getServletTemplateContext(), fromForm);
       fromForm.generateDefaults();
       String email = fromForm.getEmail().trim();
 /*
       if (fromForm.getName() == null || fromForm.getName().equals("")) {
         String name = email;
         int index = email.indexOf("@");
         if (index != -1) name = email.substring(0,index);
         fromForm.setName(name);
       }
 */
       if (email != null) fromForm.setLogin(email);
       PoemThread.withAccessToken(
           AccessToken.root,
           new PoemTask() {
             public void run() {
               userTable.create(fromForm);
             }
           });
           // gosh, it should be easier to do this!
       HttpSession session = context.getSession();
       session.removeAttribute(Login.TRIGGERING_REQUEST_PARAMETERS);
       session.removeAttribute(Login.TRIGGERING_EXCEPTION);
       context.remove("continuationURL");
       return getLoginHandler().loginSuccessfullyAs(melati, 
                                         melati.getServletTemplateContext(), fromForm);
     }
     else if (existing.getPassword_unsafe().equals(
                  context.getForm("field_password"))) {
       PoemThread.withAccessToken(
           existing,
           new PoemTask() {
             public void run() {
               Form.extractFields(melati.getServletTemplateContext(), existing);
             }
           });
 
       return getLoginHandler().loginSuccessfullyAs(melati, 
                                         melati.getServletTemplateContext(), existing);
     }
     else {
       User fromForm = (User)userTable.newPersistent();
       Form.extractFields(melati.getServletTemplateContext(), fromForm);
       context.put("loginInUse", Boolean.TRUE);
       return handleRegister(melati, context, fromForm);
     }
   }
 
   protected String bibliomaniaHandle(Melati melati, final WebContext context)
       throws Exception {
     String action = context.getForm("action");
 
     if (action == null) action = "";
 
     String login = context.getForm("field_login");
     if (login == null)
       login = context.getForm("field_email");
 
     if (action.equals("remind") && login != null)
       return handleRemind(melati, context);
     else if (action.equals("register"))
       return handleRegister(melati, context, null);
     else if (action.equals("create"))
       return handleRegisterFinish(melati, context);
     else
       return getLoginHandler().doTemplateRequest(melati,
                                                  melati.getServletTemplateContext());
   }
 }

1		package org.paneris.bibliomania;
2
3		import java.io.ByteArrayOutputStream;
4		import java.io.IOException;
5		//import java.io.PrintWriter;
6
7		import javax.servlet.http.HttpSession;
8
9		import org.melati.Melati;
10		import org.melati.servlet.Form;
11		import org.melati.login.Login;
12		import org.melati.login.LoginHandler;
13		import org.melati.poem.AccessToken;
14		import org.melati.poem.BaseFieldAttributes;
15		import org.melati.poem.Field;
16		import org.melati.poem.FieldAttributes;
17		import org.melati.poem.PoemTask;
18		import org.melati.poem.PoemThread;
19		import org.melati.servlet.TemplateServlet;
20		import org.melati.template.ServletTemplateContext;
21		import org.melati.util.Email;
22		import org.melati.util.UnexpectedExceptionException;
23		import org.webmacro.Context;
24		import org.webmacro.FastWriter;
25		import org.webmacro.WebMacroException;
26		import org.webmacro.servlet.WebContext;
27
28		//import sun.net.smtp.SmtpClient;
29
30	0	public class BibliomaniaLogin extends BibliomaniaServlet {
31
32		/**
33		*
34		*/
35		private static final long serialVersionUID = 1L;
36
37		protected void fillContext(Melati melati, BibContext it) {
38	0	it.setLogicalDatabase("bibliomania");
39	0	}
40
41		public String loginTemplate(String name) {
42	0	return bibliomaniaTemplate("bibliomanialogin/" + name);
43		}
44
45		public static class Handler extends LoginHandler {
46
47		public Handler(TemplateServlet s) {
48	0	super(s);
49	0	}
50
51		public String getLogin(ServletTemplateContext context) {
52	0	String it = context.getFormField("field_login");
53	0	return it == null ? context.getFormField("field_email") : it;
54		}
55
56		protected String loginTemplate(String name) {
57	0	return bibliomaniaTemplate("bibliomanialogin/" + name);
58		}
59
60		protected String usernameUnknownTemplate() {
61	0	return loginTemplate("Login");
62		}
63
64		protected String passwordIncorrectTemplate() {
65	0	return loginTemplate("Login");
66		}
67		}
68
69	0	Handler loginHandler = null;
70
71		protected Handler getLoginHandler() {
72	0	if (loginHandler == null)
73	0	loginHandler = new Handler(this);
74
75	0	return loginHandler;
76		}
77
78		protected String handleRemind(Melati melati, WebContext context)
79		throws Exception {
80
81	0	getLoginHandler().setupContext(melati.getServletTemplateContext());
82
83	0	final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
84
85	0	UserTable users = (UserTable)db.getUserTable();
86	0	String login = context.getForm("field_login");
87	0	if (login == null)
88	0	login = context.getForm("field_email");
89
90	0	User u = (User)users.getLoginColumn().firstWhereEq(login);
91	0	if (u == null)
92	0	u = (User)users.getEmailColumn().firstWhereEq(login);
93	0	final User user = u;
94
95	0	if (user == null)
96	0	context.put("loginUnknown", Boolean.TRUE);
97		else {
98	0	final String[] email = new String[1];
99	0	ByteArrayOutputStream buf = new ByteArrayOutputStream();
100	0	final FastWriter bufp = new FastWriter(db.getWebMacro().getBroker(),
101		buf, "UTF8");
102
103	0	PoemThread.withAccessToken(
104		AccessToken.root,
105	0	new PoemTask() {
106		public void run() {
107	0	email[0] = user.getEmail();
108	0	if (email[0] == null)
109	0	return;
110
111	0	email[0] = email[0].replace("\n\r", " ");
112	0	final org.webmacro.Template messageTemplate =
113		db.getPasswordReminderTemplate();
114	0	final Context msgContext = db.getWebMacro().getContext();
115
116	0	msgContext.put("db", db);
117	0	msgContext.put("user", user);
118
119		try {
120	0	messageTemplate.write(bufp.getOutputStream(), msgContext);
121		}
122	0	catch (WebMacroException e) {
123	0	throw new UnexpectedExceptionException(
124		"Error expanding password reminder message", e);
125		}
126	0	catch (IOException e) {
127	0	throw new UnexpectedExceptionException(
128		"Error expanding password reminder message", e);
129	0	}
130	0	}
131		});
132
133	0	if (email[0] == null)
134	0	context.put("reminded", "(no address)");
135		else {
136	0	bufp.flush();
137
138	0	String from = db.getPasswordReminderFrom().replace("\n\r", " ");
139		/*
140		SmtpClient smtp = new SmtpClient(db.getSmtpServer());
141		try {
142		smtp.from(from);
143		smtp.to(email[0]);
144		PrintWriter data = new PrintWriter(smtp.startMessage());
145		data.println(buf.toString().trim());
146		data.flush();
147		}
148		finally {
149		smtp.closeServer();
150		}
151		*/
152	0	Email.send(db.getSmtpServer(), from, email[0], null, "Your bibliomania.com password", buf.toString().trim());
153	0	context.put("reminded", email[0]);
154		}
155		}
156
157	0	return loginTemplate("Login");
158		}
159
160		protected String handleRegister(Melati melati, WebContext context,
161		User template)
162		throws Exception {
163
164	0	final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
165	0	UserTable userTable = (UserTable)db.getUserTable();
166	0	ServletTemplateContext tc = melati.getServletTemplateContext();
167	0	String email = context.getForm("field_email");
168	0	if (email == null) email = context.getForm("field_login");
169
170	0	context.put("email", new Field(email,
171		userTable.getEmailColumn()).withNullable(false));
172	0	context.put("name", new Field(Form.getFieldNulled(tc,"field_name"),
173		new BaseFieldAttributes((FieldAttributes)userTable.getNameColumn(), true)));
174	0	context.put("password",
175		new Field(Form.getFieldNulled(tc,"field_password"),
176		userTable.getPasswordColumn()));
177	0	context.put("age", new Field(Form.getIntegerField(tc,"field_age"),
178		userTable.getAgeColumn()));
179	0	context.put("sex", new Field(Form.getIntegerField(tc,"field_sex"),
180		userTable.getSexColumn()));
181	0	context.put("country",
182		new Field(
183		Form.getIntegerField(tc,"field_country"),
184		userTable.getCountryColumn()));
185	0	context.put("fulltimeeducation",
186		new Field(Form.getBooleanField(
187		tc, "field_fulltimeeducation"),
188		userTable.getFulltimeeducationColumn()));
189	0	context.put("wantemailalerts",
190		new Field(Form.getBooleanField(tc,
191		"field_wantemailalerts"),
192		userTable.getWantemailalertsColumn()));
193	0	context.put("wantspam",
194		new Field(Form.getBooleanField(tc, "field_wantspam"),
195		userTable.getWantspamColumn()));
196
197	0	if (template == null) {
198	0	final User[] t = new User[1];
199	0	PoemThread.withAccessToken(
200		AccessToken.root,
201	0	new PoemTask() {
202		public void run() {
203	0	t[0] = (User)db.getTemplateRegisterUser().duplicated();
204	0	t[0].setName_unsafe(null);
205	0	}
206		});
207	0	template = t[0];
208		}
209
210	0	return loginTemplate("Register");
211		}
212
213		protected String handleRegisterFinish(final Melati melati,
214		final WebContext context)
215		throws Exception {
216	0	final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();
217	0	final UserTable userTable = (UserTable)db.getUserTable();
218
219	0	String login = context.getForm("field_login");
220	0	if (login == null)
221	0	login = context.getForm("field_email");
222
223	0	final User existing =
224		(User)userTable.getLoginColumn().firstWhereEq(login);
225
226	0	if (existing == null) {
227	0	final User fromForm = (User)userTable.newPersistent();
228	0	Form.extractFields(melati.getServletTemplateContext(), fromForm);
229	0	fromForm.generateDefaults();
230	0	String email = fromForm.getEmail().trim();
231		/*
232		if (fromForm.getName() == null \|\| fromForm.getName().equals("")) {
233		String name = email;
234		int index = email.indexOf("@");
235		if (index != -1) name = email.substring(0,index);
236		fromForm.setName(name);
237		}
238		*/
239	0	if (email != null) fromForm.setLogin(email);
240	0	PoemThread.withAccessToken(
241		AccessToken.root,
242	0	new PoemTask() {
243		public void run() {
244	0	userTable.create(fromForm);
245	0	}
246		});
247		// gosh, it should be easier to do this!
248	0	HttpSession session = context.getSession();
249	0	session.removeAttribute(Login.TRIGGERING_REQUEST_PARAMETERS);
250	0	session.removeAttribute(Login.TRIGGERING_EXCEPTION);
251	0	context.remove("continuationURL");
252	0	return getLoginHandler().loginSuccessfullyAs(melati,
253		melati.getServletTemplateContext(), fromForm);
254		}
255	0	else if (existing.getPassword_unsafe().equals(
256		context.getForm("field_password"))) {
257	0	PoemThread.withAccessToken(
258		existing,
259	0	new PoemTask() {
260		public void run() {
261	0	Form.extractFields(melati.getServletTemplateContext(), existing);
262	0	}
263		});
264
265	0	return getLoginHandler().loginSuccessfullyAs(melati,
266		melati.getServletTemplateContext(), existing);
267		}
268		else {
269	0	User fromForm = (User)userTable.newPersistent();
270	0	Form.extractFields(melati.getServletTemplateContext(), fromForm);
271	0	context.put("loginInUse", Boolean.TRUE);
272	0	return handleRegister(melati, context, fromForm);
273		}
274		}
275
276		protected String bibliomaniaHandle(Melati melati, final WebContext context)
277		throws Exception {
278	0	String action = context.getForm("action");
279
280	0	if (action == null) action = "";
281
282	0	String login = context.getForm("field_login");
283	0	if (login == null)
284	0	login = context.getForm("field_email");
285
286	0	if (action.equals("remind") && login != null)
287	0	return handleRemind(melati, context);
288	0	else if (action.equals("register"))
289	0	return handleRegister(melati, context, null);
290	0	else if (action.equals("create"))
291	0	return handleRegisterFinish(melati, context);
292		else
293	0	return getLoginHandler().doTemplateRequest(melati,
294		melati.getServletTemplateContext());
295		}
296		}