GithubConfigurationProvider.java

/* license-start
 * 
 * Copyright (C) 2008 - 2013 Crispico, <http://www.crispico.com/>.
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation version 3.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details, at <http://www.gnu.org/licenses/>.
 * 
 * Contributors:
 *   Crispico - Initial API and implementation
 *
 * license-end
 */
package com.crispico.flower.mp.codesync.wiki.github;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.io.FileUtils;
import org.flowerplatform.model.astcache.wiki.AstCacheWikiFactory;
import org.flowerplatform.model.astcache.wiki.Page;

import com.crispico.flower.mp.codesync.wiki.IConfigurationProvider;
import com.crispico.flower.mp.codesync.wiki.WikiPlugin;
import com.crispico.flower.mp.codesync.wiki.WikiRegexConfiguration;
import com.crispico.flower.mp.codesync.wiki.WikiTextBuilder;
import com.crispico.flower.mp.codesync.wiki.WikiTreeBuilder;
import com.crispico.flower.mp.model.codesync.CodeSyncElement;
import com.crispico.flower.mp.model.codesync.CodeSyncFactory;
import com.crispico.flower.mp.model.codesync.CodeSyncRoot;

/**
 * @author Mariana
 */
public class GithubConfigurationProvider implements IConfigurationProvider {

	public static final String TECHNOLOGY = "github";
	
	private Map<String, IConfigurationProvider> configurationProviders = new HashMap<String, IConfigurationProvider>();
	
	private List<String> skipFilesExtensions = Arrays.asList(".git", ".project", ".notation", ".mindmap");
	
	public GithubConfigurationProvider() {
		configurationProviders.put("md", new MarkdownConfigurationProvider());
	}
	
	@Override
	public void buildConfiguration(WikiRegexConfiguration config, CodeSyncElement cse) {
		getConfigurationProviderForPage(cse).buildConfiguration(config, cse);
	}

	@Override
	public Class<? extends WikiTreeBuilder> getWikiTreeBuilderClass() {
		return WikiTreeBuilder.class;
	}

	@Override
	public WikiTextBuilder getWikiTextBuilder(CodeSyncElement cse) {
		return getConfigurationProviderForPage(cse).getWikiTextBuilder(cse);
	}
	
	/**
	 * @param wiki instance of {@link File}, can be a page or directory
	 */
	@Override
	public CodeSyncRoot getWikiTree(Object wiki) {
		File file = (File) wiki;
		CodeSyncRoot root = CodeSyncFactory.eINSTANCE.createCodeSyncRoot();
		root.setType(WikiPlugin.FOLDER_CATEGORY);
		if (file != null) {
			if (file.isDirectory()) {
				root.setName(file.getPath());
				for (File child : file.listFiles()) {
					createWikiPage(child, root);
				}
			} else {
				root.setName(file.getParentFile().getPath());
				createWikiPage(file, root);
			}
		}
		return root;
	}

	private void createWikiPage(File file, CodeSyncElement root) {
		int index = file.getName().lastIndexOf(".");
		if (index >= 0) {
			String extension = file.getName().substring(index);
			if (skipFilesExtensions.contains(extension)) {
				return;
			}
		}
		
		CodeSyncElement cse = CodeSyncFactory.eINSTANCE.createCodeSyncElement();
		cse.setName(file.getName());
		if (file.isDirectory()) {
			// create new dir
			cse.setType(WikiPlugin.FOLDER_CATEGORY);
			for (File child : file.listFiles()) {
				// recurse for the files in this dir
				createWikiPage(child, cse);
			}
		} else {
			// create new page
			cse.setType(WikiPlugin.PAGE_CATEGORY);
			Page page = AstCacheWikiFactory.eINSTANCE.createPage();
			String content;
			try {
				content = FileUtils.readFileToString(file);
			} catch (IOException e) {
				throw new RuntimeException(e);
			}
			page.setInitialContent(content);
			page.setLineDelimiter(WikiPlugin.getInstance().getLineDelimiter(content));
			cse.setAstCacheElement(page);
		}
		root.getChildren().add(cse);
	}

	@Override
	public void savePage(Page page) {
		String path = WikiPlugin.getInstance().getPagePath(page, "/", true);
		try {
			FileUtils.writeStringToFile(new File(path), page.getInitialContent());
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}

	private IConfigurationProvider getConfigurationProviderForPage(CodeSyncElement cse) {
		String extension = cse.getName().substring(cse.getName().lastIndexOf(".") + 1, cse.getName().length());
		return configurationProviders.get(extension);
	}
	
}