package org.paneris.jammyjoes.servlet;

 import java.sql.Timestamp;

 import org.melati.Melati;

 import org.melati.servlet.Form;

 import org.melati.PoemContext;

 import org.melati.servlet.PathInfoException;

 import org.melati.servlet.TemplateServlet;

 import org.melati.template.ServletTemplateContext;

 import org.paneris.jammyjoes.model.Affiliate;

 import org.paneris.jammyjoes.model.AffiliateTransaction;

 import org.paneris.jammyjoes.model.AffiliateTransactionTable;

 import org.paneris.jammyjoes.model.JammyjoesDatabase;

 import org.paneris.jammyjoes.util.JammyJoesContextUtil;

 import org.paneris.jammyjoes.util.JammyJoesUtil;

 import org.webmacro.servlet.WebContext;

 public abstract class JammyJoesMelatiServlet extends TemplateServlet {

   protected PoemContext poemContext(Melati melati) throws PathInfoException {

     PoemContext context = new PoemContext();

     context.setLogicalDatabase("jammyjoes");

     return context;

   }

   public String getSysAdminName() {

     return "Tim Joyce";

   }

   public String getSysAdminEmail() {

     return "timj@hoop.co.uk";

   }

   protected String doTemplateRequest(Melati melati, ServletTemplateContext templateContext) throws Exception {

     logReferer(melati, templateContext);

     JammyjoesDatabase db = (JammyjoesDatabase) melati.getDatabase();

     templateContext.put("jjutil", new JammyJoesContextUtil());

     templateContext.put("menutypes", db.getTypeTable().selection());

     return jammyjoesRequest(melati, templateContext);

   }

   protected abstract String jammyjoesRequest(Melati melati, ServletTemplateContext webContext) throws Exception;

   protected String addExtension(String templateName) {

     return templateName;

   }

   private void logReferer(Melati melati, ServletTemplateContext context) {

     try {

       JammyjoesDatabase db = (JammyjoesDatabase) melati.getDatabase();

       WebContext webContext = (WebContext)context.getContext();

       String affiliateId = JammyJoesUtil.getCookieValue(melati, "affiliate");

       if (affiliateId == null) {

         affiliateId = Form.getFormNulled(context,"affiliate");

       }

       if (affiliateId == null) {

         return;

       }

       Affiliate affiliate = (Affiliate)db.getAffiliateTable().getObject(new Integer(affiliateId));

       int days = affiliate.getCookiedays().intValue();

       melati.getResponse().addCookie(

           JammyJoesUtil.makeCookie("affiliate", affiliate.getTroid().toString(), days));

       String referrer = webContext.getCGI().getHTTP_REFERER();

       if (isExcluded(referrer)) {

         return;

       }

       AffiliateTransactionTable transTable = db.getAffiliateTransactionTable();

       AffiliateTransaction trans = (AffiliateTransaction) transTable.newPersistent();

       trans.setAffiliate(affiliate);

       trans.setHit(new Timestamp(new java.util.Date().getTime()));

       trans.setReferrerURL(referrer);

       trans.setLandingPage(melati.getRequest().getRequestURL().toString());

       trans.setUserIpAddress(melati.getRequest().getRemoteAddr());

       transTable.create(trans);

     } catch (Exception e) {

       e.printStackTrace(System.err);

     }

   }

   private boolean isExcluded(String referrer) {

     if (referrer == null) {

       return false;

     }

     if (referrer.startsWith("http://localhost")) {

       return true;

     }

     if (referrer.startsWith("http://www.jammyjoes")) {

       return true;

     }

     return false;

   }

 }