src/main/java/org/apache/freemarker/generator/JsonPropertiesProvider.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.apache.freemarker.generator;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.reflect.TypeToken;
import com.google.gson.stream.JsonReader;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.lang.reflect.Type;
import java.nio.file.Path;
import java.util.HashMap;
import java.util.Map;

public class JsonPropertiesProvider implements OutputGeneratorPropertiesProvider {
	private final Gson gson;
	private final Type stringObjectMap;
	private final File dataDir;
	private final File templateDir;
	private final File outputDir;

	private JsonPropertiesProvider(File dataDir, File templateDir, File outputDir) {
		this.dataDir = dataDir;
		this.templateDir = templateDir;
		this.outputDir = outputDir;
		gson = new GsonBuilder().setLenient().create();
		stringObjectMap = new TypeToken<Map<String, Object>>() { } .getType();
	}

	public static JsonPropertiesProvider create(File dataDir, File templateDir, File outputDir) {
		return new JsonPropertiesProvider(dataDir, templateDir, outputDir);
	}

	@Override
	public void providePropertiesFromFile(Path path, OutputGenerator.OutputGeneratorBuilder builder) {
		File jsonDataFile = path.toFile();
		Map<String,Object> data = parseJson(jsonDataFile);

		Object obj = data.get("dataModel");
		if (obj != null) {
			builder.addDataModel((Map<String, Object>) obj);
		} else {
			builder.addDataModel(new HashMap<String,Object>());
		}

		obj = data.get("templateName");
		if (obj == null) {
			throw new RuntimeException("Require json data property not found: templateName");
		}
		builder.addTemplateLocation(templateDir.toPath().resolve(obj.toString()));

		String dataDirName = dataDir.getAbsolutePath();
		String jsonFileName = jsonDataFile.getAbsolutePath();
		if (!jsonFileName.startsWith(dataDirName)) {
			throw new IllegalStateException("visitFile() given file not in sourceDirectory: " + jsonDataFile);
		}

		String outputFileName = jsonFileName.substring(dataDirName.length()+1);
		outputFileName = outputFileName.substring(0, outputFileName.length() - 5);
		Path outputPath = outputDir.toPath();
		Path resolved = outputPath.resolve(outputFileName);
		builder.addOutputLocation(resolved);
	}

	private Map<String, Object> parseJson(File jsonDataFile) {
		try (JsonReader reader = new JsonReader(new InputStreamReader(new FileInputStream(jsonDataFile), "UTF-8"))) {
			return gson.fromJson(reader, stringObjectMap);
		} catch (Throwable t) {
			throw new RuntimeException("Could not parse json data file: " + jsonDataFile, t);
		}
	}
}