AbstractTemplateVisitor.java

/*
 * Copyright (C) 2012 Facebook, Inc.
 *
 * 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 com.facebook.swift.generator.visitors;

import com.facebook.swift.generator.SwiftDocumentContext;
import com.facebook.swift.generator.SwiftGeneratorConfig;
import com.facebook.swift.generator.SwiftGeneratorTweak;
import com.facebook.swift.generator.template.JavaContext;
import com.facebook.swift.generator.template.TemplateContextGenerator;
import com.facebook.swift.generator.util.TemplateLoader;
import com.facebook.swift.parser.visitor.DocumentVisitor;
import com.google.common.base.Charsets;
import com.google.common.base.Splitter;
import org.antlr.stringtemplate.NoIndentWriter;
import org.antlr.stringtemplate.StringTemplate;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;

public abstract class AbstractTemplateVisitor implements DocumentVisitor
{
    private final File outputFolder;
    private final TemplateLoader templateLoader;
    protected final TemplateContextGenerator contextGenerator;
    protected final SwiftGeneratorConfig config;

    protected AbstractTemplateVisitor(final TemplateLoader templateLoader,
                                      final SwiftDocumentContext context,
                                      final SwiftGeneratorConfig config,
                                      final File outputFolder)
    {
        this.outputFolder = outputFolder;
        this.templateLoader = templateLoader;
        this.contextGenerator = context.getTemplateContextGenerator();
        this.config = config;
    }

    protected void render(final JavaContext context, final String templateName)
        throws IOException
    {
        final StringTemplate template = templateLoader.load(templateName);
        template.setAttribute("context", context);

        final Map<String, Boolean> tweakMap = new HashMap<>();
        for (SwiftGeneratorTweak tweak: SwiftGeneratorTweak.values()) {
            tweakMap.put(tweak.name(), config.containsTweak(tweak));
        }
        template.setAttribute("tweaks", tweakMap);


        final Iterable<String> packages = Splitter.on('.').split(context.getJavaPackage());
        File folder = outputFolder;

        for (String pkg : packages) {
            folder = new File(folder, pkg);
            folder.mkdir();
        }

        final File file = new File(folder, context.getJavaName() + ".java");

        try (OutputStreamWriter osw = new OutputStreamWriter(new FileOutputStream(file), Charsets.UTF_8)) {
            template.write(new NoIndentWriter(osw));
            osw.flush();
        }
    }
}