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;
     }
   }
   
 }