DocGenerator.java

package org.develnext.jphp.genapi;

import org.develnext.jphp.core.syntax.SyntaxAnalyzer;
import org.develnext.jphp.core.tokenizer.Tokenizer;
import org.develnext.jphp.genapi.description.ClassDescription;
import org.develnext.jphp.genapi.template.SphinxTemplate;
import php.runtime.common.collections.map.HashedMap;
import php.runtime.env.Context;
import php.runtime.env.Environment;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class DocGenerator {
    public static final Map<String, String> languages = new HashedMap<String, String>(){{
        put("en", "English");
        //put("ru", "Русский");
    }};

    protected List<File> files = new ArrayList<File>();
    protected Environment environment;

    public DocGenerator(File... directories) {
        environment = new Environment(System.out);

        if (directories != null)
            for(File dir : directories) {
                addDirectory(dir, true);
            }
    }

    public void addDirectory(File directory, boolean recursive) {
        File[] list = directory.listFiles();
        if (list != null) {
            for(File file : list) {
                if (file.isDirectory()) {
                    if (recursive) {
                        addDirectory(file, recursive);
                    }
                } else if (file.isFile()) {
                    if (file.getName().endsWith(".php") || file.getPath().endsWith(".inc")) {
                        files.add(file);
                    }
                }
            }
        }
    }

    protected SyntaxAnalyzer parseFile(File file) {
        try {
            Tokenizer tokenizer = new Tokenizer(new Context(file));
            return new SyntaxAnalyzer(environment, tokenizer);
        } catch (IOException e) {
            throw new RuntimeException(e);
        } catch (Exception e) {
            environment.catchUncaught(e);
            try {
                environment.doFinal();
            } catch (Throwable throwable) {
                throw new RuntimeException(throwable);
            }
            return null;
        }
    }

    public void generate(File targetDirectory, String language) {
        if (!targetDirectory.exists())
            if (!targetDirectory.mkdirs())
                throw new IllegalStateException("Cannot create target directory");

        SphinxTemplate sphinxTemplate = new SphinxTemplate(language, languages.get(language));
        List<ApiDocument> documents = new ArrayList<ApiDocument>();
        for(File file : files) {
            ApiDocument document = new ApiDocument(parseFile(file), sphinxTemplate);
            documents.add(document);
        }

        Map<String, ClassDescription> classMap = new HashedMap<String, ClassDescription>();
        for(ApiDocument document : documents) {
            classMap.putAll(document.getClasses());
        }

        for(ClassDescription el : classMap.values()) {
            if (el.getExtends() != null) {
                ClassDescription parent = classMap.get(el.getExtends().toLowerCase());
                if (parent != null) {
                    parent.addChildClass(el);
                }
            }
        }

        for(ApiDocument document : documents) {
            document.generate(targetDirectory, language, classMap);
        }
    }

    public static void main(String[] args) throws IOException {
        File root = new File(".").getCanonicalFile();
        DocGenerator generator = new DocGenerator();

        for(File file : root.listFiles(new FileFilter() {
            @Override
            public boolean accept(File pathname) {
                return pathname.isDirectory();
            }
        })) {
            File sdkFile1 = new File(file, "resources/JPHP-INF/sdk/");
            File sdkFile2 = new File(file, "src/main/resources/JPHP-INF/sdk/");

            if (sdkFile1.isDirectory()) {
                generator.addDirectory(sdkFile1, true);
            } else if (sdkFile2.isDirectory()) {
                generator.addDirectory(sdkFile2, true);
            }
        }

        for(Map.Entry<String, String> entry : languages.entrySet()) {
            generator.generate(new File("./docs/api_" + entry.getKey()), entry.getKey());
        }
    }
}