AbsArchiveModifier.java

package org.objectweb.jonas_lib.genclientstub.modifier;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.archive.Archive;
import org.objectweb.jonas_lib.genbase.archive.J2EEArchive;
import org.objectweb.jonas_lib.genbase.archive.WebApp;
import org.objectweb.jonas_lib.genbase.generator.Config;
import org.objectweb.jonas_lib.genbase.modifier.ArchiveModifier;
import org.objectweb.jonas_lib.genclientstub.ClientStubGenException;
import org.objectweb.jonas_lib.genclientstub.generator.Generator;

public abstract class AbsArchiveModifier extends ArchiveModifier {

    public AbsArchiveModifier(J2EEArchive archive) {
        super(archive);
    }

    public abstract Archive modify() throws GenBaseException;

    protected void generateFoundStubs(Config config, Archive archive) throws ClientStubGenException {

        // List of classes on which generate stubs
        List classesStub = new ArrayList();
        try {
            List files = archive.getContainedFiles();

            for (Iterator itFiles = files.iterator(); itFiles.hasNext();) {
                String clName = (String) itFiles.next();
                if (clName.endsWith("_Stub.class") || clName.endsWith("_Tie.class")) {
                    // Extract classname from stub and add entry in the list
                    if (archive instanceof WebApp) {
                        if (clName.startsWith("WEB-INF/classes/")) {
                            classesStub.add(clName.substring("WEB-INF/classes/".length()));
                        }
                    } else {
                        classesStub.add(clName);
                    }
                }
            }

            // launch generation for all classes found
            for (Iterator itClass = classesStub.iterator(); itClass.hasNext();) {
                String clName = (String) itClass.next();

                // Remove stub name
                int removeStubIdx = 0;
                if (clName.endsWith("_Stub.class")) {
                    removeStubIdx = clName.length() - "_Stub.class".length();
                } else if (clName.endsWith("_Tie.class")) {
                    removeStubIdx = clName.length() - "_Tie.class".length();
                } else {
                    continue;
                }
                clName = clName.substring(0, removeStubIdx);

                // Remove the _ from the className
                int idx = clName.lastIndexOf('/') + 1;
                if (clName.charAt(idx) == '_') {
                    String pkgName = clName.substring(0, idx);
                    clName = pkgName + clName.substring(idx + 1, clName.length());
                }

                // Translate / into . (/ is in jar entry, classname should contain .)
                clName = clName.replaceAll("/", ".");

                // javax object and existing omg stubs
                if (clName.indexOf("javax.") != -1 || clName.indexOf("org.omg.stub") != -1) {
                    continue;
                }


                // Doesn't take into account Home and Remote
                if (clName.indexOf("Home") != -1 || clName.indexOf("Remote") != -1) {
                    continue;
                }

                Generator g = new Generator(config, null, clName, archive);
                try {
                    g.generate();
                    g.compile();
                } catch (ClientStubGenException cstge) {
                    continue;
                }
                // add files in archive
                g.addFiles(archive);
            }
        } catch (Exception e) {
            throw new ClientStubGenException("Cannot find/build stubs from the file " + archive.getRootFile(), e);
        }

    }

}

---