/**
* Copyright (C) 2013-2015 all@code-story.net
*
* 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 net.codestory.http.templating;
import static java.nio.file.Files.*;
import static java.util.Collections.emptyMap;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toMap;
import static net.codestory.http.io.FileVisitor.*;
import static net.codestory.http.io.Resources.*;
import static net.codestory.http.misc.MemoizingSupplier.*;
import java.io.*;
import java.nio.file.*;
import java.util.*;
import java.util.function.*;
import net.codestory.http.convert.*;
import net.codestory.http.io.*;
import net.codestory.http.misc.*;
import net.codestory.http.templating.yaml.*;
import com.github.jknack.handlebars.*;
public class Site {
private final Resources resources;
private final Supplier<Set<String>> resourceList;
private final Supplier<Map<String, Object>> yaml;
private final Supplier<Map<String, Object>> data;
private final Supplier<List<Map<String, Object>>> pages;
private final Supplier<Map<String, List<Map<String, Object>>>> tags;
private final Supplier<Map<String, List<Map<String, Object>>>> categories;
public Site(Env env, Resources resources) {
this.resources = resources;
resourceList = memoize(() -> list(env));
yaml = memoize(() -> loadYamlConfig("_config.yml"));
data = memoize(() -> getResourceList()
.stream()
.filter(path -> path.startsWith("_data/"))
.collect(toMap(Site::nameWithoutExtension, this::readYaml))
);
pages = memoize(() -> getResourceList()
.stream()
.filter(path -> !path.startsWith("_"))
.map(this::pathToMap)
.collect(toList())
);
tags = memoize(() -> {
Map<String, List<Map<String, Object>>> pagesPerTag = new TreeMap<>();
for (Map<String, Object> page : getPages()) {
for (String tag : tags(page)) {
pagesPerTag.computeIfAbsent(tag, key -> new ArrayList<>()).add(page);
}
}
return pagesPerTag;
});
categories = memoize(() -> getPages().stream().collect(groupingBy(Site::category, TreeMap::new, toList())));
}
private static Set<String> list(Env env) {
Set<String> paths = new TreeSet<>();
try {
if (env.classPath()) {
Path rootPath = Paths.get(env.appFolder());
new ClasspathScanner().getResources(env.appFolder()).forEach(resource -> paths.add(relativePath(rootPath, Paths.get(resource))));
}
if (env.filesystem()) {
Path rootPath = new File(env.workingDir(), env.appFolder()).toPath();
walkFileTree(rootPath, onFile(path -> paths.add(relativePath(rootPath, path))));
}
} catch (IOException e) {
// Ignore
}
paths.remove("");
return paths;
}
private Map<String, Object> configYaml() {
return yaml.get();
}
public Set<String> getResourceList() {
return resourceList.get();
}
public Object get(String key) {
return yaml.get().get(key);
}
public <T> T getAs(String key, Class<T> type) {
return TypeConvert.convertValue(get(key), type);
}
public Map<String, Object> getData() {
return data.get();
}
public List<Map<String, Object>> getPages() {
return pages.get();
}
public Map<String, List<Map<String, Object>>> getTags() {
return tags.get();
}
public Map<String, List<Map<String, Object>>> getCategories() {
return categories.get();
}
private Map<String, Object> pathToMap(String path) {
try {
Path yamlPath = Paths.get(path);
return YamlFrontMatter.parse(resources.sourceFile(yamlPath)).getVariables();
} catch (IOException e) {
throw new IllegalStateException("Unable to read file: " + path, e);
}
}
private Object readYaml(String path) {
try {
return YamlParser.INSTANCE.parse(resources.sourceFile(Paths.get(path)));
} catch (IOException e) {
throw new IllegalStateException("Unable to read file: " + path, e);
}
}
private static String category(Map<String, Object> page) {
return page.getOrDefault("category", "").toString().trim();
}
@SuppressWarnings("unchecked")
private static String[] tags(Map<String, Object> page) {
Object tags = page.getOrDefault("tags", "");
if (tags instanceof List<?>) {
return ((List<String>) tags).stream().toArray(String[]::new);
}
return tags.toString().trim().split("\\s*,\\s*");
}
@SuppressWarnings("unchecked")
private Map<String, Object> loadYamlConfig(String configFile) {
Path configPath = Paths.get(configFile);
if (!resources.exists(configPath)) {
return emptyMap();
}
try {
return YamlParser.INSTANCE.parseMap(resources.sourceFile(configPath));
} catch (IOException e) {
throw new IllegalStateException("Unable to read " + configFile, e);
}
}
private static String nameWithoutExtension(String path) {
return Strings.substringBeforeLast(Paths.get(path).getFileName().toString(), ".");
}
enum SiteValueResolver implements ValueResolver {
INSTANCE;
@Override
public Object resolve(Object context, String name) {
if (context instanceof Site) {
return ((Site) context).configYaml().getOrDefault(name, UNRESOLVED);
}
return UNRESOLVED;
}
@Override
public Object resolve(Object o) {
return UNRESOLVED;
}
@Override
public Set<Map.Entry<String, Object>> propertySet(Object context) {
if (context instanceof Site) {
return ((Site) context).configYaml().entrySet();
}
return Collections.emptySet();
}
}
}