Subversion Repositories SmartDukaan

package com.amazonaws.mws.samples;

import in.shop2020.feeds.products.AmazonEnvelope;
import in.shop2020.feeds.products.AmazonEnvelope.Message;
import in.shop2020.feeds.products.Header;
import in.shop2020.feeds.products.OrderAcknowledgement;
import in.shop2020.model.v1.order.AmazonOrdersToAcknowledge;
import in.shop2020.model.v1.order.OrderSource;
import in.shop2020.thrift.clients.TransactionClient;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.math.BigInteger;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;

import org.apache.commons.codec.binary.Base64;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;


public class CreateOrderAcknowledgementFeed {
        static java.text.SimpleDateFormat sdf = new java.text.SimpleDateFormat("yyyy-MM-dd-HH:mm:ss");
        static Calendar cal=GregorianCalendar.getInstance();
        private static final String AMAZON_ORDER_ACKNOWLEDGEMENT_XML = "/home/amazon/OrderAck/OrderAcknowledgement"+sdf.format(cal.getTime())+".xml";

        public static void main(String... args) {
                createOrderAcknowlegementFeed();
        }


        public static void createOrderAcknowlegementFeed() {
                TransactionClient tcl=null;
                List<AmazonOrdersToAcknowledge> amazonOrdersToAcknowledge=null;
                try {
                        tcl = new TransactionClient();
                        amazonOrdersToAcknowledge = tcl.getClient().getAmazonOrdersToAcknowledge(OrderSource.AMAZON);
                } catch (TTransportException e2) {
                        // TODO Auto-generated catch block
                        e2.printStackTrace();
                } catch (TException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                if(amazonOrdersToAcknowledge!=null && amazonOrdersToAcknowledge.size() > 0){
                        File file = new File(AMAZON_ORDER_ACKNOWLEDGEMENT_XML);
                        AmazonEnvelope amazonenvelope = new AmazonEnvelope();
                        Header header = new Header();
                        header.setDocumentVersion("1.0");
                        header.setMerchantIdentifier("AF6E3O0VE0X4D");
                        header.setOverrideReleaseId(null);
                        amazonenvelope.setHeader(header);
                        amazonenvelope.setMessageType("OrderAcknowledgement");
                        amazonenvelope.setMarketplaceName(null);
                        List<Message> messages = amazonenvelope.getMessage();
                        Map<String,Long> amazonOrdertxnMap = new HashMap<String,Long>();
                        Map<String,List<String>> amazonOrderIdOrderItemMap = new HashMap<String,List<String>>();
                        Map<String,Integer> amazonOrderItemSuccessMap = new HashMap<String,Integer>();
                        Map<String,Integer> amazonOrderItemFailureMap = new HashMap<String,Integer>();
                        for(AmazonOrdersToAcknowledge amazonOrderToAcknowledge:amazonOrdersToAcknowledge ){
                                if(!amazonOrdertxnMap.containsKey(amazonOrderToAcknowledge.getAmazonOrderId())){
                                        amazonOrdertxnMap.put(amazonOrderToAcknowledge.getAmazonOrderId(),amazonOrderToAcknowledge.getTransaction_id());
                                }
                                if(!amazonOrderIdOrderItemMap.containsKey(amazonOrderToAcknowledge.getAmazonOrderId())){
                                        List<String> orderItemsList = new ArrayList<String>();
                                        orderItemsList.add(amazonOrderToAcknowledge.getAmazonOrderItemId());
                                        amazonOrderIdOrderItemMap.put(amazonOrderToAcknowledge.getAmazonOrderId(),orderItemsList);
                                }
                                else{
                                        if(!amazonOrderIdOrderItemMap.get(amazonOrderToAcknowledge.getAmazonOrderId()).contains(amazonOrderToAcknowledge.getAmazonOrderItemId()))
                                                amazonOrderIdOrderItemMap.get(amazonOrderToAcknowledge.getAmazonOrderId()).add(amazonOrderToAcknowledge.getAmazonOrderItemId());
                                }
                                if(amazonOrderToAcknowledge.getStatus().equals("Order-Payment-Success")){
                                        amazonOrderItemSuccessMap.put(amazonOrderToAcknowledge.getAmazonOrderId()+amazonOrderToAcknowledge.getAmazonOrderItemId(),amazonOrderToAcknowledge.getCount());
                                }
                                if(amazonOrderToAcknowledge.getStatus().equals("Order-Payment-Failure")){
                                        amazonOrderItemFailureMap.put(amazonOrderToAcknowledge.getAmazonOrderId()+amazonOrderToAcknowledge.getAmazonOrderItemId(),amazonOrderToAcknowledge.getCount());
                                }       
                        }
                        JAXBContext context;
                        Marshaller m=null;
                        try {
                                context = JAXBContext.newInstance(AmazonEnvelope.class);
                                m = context.createMarshaller();
                        } catch (JAXBException e1) {
                                // TODO Auto-generated catch block
                                e1.printStackTrace();
                        }

                        for (Map.Entry<String,List<String>> entry : amazonOrderIdOrderItemMap.entrySet()){
                                OrderAcknowledgement orderAcknowLedgement = new OrderAcknowledgement();
                                List<OrderAcknowledgement.Item> acknowledgementItems = orderAcknowLedgement.getItem();
                                String orderid = entry.getKey();
                                orderAcknowLedgement.setAmazonOrderID(orderid);
                                orderAcknowLedgement.setMerchantOrderID(amazonOrdertxnMap.get(orderid).toString());
                                orderAcknowLedgement.setStatusCode("Failure");
                                for(String orderItemId :entry.getValue()){
                                        OrderAcknowledgement.Item orderAcknowledgementItem = new OrderAcknowledgement.Item();
                                        if(amazonOrderItemSuccessMap.get(orderid+orderItemId)!=null && amazonOrderItemSuccessMap.get(orderid+orderItemId) > 0){
                                                orderAcknowledgementItem.setAmazonOrderItemCode(orderItemId);
                                                orderAcknowLedgement.setStatusCode("Success");
                                        }
                                        else{
                                                orderAcknowledgementItem.setAmazonOrderItemCode(orderItemId);
                                                orderAcknowledgementItem.setCancelReason("ShippingAddressUndeliverable");
                                        }
                                        acknowledgementItems.add(orderAcknowledgementItem);

                                }
                                AmazonEnvelope.Message message = new AmazonEnvelope.Message();
                                Random rand = new Random();
                                message.setMessageID(BigInteger.valueOf(Math.abs(rand.nextInt())));
                                message.setOrderAcknowledgement(orderAcknowLedgement);
                                messages.add(message);
                        }

                        System.out.println("Messages is " + messages);
                        try {
                                m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
                                m.marshal(amazonenvelope, file);
                                m.marshal(amazonenvelope, System.out);

                        } catch (Exception e) {
                                // TODO Auto-generated catch block
                                e.printStackTrace();

                        }
                                //                      GmailUtils  gm = GmailUtils();
                        //                      gm.sendSSLMessage(arg0, arg1, arg2, arg3, arg4, arg5)
                }


        }

                
        public static String computeContentMD5HeaderValue( FileInputStream fis) {
                try {
                        DigestInputStream dis = new DigestInputStream( fis,
                                        MessageDigest.getInstance( "MD5" ));
                        byte[] buffer = new byte[8192];
                        while( dis.read( buffer ) > 0 );
                        String md5Content = new String(Base64.encodeBase64(dis.getMessageDigest().digest())
                        );
                        // Effectively resets the stream to be beginning of the file via a
                        fis.getChannel().position( 0 );
                        return md5Content;
                } catch (NoSuchAlgorithmException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                return null;

        }

        
        
        
}