com.dyuproject.protostuff.compiler.CompilerUtil Maven / Gradle / Ivy

Go to download
//========================================================================
//Copyright 2007-2010 David Yu [email protected]
//------------------------------------------------------------------------
//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.dyuproject.protostuff.compiler;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Utilities and helpers for compiling protos.
 *
 * @author David Yu
 * @created Jan 5, 2010
 */
public final class CompilerUtil
{

    
    
    public static Writer newWriter(ProtoModule module, String packageName, String fileName) 
    throws IOException
    {
        String encoding = module.getEncoding();
        if(encoding==null || encoding.isEmpty())
            encoding = "UTF-8";
        
        File packageDir = new File(module.getOutputDir(), packageName.replace('.', '/'));
        if(!packageDir.exists())
            packageDir.mkdirs();
        
        File outputFile = new File(packageDir, fileName);
        FileOutputStream out = new FileOutputStream(outputFile);
        return new BufferedWriter(new OutputStreamWriter(out, encoding));
    }
    
    public static List getProtoFiles(File dir)
    {
        return getFilesByExtension(dir, new String[]{".proto"});
    }
    
    /**
     * Returns a list of file filtered by their file types/extensions; (E.g ".proto")
     */
    public static List getFilesByExtension(File dir, String[] extensions)
    {
        if(!dir.isDirectory() || extensions==null)
            return Collections.emptyList();
        List files = new ArrayList();
        addFilesByExtension(files, dir, extensions);
        return files;
    }
    
    static void addFilesByExtension(List list, File dir, String[] extensions)
    {
        File[] files = dir.listFiles();
        for(int i=0; i