package org.paneris.bibliomania.shopping;

 import java.io.PrintWriter;

 import java.io.StringWriter;

 import java.util.Enumeration;

 import java.util.Hashtable;

 import java.util.Locale;

 import org.melati.Melati;

 import org.melati.servlet.Form;

 import org.melati.poem.AccessToken;

 import org.melati.poem.Field;

 import org.melati.poem.PoemTask;

 import org.melati.poem.PoemThread;

 import org.melati.template.ServletTemplateContext;

 import org.melati.template.TemplateEngine;

 import org.melati.util.Email;

 import org.melati.util.InstantiationPropertyException;

 import org.melati.util.MelatiWriter;

 import org.paneris.bibliomania.BibliomaniaDatabase;

 import org.paneris.bibliomania.Country;

 import org.paneris.bibliomania.Currency;

 import org.paneris.bibliomania.DeliveryCharge;

 import org.paneris.bibliomania.NotLoggedInException;

 import org.paneris.bibliomania.ShopOrder;

 import org.paneris.bibliomania.ShopOrderItem;

 import org.paneris.bibliomania.User;

 import org.paneris.bibliomania.UserTable;

 import org.paneris.bibliomania.util.BibliomaniaUtil;

 import org.paneris.melati.shopping.MelatiShoppingConfig;

 import org.paneris.melati.shopping.ShoppingTrolley;

 import org.paneris.melati.shopping.ShoppingTrolleyItem;

 import org.webmacro.servlet.WebContext;

 /**

 * <p> A Shopping Trolley stored information in the user's Shopping Trolley.<p>

 * <p> It does this by storing itself in the session.</p>

 * <p> For this reason, the constructors are private, and you will be expected to

 * always get the Shopping Trolley using getInstance();</p>

 *

 * usage example:

 *

 * ShoppingTrolley trolley = ShoppingTrolley.getInstance(Melati melati);

 * context.put("trolley", trolley);

 *

 **/

 public class BibliomaniaShoppingTrolley extends ShoppingTrolley {

   private static String MY_TROLLEY = "org.paneris.bibliomania.shopping.BibliomaniaShoppingTrolley";

   private ShopOrder order = null;

   public Country countryCode;

  /**

   * Get the Locale for this trolley.

   */

   public Locale getLocale(){

     // attempt to get from user preferences

     if (locale == null) {

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

       User user = (User)melati.getUser();

       if (user.isRegisteredUser(db) && user.getCurrency() != null) {

         locale = user.getCurrency().toLocale();

       }

       // attempt to find from user's http headers

       if (locale == null) {

         locale = BibliomaniaUtil.makeLocale(

                               melati.getRequest().getHeader("accept-language"));

       }

     }

     return locale; 

   }

  /**

   * Public Constructor to build a trolley from some id.

   */

   public void initialise(Melati melatiP, MelatiShoppingConfig configP) {

     super.initialise(melatiP, configP);

   }

  /**

   * Load a trolley from something persistent.

   */

   public void load(Integer id) throws InstantiationPropertyException {

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

     order = db.getShopOrderTable().getShopOrderObject(id);

     setCountryCode(order.getCountry());

     setCountry(countryCode.getDisplayname());

     setName(order.getName());

     setEmail(order.getEmail());

     setTel(order.getTel());

     setDeliveryAddress(order.getAddress());

     setTown(order.getTown());

     setCounty(order.getCounty());

     setPostcode(order.getPostcode());

     setLocale(order.getCurrency().toLocale());

     Enumeration e = order.getItems();

     while (e.hasMoreElements()) {

       ShopOrderItem line = (ShopOrderItem)e.nextElement();

       ShoppingTrolleyItem item = newItem(line.getProduct().getTroid(), null, null);

       if (line.getQuantity() != null) {

         item.setQuantity(line.getQuantity().doubleValue());

       }

     }

   }

  /**

   * Save a trolley to something persistent.

   */

   public void save() {

     final BibliomaniaDatabase db = (BibliomaniaDatabase)melati.getDatabase();

     User user = (User)melati.getUser();

     if (user == null || !user.isRegisteredUser(db)) {

       user = (User)((UserTable)db.getUserTable()).getEmailColumn()

                                                  .firstWhereEq(email);

       if (user == null) {

         // we have new user, save their details for next time

         final User newUser;

         newUser = (User)db.getUserTable().newPersistent();

         newUser.setName(name);

         newUser.setEmail(email);

         newUser.setTel(tel);

         newUser.setAddress(address);

         newUser.setTown(town);

         newUser.setCounty(county);

         newUser.setRegion(postcode);

         newUser.setCountry(countryCode);

         newUser.generateDefaults();

         PoemThread.withAccessToken(

           AccessToken.root,

           new PoemTask() {

             public void run() {

               db.getUserTable().create(newUser);

             }

           }

         );

         user = newUser;

       }

     } else {

       // we have a genuinely logged in user, save their details for next time

       user.setName(name);

       user.setEmail(email);

       user.setTel(tel);

       user.setAddress(address);

       user.setTown(town);

       user.setCounty(county);

       user.setRegion(postcode);

       user.setCountry(countryCode);

     }

     if (order == null) {

       order = (ShopOrder)db.getShopOrderTable().newPersistent();

     } else {

       order.removeItems();

     }

     order.setUser(user);

     order.setName(name);

     order.setEmail(email);

     order.setTel(tel);

     order.setAddress(address);

     order.setTown(town);

     order.setCounty(county);

     order.setPostcode(postcode);

     order.setCountry(countryCode);

     order.setStatus(db.getOrderStatusTable().getNotAuthorised());

     order.setAmount(convertFromUK(getTotalValue()));

     order.setDelivery(convertFromUK(getTotalDeliveryValue()));

     order.setAmountUK(getTotalValue());

     order.setDeliveryUK(getTotalDeliveryValue());

     order.setCurrency(getCurrency());

     if (order.getTroid() == null) {

       db.getShopOrderTable().create(order);

     }

     for (Enumeration en = getItems(); en.hasMoreElements();) {

       BibliomaniaShoppingTrolleyItem item = (BibliomaniaShoppingTrolleyItem) en.nextElement();

       item.save(db, order);

     }

   }

  /**

   * Provide a mechanism for working out if 

   * this order should include a delivery charge.

   */

   public boolean hasDelivery(){

     if (countryCode == null) return false;

     return true;

   }

  /**

   * Can we deliver this - we need to check that each item has a supplier.

   */

   public boolean canSupply(){

     boolean canSupply = true;

     for (Enumeration en = getItems(); en.hasMoreElements();) {

       BibliomaniaShoppingTrolleyItem item = (BibliomaniaShoppingTrolleyItem) en.nextElement();

       if (item.getDeliveryCharge() == null) canSupply = false;

     }

     return canSupply;

   }

  /**

   *  Work out the cost of delivery.

   */

   public double getDeliveryValue() {

     double cost = 0;

     Enumeration f = getSuppliersCharges();

     while (f.hasMoreElements()) {

       cost += ((SupplierCharge)f.nextElement()).charge.getOrdercharge().doubleValue();

     }

     return cost;

   }

  /**

   *  Extract the list of supplier's charges.

   */

   public Enumeration getSuppliersCharges() {

     // get the cost for each supplier

     Hashtable charges = new Hashtable();

     Enumeration e = getItems();

     while (e.hasMoreElements()) {

       BibliomaniaShoppingTrolleyItem item = (BibliomaniaShoppingTrolleyItem)e.nextElement();

       DeliveryCharge charge = item.getDeliveryCharge();

       if (charge != null) {

         Integer troid = charge.getTroid();

         SupplierCharge sc = new SupplierCharge(charge,item);

         if (charges.containsKey(troid))

           sc.amount += ((SupplierCharge)charges.get(troid)).getAmount();

         charges.put(troid,sc);

       }

     }

     return charges.elements();

   }

  /**

   *  Provide a mechanism for working out if 

   *  this order should include a discount.

   */

   public boolean hasDiscount() {

     return false;

   }

  /**

   * If you want to apply a discount to this order, do it here.

   */

   public double getDiscountRate() {

     return 0;

   }

  /**

   *  Provide a mechanism for working out if 

   *  this order should include VAT (default should be true).

   */

   public boolean hasVAT() {

     return true;

   }

   public Field getCountries() {

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

     Integer current = null;

     if (countryCode != null) current = countryCode.getTroid();

     return new Field(current, db.getShopOrderTable().getCountryColumn());

   }

   public Country getCountryCode() {

     return countryCode;

   }

   public void setFromForm(Melati melati) {

     super.setFromForm(melati);

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

     Integer countryCodeId = 

          Form.getIntegerField(melati.getServletTemplateContext(),"field_country");

     Country newCode = (Country)db.getCountryTable()

                   .getIdColumn().firstWhereEq(countryCodeId);

     setCountryCode(newCode);

   }

   public void setCountryCode(Country newCode) {

     if (newCode != countryCode) {

       // need to unset charges so that they can be recalculated

       Enumeration e = getItems();

       while (e.hasMoreElements()) {

         BibliomaniaShoppingTrolleyItem line = (BibliomaniaShoppingTrolleyItem)e.nextElement();

         line.setGotCharge(false);

       }

     }

     countryCode = newCode;

   }

   public void setDefaultDetails(Melati melati) {

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

     User user = (User)melati.getUser();

     if (user != null && user.isRegisteredUser(db)) {

       if (address == null) address = user.getAddress();

       if (town == null) town = user.getTown();

       if (county == null) county = user.getCounty();

       if (name == null) name = user.getName();

       if (tel == null) tel = user.getTel();

       if (countryCode == null) setCountryCode(user.getCountry());

       if (postcode == null) postcode = user.getRegion();

       if (email == null) email = user.getEmail();

     }

   }

   public void assertLogin(Melati melatiP) {

     BibliomaniaDatabase db = (BibliomaniaDatabase)melatiP.getDatabase();

     if (!PoemThread.accessToken().givesCapability(

         db.getRegisteredUserCapability()))

       throw new NotLoggedInException();

   }

   public void confirmPayment(Melati melatiP) {

     BibliomaniaDatabase db = (BibliomaniaDatabase)melatiP.getDatabase();

     ServletTemplateContext context = melatiP.getServletTemplateContext();

     String amountString = context.getFormField("amount");

     String uniqueString = melatiP.getRequest().getQueryString();

     Integer orderInteger = Form.getIntegerField(context,"orderref");

     ServletTemplateContext templateContext = melatiP.getServletTemplateContext();

     boolean error = false;

     try {

       double amount = new Double(amountString).doubleValue();

       load(orderInteger);

       amount /= 100;

       templateContext.put("trolley",this);

       if (!uniqueString.equals("foo")) {

         error = true;

         context.put("error", "2");

       }

       if (getTotalValue() != amount) {

         error = true;

         context.put("expect", new Double(convertFromUK(getTotalValue())));

         context.put("paid", new Double(convertFromUK(amount)));

         context.put("error", "1");

       } 

       if (!error) {

         TemplateEngine templateEngine = melati.getTemplateEngine();

         templateContext.put("melati",melati);

         // email to bibliomaia

         MelatiWriter sw = templateEngine.getStringWriter();

         String templateName = "email/Bibliomania.wm";

         templateEngine.expandTemplate(sw, templateName, templateContext);

         String emailText = sw.toString();

         sw = templateEngine.getStringWriter();

         Email.send(db.getSettingTable().get(Email.SMTPSERVER), getEmail(), db.getOrderEmailTo(), "", 

                    "Bibliomania Order (number: " + orderInteger + ")", emailText);

         //email to customer

         sw = templateEngine.getStringWriter();

         templateName = "email/Customer.wm";

         templateEngine.expandTemplate(sw, templateName, templateContext);

         emailText = sw.toString();

         Email.send(db.getSettingTable().get(Email.SMTPSERVER), db.getOrderEmailFrom(), getEmail(), "", 

                    "Your order at Bibliomania.com (number: " + orderInteger + ")", emailText);

         //email to suppliers

         Enumeration f = getSuppliersCharges();

         while (f.hasMoreElements()) {

           SupplierCharge charge = (SupplierCharge)f.nextElement();

           templateContext.put("charge",charge);

           sw = templateEngine.getStringWriter();

           templateName = "email/Supplier.wm";

           templateEngine.expandTemplate(sw, templateName, templateContext);

           emailText = sw.toString();

           Email.send(db.getSettingTable().get(Email.SMTPSERVER), db.getOrderEmailFrom(), charge.getCharge().getSupplier().getEmail(), "", 

                      "Book orders via Bibliomania.com (number: " + orderInteger + ")", emailText);

         }

         order.setStatus(db.getOrderStatusTable().getAuthorised());

         Enumeration e = order.getItems();

         while (e.hasMoreElements()) {

           ShopOrderItem purchase = (ShopOrderItem)e.nextElement();

           purchase.setStatus(db.getOrderStatusTable().getAuthorised());

         }

         remove(melati);

       }

     } catch (Exception e) {

       error = true;

       StringWriter sw = new StringWriter();

       PrintWriter pw = new PrintWriter(sw);

       e.printStackTrace(pw);

       context.put("error", sw);

     }

   }

  /**

   * Return the name of the trolley (for storing in the session).

   */

   public static String name() {

     return MY_TROLLEY;

   }

   public ShopOrder getOrder() {

     return order;

   }

   public static double round(double num) {

     int a = Math.round(new Float(num * 100).floatValue());

     double b = new Double(a).doubleValue();

     return (b/100);

   }

  /**

   * Format a number in the locale currency.

   */

   public String convertFromUKandFormat(double value) {

     return getCurrency().convertFromUKandFormat(value);

   }

  /**

   * Convert from foreign currency back to sterling.

   */

   public double convertFromUK(double value) {

     return getCurrency().convertFromUK(value);

   }

  /**

   * Convert from foreign currency back to sterling.

   */

   public double convertToUK(double value) {

     return getCurrency().convertToUK(value);

   }

   public Currency getCurrency(Locale localeP) {

     return ((BibliomaniaDatabase) melati.getDatabase())

                               .getCurrencyTable().getCurrency(localeP);

   }

   public Currency getCurrency() {

     return getCurrency(getLocale());

   }

   public String displayCurrency(double value) {

     return convertFromUKandFormat(value);

   }

   public void configureRequest(Melati melatiP) {

     this.melati = melatiP;

     BibliomaniaDatabase db = (BibliomaniaDatabase)melatiP.getDatabase();

     db.setupContext(melatiP, (WebContext)melatiP.getTemplateContext().getContext(),

                     db.getShopSectionGroup());

   }  

   public class SupplierCharge {

     public DeliveryCharge charge;     

     public double amount;

     public SupplierCharge (DeliveryCharge c, BibliomaniaShoppingTrolleyItem item) {

       charge = c;

       amount = item.getValue();

     }

     public DeliveryCharge getCharge() {

       return charge;

     }

     public double getAmount() {

       return amount;

     }

   }

 }