/*
* Copyright 2013 Alex Lin.
*
* 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://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License 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.
*/
package org.opoo.press.impl;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.opoo.press.Site;
import org.opoo.press.SiteManager;
import org.opoo.press.importer.Importer;
import org.opoo.press.util.ClassUtils;
import org.opoo.press.util.LinkUtils;
import org.opoo.util.ClassPathUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* @author Alex Lin
*
*/
public class SiteManagerImpl extends SiteServiceImpl implements SiteManager {
private static final Logger log = LoggerFactory.getLogger(SiteManagerImpl.class);
public static final String NEW_POST_TEMPLATE = "new_post.ftl";
public static final String NEW_PAGE_TEMPLATE = "new_page.ftl";
public static final String DEFAULT_NEW_POST_FILE = "article/${year}-${month}-${day}-${name}.${format}";
public static final String DEFAULT_NEW_PAGE_FILE = "${name}.${format}";
public static final String NEW_POST_FILE_KEY = "new_post";
public static final String NEW_PAGE_FILE_KEY = "new_page";
public static final String SAMPLE_POST_TEMPLATE = "sample-post.ftl";
public static final String SAMPLE_POST_NAME = "hello-world";
private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm");
/* (non-Javadoc)
* @see org.opoo.press.SiteManager#install(java.io.File, java.util.Locale, boolean)
*/
@Override
public Site install(File siteDir, Locale locale, boolean createSamplePost) throws Exception {
if(siteDir.exists()){
throw new Exception("Site already initialized - " + siteDir.getAbsolutePath());
}
siteDir.mkdirs();
ClassLoader threadLoader = Thread.currentThread().getContextClassLoader();
ClassPathUtils.copyPath(threadLoader, "site", siteDir);
if(locale == null){
locale = Locale.getDefault();
}
updateConfigFile(siteDir, locale);
Site site = createSite(siteDir);
if(createSamplePost){
log.info("Creating sample post.");
//create sample post: hello world
createSamplePost(site, locale);
}
updateCssFile(siteDir);
return site;
}
/**
* @param siteDir
* @param locale
*/
private void updateConfigFile(File siteDir, Locale locale) {
boolean isZH = "zh".equals(locale.getLanguage());
if(isZH){
FileUtils.deleteQuietly(new File(siteDir, "config.yml"));
File file = new File(siteDir, "config_zh_CN.yml");
file.renameTo(new File(siteDir, "config.yml"));
}else{
FileUtils.deleteQuietly(new File(siteDir, "config_zh_CN.yml"));
}
}
private void createSamplePost(Site site, Locale locale) throws IOException{
renderFile(site, "", SAMPLE_POST_NAME, getNewPostFileStyle(site), SAMPLE_POST_TEMPLATE, false, "markdown");
}
private void updateCssFile(File siteDir){
File css = new File(siteDir, "assets/stylesheets/screen.css");
if(css.exists() && css.isFile() && css.canWrite()){
log.info("Update timestamp for {}", css);
css.setLastModified(System.currentTimeMillis() + 10000);
}
}
/* (non-Javadoc)
* @see org.opoo.press.SiteManager#clean(org.opoo.press.Site)
*/
@Override
public void clean(Site site) throws Exception {
File destination = site.getDestination();
File working = site.getWorking();
log.info("Cleaning destination directory " + destination);
FileUtils.deleteDirectory(destination);
log.info("Cleaning working directory " + working);
FileUtils.deleteDirectory(working);
}
/* (non-Javadoc)
* @see org.opoo.press.SiteManager#newPage(org.opoo.press.Site, java.lang.String, java.lang.String)
*/
@Override
public File newPage(Site site, String title, String name) throws Exception {
return newPage(site, title, name, null);
}
public File newPage(Site site, String title, String name, String format) throws Exception {
if(StringUtils.isBlank(title)){
throw new IllegalArgumentException("Title is required.");
}
String newPageFile = (String) site.getConfig().get(NEW_PAGE_FILE_KEY);
if(StringUtils.isBlank(newPageFile)){
newPageFile = DEFAULT_NEW_PAGE_FILE;
}
return renderFile(site, title, name, newPageFile, NEW_PAGE_TEMPLATE, false, format);
}
private String getNewPostFileStyle(Site site){
String newPostFile = (String) site.getConfig().get(NEW_POST_FILE_KEY);
if(StringUtils.isBlank(newPostFile)){
newPostFile = DEFAULT_NEW_POST_FILE;
}
return newPostFile;
}
/* (non-Javadoc)
* @see org.opoo.press.SiteManager#newPost(org.opoo.press.Site, java.lang.String, java.lang.String, boolean)
*/
@Override
public File newPost(Site site, String title, String name, boolean draft) throws Exception {
return newPost(site, title, name, draft, null);
}
public File newPost(Site site, String title, String name, boolean draft, String format) throws Exception {
if(StringUtils.isBlank(title)){
throw new IllegalArgumentException("Title is required.");
}
return renderFile(site, title, name, getNewPostFileStyle(site), NEW_POST_TEMPLATE, draft, format);
}
private File renderFile(Site site, String title, String name, String newFileStyle, String newFileTemplate, boolean isDraft, String format) throws IOException{
name = processName(site, title, name);
if(format == null){
format = "markdown";
}
Date date = new Date();
Map<String,Object> map = new HashMap<String,Object>();
map.put("title", title);
map.put("name", name);
map.put("format", format);
map.put("date", DATE_FORMAT.format(date) );
LinkUtils.addDateParams(map, date);
String filename = site.getRenderer().renderContent(newFileStyle, map);
File file = new File(site.getSource(), filename);
map.put("site", site);
map.put("file", file);
map.put("published", !isDraft);
try{
map.put("version", Site.class.getPackage().getSpecificationVersion());
}catch(Exception e){
map.put("version", "unkown_version");
}
FileOutputStream os = null;
OutputStreamWriter out = null;
try {
File dir = file.getParentFile();
if(!dir.exists()){
dir.mkdirs();
}
os = new FileOutputStream(file);
out = new OutputStreamWriter(os, "UTF-8");
//List<TemplateLoader> list = new ArrayList<TemplateLoader>();
//list.add(new ClassTemplateLoader(Site.class, "templates"));
//RendererImpl rendererImpl = new RendererImpl(site, list);
//rendererImpl.render(newFileTemplate, map, out);
site.getRenderer().render(newFileTemplate, map, out);
} finally{
IOUtils.closeQuietly(out);
IOUtils.closeQuietly(os);
}
log.info("Write to file " + file);
return file;
}
private String processName(Site site, String title, String name) {
if(name == null){
log.info("Using title as post name.");
name = title;
}else{
name = name.trim();
}
name = site.toSlug(name);
return name;
}
public void build(Site site){
long start = System.currentTimeMillis();
try{
site.build();
}catch(Exception e){
log.error("Generate site exception", e);
}finally{
long time = System.currentTimeMillis() - start;
log.info("Generate time: " + time + "ms");
}
}
/* (non-Javadoc)
* @see org.opoo.press.SiteManager#doImport(org.opoo.press.Site, java.lang.String, java.util.Map)
*/
@Override
public void doImport(Site site, String importer, Map<String, Object> params) throws Exception {
@SuppressWarnings("unchecked")
Map<String,String> importers = (Map<String, String>) site.getConfig().get("importers");
Importer importerInstance = null;
if(importers != null){
String className = importers.get(importer);
if(className != null){
importerInstance = (Importer) ClassUtils.newInstance(className, site);
}
}
if(importerInstance == null){
throw new Exception("No valid importer: " + importer);
}
importerInstance.doImport(site, params);
}
}