Subversion Repositories SmartDukaan

package in.shop2020.util;

import java.io.BufferedOutputStream;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.StringTokenizer;

import in.shop2020.metamodel.core.Entity;

import in.shop2020.metamodel.core.FreeformContent;

import in.shop2020.metamodel.core.Media;

import in.shop2020.metamodel.core.Slide;

import in.shop2020.metamodel.definitions.Catalog;

import in.shop2020.metamodel.definitions.DefinitionsContainer;

import in.shop2020.metamodel.definitions.EntityContainer;

import in.shop2020.metamodel.util.CreationUtils;

//import in.shop2020.metamodel.util.OldCreationUtils;

/*

 * @author rajveer 

 * 

 *

 */

public class ContentMigrator {

	private String sourceDbPath;

	private String destinationDbPath;

	public static void main(String[] args) throws Exception {

		String usage = "Usage: ContentMigrator {source db path}{destination db path}";

		String sourceDbPath = null, destinationDbPath = null;

		if(args.length < 2) {

			System.out.println(usage);

			System.exit(-1);

		}else{

			sourceDbPath = args[0];

			destinationDbPath = args[1];

		}

		ContentMigrator contentmigrator = new ContentMigrator(sourceDbPath, destinationDbPath);

		boolean result = contentmigrator.migrate();

		System.out.println("Content migration status:  "+ result);

	}

	public ContentMigrator(String sourceDbPath, String destinationDbPath) {

		this.sourceDbPath = sourceDbPath;

		this.destinationDbPath = destinationDbPath;

	}

	/**

	 * this function will read source definition and source entities, and will convert

	 * source entities to destination entities according to destination definitions.  

	 * 

	 * @return boolean

	 * @throws Exception 

	 */

