Subversion Repositories SmartDukaan

/*******************************************************************************

 *  Copyright 2009 Amazon Services.

 *  Licensed under the Apache License, Version 2.0 (the "License"); 

 *  

 *  You may not use this file except in compliance with the License. 

 *  You may obtain a copy of the License at: http://aws.amazon.com/apache2.0

 *  This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 

 *  CONDITIONS OF ANY KIND, either express or implied. See the License for the 

 *  specific language governing permissions and limitations under the License.

 * ***************************************************************************** 

 *

 *  Marketplace Web Service Java Library

 *  API Version: 2009-01-01

 *  Generated: Wed Feb 18 13:28:48 PST 2009 

 * 

 */

package com.amazonaws.mws.samples;

import java.io.ByteArrayInputStream;

import in.shop2020.model.v1.catalog.Amazonlisted;

import in.shop2020.logistics.LogisticsInfo;

import in.shop2020.logistics.DeliveryType;

import in.shop2020.logistics.LogisticsServiceException;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.math.BigDecimal;

import java.math.BigInteger;

import java.nio.channels.FileChannel;

import java.security.DigestInputStream;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.util.Arrays;

import java.util.Calendar;

import java.util.Date;

import java.util.GregorianCalendar;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.ArrayList;

import java.util.Map;

import java.util.Random;

import java.util.TimeZone;

import javax.xml.bind.JAXBContext;

import javax.xml.bind.JAXBException;

import javax.xml.bind.Marshaller;

import javax.xml.datatype.DatatypeConfigurationException;

import javax.xml.datatype.DatatypeConstants;

import javax.xml.datatype.DatatypeFactory;

import javax.xml.datatype.XMLGregorianCalendar;

import com.amazonaws.mws.*;

import com.amazonaws.mws.model.*;

import com.amazonaws.mws.mock.MarketplaceWebServiceMock;

import org.apache.commons.codec.binary.*;

import in.shop2020.feeds.products.AmazonEnvelope;

import in.shop2020.feeds.products.BaseCurrencyCodeWithDefault;

import in.shop2020.feeds.products.CE;

import in.shop2020.feeds.products.HandheldOrPDA;

import in.shop2020.feeds.products.Header;

import in.shop2020.feeds.products.Inventory;

import in.shop2020.feeds.products.Listings;

import in.shop2020.feeds.products.OverrideCurrencyAmount;

import in.shop2020.feeds.products.Price;

import in.shop2020.feeds.products.Product;

import in.shop2020.feeds.products.StandardProductID;

import in.shop2020.model.v1.catalog.CatalogService.Client;

import in.shop2020.model.v1.catalog.CatalogServiceException;

import in.shop2020.model.v1.catalog.Item;

import in.shop2020.model.v1.catalog.status;

import in.shop2020.model.v1.inventory.AmazonInventorySnapshot;

import in.shop2020.model.v1.inventory.InventoryServiceException;

import in.shop2020.model.v1.inventory.ItemInventory;

import in.shop2020.model.v1.inventory.Warehouse;

import in.shop2020.model.v1.inventory.WarehouseType;

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.InventoryClient;

import in.shop2020.thrift.clients.LogisticsClient;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

/**

 * 

 * Submit Feed Samples

 * 

 * 

 */

public class SubmitNonFBAInventoryFeed {

    /**

     * Just add a few required parameters, and try the service Submit Feed

     * functionality

     * 

     * @param args

     *            unused

     *            

     */

    /**

     * @param args

     * @throws TException 

     * @throws CatalogServiceException 

     */

	final public static String AMAZON_NON_FBA_INVENTORY_XML = "/home/vikram/Nonfbainventory.xml";

	final public static boolean PURGE_OR_REPLACE   = false;

	public static  List<Amazonlisted> amazonNonFbaListedItems=null;

	public static List<in.shop2020.model.v1.catalog.Item> aliveItemsMap=null;

	public static Map<Long,AmazonItem> allItemsAmazonMap   = new HashMap<Long,AmazonItem>();

	static{

		CatalogClient catalogServiceClient = null;

		Map<Long,ItemInventory> availability= new HashMap<Long,ItemInventory>();

		try {

			catalogServiceClient = new CatalogClient();

		} catch (TTransportException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		Client catalogClient = catalogServiceClient.getClient();

		try {

			amazonNonFbaListedItems = catalogClient.getAllNonFbaListedItems();

			aliveItemsMap = catalogClient.getAllAliveItems();

			InventoryClient inventoryServiceClient = null;

			try {

				inventoryServiceClient = new InventoryClient();

			} catch (TTransportException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

			in.shop2020.model.v1.inventory.InventoryService.Client inventoryClient = inventoryServiceClient.getClient();

			availability = inventoryClient.getInventorySnapshot(0);

			List<Warehouse> warehouses = inventoryClient.getWarehouses(WarehouseType.OURS_THIRDPARTY,null,0,0,0);

			List<Long> thirdpartywarehouseids = new ArrayList<Long>();

			for(Warehouse warehouse:warehouses){

				thirdpartywarehouseids.add(warehouse.getId());

			}

			//System.out.println("Availability is " + availability );

			for(in.shop2020.model.v1.catalog.Item thriftItem:aliveItemsMap){

				long available=0;

				long reserve=0;

				AmazonItem item;

				if(availability.get(thriftItem.getId())!=null){ 

					ItemInventory iteminventory = availability.get(thriftItem.getId());

					Map<Long,Long> itemavailability = new HashMap<Long,Long>();

					itemavailability = iteminventory.getAvailability();

					for (Map.Entry<Long,Long> entry : itemavailability.entrySet()) {

						if(!thirdpartywarehouseids.contains(entry.getKey())){

							available =  available + entry.getValue();	

						}

					}

					Map<Long,Long> itemreserve = new HashMap<Long,Long>();

					itemreserve = iteminventory.getReserved();

					for (Map.Entry<Long,Long> entry : itemreserve.entrySet()) {

						if(!thirdpartywarehouseids.contains(entry.getKey())){

							reserve =  reserve + entry.getValue();

						}

					}

					item= new AmazonItem(thriftItem.getId(),available,reserve,thriftItem.getHoldInventory(),thriftItem.getDefaultInventory(),thriftItem.isRisky(),thriftItem.getItemStatus());

				}

				else{

					item = new AmazonItem(thriftItem.getId(),0,0,thriftItem.getHoldInventory(),thriftItem.getDefaultInventory(),thriftItem.isRisky(),thriftItem.getItemStatus());

				}

				//System.out.println(" Item details are " + thriftItem.getId() +" " + available + " " + reserve + " " + thriftItem.getHoldInventory() + " "+ thriftItem.getDefaultInventory() + " " + thriftItem.isRisky());

				//System.out.println("+++++++++++++++++++++++");

				allItemsAmazonMap.put(thriftItem.getId(),item);

			}

		} catch (TException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}

    public static void main(String... args) throws CatalogServiceException, TException {

        /**************************************************	**********************

         * Access Key ID and Secret Access Key ID, obtained from:

         * http://aws.amazon.com

         ***********************************************************************/

    	//public static final String marketplaceId = "A21TJRUUN4KGV";

        final String accessKeyId = "AKIAII3SGRXBJDPCHSGQ";

        final String secretAccessKey = "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";

        final String appName = "Test";

        final String appVersion = "1.0";

        MarketplaceWebServiceConfig config = new MarketplaceWebServiceConfig();

        /************************************************************************

         * Uncomment to set the appropriate MWS endpoint.

         ************************************************************************/

        // US

        // config.setServiceURL("https://mws.amazonservices.com");

        // UK

        // config.setServiceURL("https://mws.amazonservices.co.uk");

        // Germany

        // config.setServiceURL("https://mws.amazonservices.de");

        // France

        // config.setServiceURL("https://mws.amazonservices.fr");

        // Italy

        // config.setServiceURL("https://mws.amazonservices.it");

        // Japan

        // config.setServiceURL("https://mws.amazonservices.jp");

        // China

        // config.setServiceURL("https://mws.amazonservices.com.cn");

        // Canada

        // config.setServiceURL("https://mws.amazonservices.ca");

        // India

        config.setServiceURL("https://mws.amazonservices.in");

        /************************************************************************

         * You can also try advanced configuration options. Available options are:

         *

         *  - Signature Version

         *  - Proxy Host and Proxy Port

         *  - User Agent String to be sent to Marketplace Web Service

         *

         ***********************************************************************/

        /************************************************************************

         * Instantiate Http Client Implementation of Marketplace Web Service        

         ***********************************************************************/

        MarketplaceWebService service = new MarketplaceWebServiceClient(

                accessKeyId, secretAccessKey, appName, appVersion, config);

        /************************************************************************

         * Setup request parameters and uncomment invoke to try out sample for

         * Submit Feed

         ************************************

         ************************************/

        /************************************************************************

         * Marketplace and Merchant IDs are required parameters for all

         * Marketplace Web Service calls.

         ***********************************************************************/

        //final String merchantId = "AF6E300VE0X4D";

        final String merchantId = "AF6E3O0VE0X4D";	

        // marketplaces to which this feed will be submitted; look at the

        // API reference document on the MWS website to see which marketplaces are

        // included if you do not specify the list yourself

        final IdList marketplaces = new IdList(Arrays.asList(

        		"A21TJRUUN4KGV"));

        SubmitFeedRequest requestinventory = new SubmitFeedRequest();

        requestinventory.setMerchant(merchantId);

        requestinventory.setMarketplaceIdList(marketplaces);

        requestinventory.setFeedType("_POST_INVENTORY_AVAILABILITY_DATA_");

        // MWS exclusively offers a streaming interface for uploading your

        // feeds. This is because

        // feed sizes can grow to the 1GB+ range - and as your business grows

        // you could otherwise

        // silently reach the feed size where your in-memory solution will no

        // longer work, leaving you

        // puzzled as to why a solution that worked for a long time suddenly

        // stopped working though

        // you made no changes. For the same reason, we strongly encourage you

        // to generate your feeds to

        // local disk then upload them directly from disk to MWS via Java -

        // without buffering them in JavaFileNotFoundException

        // memory in their entirety.

        // Note: MarketplaceWebServiceClient will not retry a submit feed request

        // because there is no way to reset the InputStream from our client. 

        // To enable retry, recreate the InputStream and resubmit the feed

        // with the new InputStream. 

        //

        try {

        	createNonFbaInventoryFeed();

        	FileInputStream inventoryfis = new FileInputStream(AMAZON_NON_FBA_INVENTORY_XML);

        	requestinventory.setContentMD5(computeContentMD5HeaderValue(inventoryfis));

			requestinventory.setFeedContent(inventoryfis);

		} catch (FileNotFoundException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		try {

				invokeSubmitFeed(service,requestinventory);

				Thread.sleep(30);

			} catch (InterruptedException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

    }

    /**

     * Submit Feed request sample Uploads a file for processing together with

     * the necessary metadata to process the file, such as which type of feed it

     * is. PurgeAndReplace if true means that your existing e.g. inventory is

     * wiped out and replace with the contents ocf this feed - use with caution

     * (the default is false).

     * 

     * @param service

     *            instance of MarketplaceWebService service

     * @param request

     *            Action to invoke

     */

    public static void invokeSubmitFeed(MarketplaceWebService service,

            SubmitFeedRequest request) {

        try {

            SubmitFeedResponse response = service.submitFeed(request);

            System.out.println("SubmitFeed Action Response");

            System.out

            .println("=============================================================================");

            System.out.println();

            System.out.print("    SubmitFeedResponse");

            System.out.println();

            if (response.isSetSubmitFeedResult()) {

                System.out.print("        SubmitFeedResult");

                System.out.println();

                SubmitFeedResult submitFeedResult = response

                .getSubmitFeedResult();

                if (submitFeedResult.isSetFeedSubmissionInfo()) {

                    System.out.print("            FeedSubmissionInfo");

                    System.out.println();

                    FeedSubmissionInfo feedSubmissionInfo = submitFeedResult

                    .getFeedSubmissionInfo();

                    if (feedSubmissionInfo.isSetFeedSubmissionId()) {

                        System.out.print("                FeedSubmissionId");

                        System.out.println();

                        System.out.print("                    "

                                + feedSubmissionInfo.getFeedSubmissionId());

                        System.out.println();

                    }

                    if (feedSubmissionInfo.isSetFeedType()) {

                        System.out.print("                FeedType");

                        System.out.println();

                        System.out.print("                    "

                                + feedSubmissionInfo.getFeedType());

                        System.out.println();

                    }

                    if (feedSubmissionInfo.isSetSubmittedDate()) {

                        System.out.print("                SubmittedDate");

                        System.out.println();

                        System.out.print("                    "

                                + feedSubmissionInfo.getSubmittedDate());

                        System.out.println();

                    }

                    if (feedSubmissionInfo.isSetFeedProcessingStatus()) {

                        System.out

                        .print("                FeedProcessingStatus");

                        System.out.println();

                        System.out.print("                    "

                                + feedSubmissionInfo.getFeedProcessingStatus());

                        System.out.println();

                    }

                    if (feedSubmissionInfo.isSetStartedProcessingDate()) {

                        System.out

                        .print("                StartedProcessingDate");

                        System.out.println();

                        System.out

                        .print("                    "

                                + feedSubmissionInfo

                                .getStartedProcessingDate());

                        System.out.println();

                    }

                    if (feedSubmissionInfo.isSetCompletedProcessingDate()) {

                        System.out

                        .print("                CompletedProcessingDate");

                        System.out.println();

                        System.out.print("                    "

                                + feedSubmissionInfo

                                .getCompletedProcessingDate());

                        System.out.println();

                    }

                }

            }

            if (response.isSetResponseMetadata()) {

                System.out.print("        ResponseMetadata");

                System.out.println();

                ResponseMetadata responseMetadata = response

                .getResponseMetadata();

                if (responseMetadata.isSetRequestId()) {

                    System.out.print("            RequestId");

                    System.out.println();

                    System.out.print("                "

                            + responseMetadata.getRequestId());

                    System.out.println();

                }

            }

            System.out.println(response.getResponseHeaderMetadata());

            System.out.println();

            System.out.println();

        } catch (MarketplaceWebServiceException ex) {

            System.out.println("Caught Exception: " + ex.getMessage());

            System.out.println("Response Status Code: " + ex.getStatusCode());

            System.out.println("Error Code: " + ex.getErrorCode());

            System.out.println("Error Type: " + ex.getErrorType());

            System.out.println("Request ID: " + ex.getRequestId());

            System.out.print("XML: " + ex.getXML());

            System.out.println("ResponseHeaderMetadata: " + ex.getResponseHeaderMetadata());

        }

    }

    public static void createNonFbaInventoryFeed() throws CatalogServiceException, TException{

    	File file = new File(AMAZON_NON_FBA_INVENTORY_XML);

    	AmazonEnvelope amazonenvelope =  new AmazonEnvelope();

    	Header header = new Header();

    	header.setDocumentVersion("1.0");

    	header.setMerchantIdentifier("AF6E3O0VE0X4D");

    	header.setOverrideReleaseId(null);

    	amazonenvelope.setHeader(header);

    	amazonenvelope.setMessageType("Inventory");

    	amazonenvelope.setMarketplaceName(null);

    	//amazonenvelope.setPurgeAndReplace(PURGE_OR_REPLACE);

    	amazonenvelope.setPurgeAndReplace(true);

    	JAXBContext context;

    	Marshaller m=null;

    	try {

    		context = JAXBContext.newInstance(AmazonEnvelope.class);

    		m = context.createMarshaller();

    	} catch (JAXBException e1) {

    		// TODO Auto-generated catch block

    		e1.printStackTrace();

    	}

    	XMLGregorianCalendar today = null;

    	XMLGregorianCalendar restockdate = null;

    	//Calendar localCalendar = Calendar.getInstance();

    	Calendar localCalendar = Calendar.getInstance(TimeZone.getDefault());

    	int currentMonth = localCalendar.get(Calendar.MONTH);

    	int currentYear = localCalendar.get(Calendar.YEAR);

    	int currentDayOfMonth = localCalendar.get(Calendar.DAY_OF_MONTH);

    	try {

    		today = DatatypeFactory.newInstance()

    		.newXMLGregorianCalendar(

    				new GregorianCalendar(currentYear,currentMonth,currentDayOfMonth));

    		restockdate = DatatypeFactory.newInstance()

    		.newXMLGregorianCalendar(

    				new GregorianCalendar(currentYear,currentMonth,currentDayOfMonth));

    		restockdate.setTimezone(DatatypeConstants.FIELD_UNDEFINED);

    		restockdate.setTime(DatatypeConstants.FIELD_UNDEFINED, DatatypeConstants.FIELD_UNDEFINED, DatatypeConstants.FIELD_UNDEFINED);  

    	} catch (DatatypeConfigurationException e) {

    		// TODO Auto-generated catch block

    		e.printStackTrace();

    	}

    	amazonenvelope.setEffectiveDate(today);

    	LogisticsClient logisticsServiceClient;

		in.shop2020.logistics.LogisticsService.Client logisticsClient=null;

    	try {

    		logisticsServiceClient = new LogisticsClient("logistics_service_prod_server_host","logistics_service_prod_server_port");

    		logisticsClient = logisticsServiceClient.getClient();

		} catch (TTransportException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

    	List <AmazonEnvelope.Message> messages = amazonenvelope.getMessage();

    	InventoryClient inventoryServiceClient = new InventoryClient();

    	in.shop2020.model.v1.inventory.InventoryService.Client inventoryClient = inventoryServiceClient.getClient();

    	long time = System.currentTimeMillis();

    	for(Amazonlisted AItem:amazonNonFbaListedItems){

    		if(!AItem.isIsInventoryOverride()){

    			AmazonEnvelope.Message message = new AmazonEnvelope.Message();	

    			Inventory inventory = new Inventory();

    			AmazonItem item = allItemsAmazonMap.get(AItem.getItemid());

    			if(item!=null && !item.getStatus().equals(status.PAUSED)){

    				long actualavailibility = (item.getAvailability() - item.getReserve());

    				long holdinventory = item.getHoldInventory();

    				long defaultinventory = item.getDefaultInventory();

    				if(item.getRisky()){

    					if(actualavailibility > holdinventory){

    						inventory.setQuantity(BigInteger.valueOf(actualavailibility - holdinventory));

    					}

    					else{

    						inventory.setQuantity(BigInteger.valueOf(0));

    					}

    				}		

    				else{

    					if(actualavailibility > defaultinventory){

    						inventory.setQuantity(BigInteger.valueOf(actualavailibility));

    					}

    					else{

    						inventory.setQuantity(BigInteger.valueOf(defaultinventory));

    					}

    				}

    			}

    			else{

    				inventory.setQuantity(BigInteger.valueOf(0));

    			}

    			AmazonInventorySnapshot amazonInventorySnapshot = new AmazonInventorySnapshot();

			    amazonInventorySnapshot.setAvailability(inventory.getQuantity().longValue());

        		amazonInventorySnapshot.setItem_id(AItem.getItemid());

    			inventoryClient.addOrUpdateAmazonInventoryForItem(amazonInventorySnapshot,time);

    			inventory.setSKU(Long.toString(AItem.getItemid()));

    			LogisticsInfo logisticinfo=new LogisticsInfo();

    			try {

				     logisticinfo = logisticsClient.getLogisticsEstimation(AItem.getItemid(),"110001",DeliveryType.COD);

				} catch (LogisticsServiceException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

				if(AItem.isIsCustomTime()){

					inventory.setFulfillmentLatency(BigInteger.valueOf(AItem.getHandlingTime()));

				}

				else{

					inventory.setFulfillmentLatency(BigInteger.valueOf(logisticinfo.getShippingTime()));

				}

				if(inventory.getFulfillmentLatency().intValue() <= 0){

					inventory.setFulfillmentLatency(BigInteger.valueOf(1));

				}

    			inventory.setSwitchFulfillmentTo("MFN");

    			Random rand = new Random();

    			message.setMessageID(BigInteger.valueOf(Math.abs(rand.nextInt())));

    			message.setInventory(inventory);

    			message.setOperationType("Update");

    			messages.add(message);

    		}

    	}

    	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();

    	}

    }

    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;

 }

}