	public boolean migrate() throws Exception{

		/*

		 * 

		DefinitionsContainer sourceDefs = new DefinitionsContainer(sourceDbPath);

		DefinitionsContainer destinationDefs = new DefinitionsContainer(destinationDbPath);

		EntityContainer sourceEnts = new EntityContainer(sourceDbPath);

		EntityContainer destinationEnts = new EntityContainer(destinationDbPath);

		Map<Long, Entity> entities = sourceEnts.getEntities();

		File f = new File("/home/rajveer/Desktop/info.txt");

		FileWriter fstream = new FileWriter(f);

        BufferedWriter out = new BufferedWriter(fstream);

        StringBuilder sb = new StringBuilder();

		for(Entity entity: entities.values()){

			List<Slide> slides = entity.getSlides();

			//List<String> vidLabels = CreationUtils.getMediaLabels(entity.getID(),"youtube");

			//List<String> imgLabels = CreationUtils.getMediaLabels(entity.getID(),"image");

			//Map<String, Media> rawMedia = CreationUtils.getRawMedia(entity.getID());

//			List<String> vidLabels = OldCreationUtils.getMediaLabels(entity.getID(),"youtube");

//			List<String> imgLabels = OldCreationUtils.getMediaLabels(entity.getID(),"image");

			Entity newEntity = new Entity(entity.getID(), entity.getCategoryID());

			newEntity.setBrand(entity.getBrand());

			newEntity.setModelName(entity.getModelName());

			newEntity.setModelNumber(entity.getModelNumber());

			Map<String, in.shop2020.creation.util.Media> rawMedia = OldCreationUtils.getRawMedia(entity.getID());

			Map<String, Media> newrawMedia = new HashMap<String, Media>();

			int totalRawVideos = 0;

			int totalRawImages = 0;

			if(rawMedia!=null){

				for(String label: rawMedia.keySet()){

					in.shop2020.creation.util.Media media = rawMedia.get(label);

					label = media.getLabel();

					String type = media.getType();

					String location = media.getLocation();

					Media newMedia = new Media(label, type, location);

					newMedia.setTitle("Title");

					if(type.equals("image")){

						totalRawImages++;

						newMedia.setFileName(media.getFileName());

						newMedia.setVideoType("");

					}else{

						totalRawVideos++;

						newMedia.setVideoType("withoutskin");

						newMedia.setFileName("");

					}

					newrawMedia.put(label, newMedia);

					System.out.println("old " + media.toString());

	//				System.out.println("new " + newMedia.toString());

	//				OldCreationUtils.removeMedia(entity.getID(), label);

					CreationUtils.addMedia(entity.getID(), newMedia);

				}

			}

			int totalVideos = 0;

			int totalImages = 0;

			for(Slide slide: slides ){

				FreeformContent ffc = slide.getFreeformContent();

				List<String> vidList = new ArrayList<String>();

				List<String> imgList = new ArrayList<String>();

				if(ffc!=null){

					List<String> vidFfcLabels = ffc.getYoutubeRefs();

					if(vidFfcLabels!=null){

						for(String vidFfcLabel: vidFfcLabels){

							StringTokenizer strTkn = new StringTokenizer(vidFfcLabel,"~!~");

							String vidLabel = null;

							if(strTkn.hasMoreTokens()){

								vidLabel = strTkn.nextToken();

								vidList.add(vidFfcLabel);

//								ffc.removeMedia("youtube", vidFfcLabel);

//								ffc.addMedia("youtube", vidLabel);

							}

							if(vidLabel == null){

								System.out.println("Something is really breaking .. bu ha ha !!!");

							}

							totalVideos++;

						}

					}

					List<String> imgFfcLabels = ffc.getImageRefs();

					if(imgFfcLabels!=null){

						for(String imgFfcLabel: imgFfcLabels){

							StringTokenizer strTkn = new StringTokenizer(imgFfcLabel,"~!~");

							String imgLabel = null;

							if(strTkn.hasMoreTokens()){

								imgLabel = strTkn.nextToken();

								imgList.add(imgFfcLabel);

//								ffc.removeMedia("image", imgFfcLabel);

//								ffc.addMedia("image", imgLabel);							

							}

							if(imgLabel == null){

								System.out.println("Something is really breaking .. bu ha ha !!!");

							}

							totalImages++;

						}

					}

					for(String vidlab : vidList){

						ffc.removeMedia("youtube", vidlab);

						ffc.addMedia("youtube", (new StringTokenizer(vidlab, "~!~")).nextToken());

					}

					for(String imglab : imgList){

						ffc.removeMedia("image",imglab);

						ffc.addMedia("image", (new StringTokenizer(imglab, "~!~")).nextToken());

					}

					slide.setFreeformContent(ffc);

					System.out.println("Slide id is " + slide.getSlideDefinitionID() + "   vid labels:  " + vidFfcLabels + "   img labels" + imgFfcLabels);

				}

				newEntity.addSlide(slide);

			}

			System.out.println("entity:"+ entity.getID() + ":trv:" + totalRawVideos + ":tri:" + totalRawImages +":tv:" + totalVideos +":ti:" + totalImages);

			sb.append("entity:"+ entity.getID() + ":trv:" + totalRawVideos + ":tri:" + totalRawImages +":tv:" + totalVideos +":ti:" + totalImages+ ":tr:" +(totalRawVideos+totalRawImages) + ":t:" +(totalVideos+totalImages) + ":vd:" +(totalRawVideos-totalVideos) + ":id:" +(totalRawImages-totalImages) + "\n");

			destinationEnts.updateEntity(newEntity);

			CreationUtils.storeEntity(newEntity);

			Map<Long, List<Entity>> entitiesByCategory = 

				sourceEnts.getEntitiesbyCategory();

			Map<Long, Entity> xentities = sourceEnts.getEntities();

			CreationUtils.rewriteRepository(xentities, entitiesByCategory);

		}

		out.write(sb.toString());

		out.close();

		*/

		return true;

	}

}