org.drools.compiler.lang.descr.PackageDescr Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of drools-compiler Show documentation
There is a newer version: 10.0.0
/*
 * Copyright 2005 JBoss 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 org.drools.compiler.lang.descr;

import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.drools.core.rule.Namespaceable;
import org.kie.internal.definition.KnowledgeDescr;
import org.kie.api.io.Resource;

public class PackageDescr extends BaseDescr
    implements
    Namespaceable,
    KnowledgeDescr {
    private static final long               serialVersionUID       = 530l;
    private String                          documentation;

    private List               imports                = Collections.emptyList();
    private List       functionImports        = Collections.emptyList();
    private List            attributes             = Collections.emptyList();
    private List               globals                = Collections.emptyList();
    private List             functions              = Collections.emptyList();
    private List                 rules                  = Collections.emptyList();
    private List      typeDeclarations       = Collections.emptyList();
    private Set entryPointDeclarations = Collections.emptySet();
    private Set     windowDeclarations     = Collections.emptySet();
    private List      enumDeclarations       = Collections.emptyList();

    public PackageDescr() {
        this( "",
              "");
    }

    public PackageDescr(final String namespace) {
        this( namespace,
              "" );
    }

    public PackageDescr(final String namespace,
                        final String documentation) {
        setNamespace(namespace);
        this.documentation = documentation;
    }

    @SuppressWarnings("unchecked")
    public void readExternal( ObjectInput in ) throws IOException,
                                              ClassNotFoundException {
        super.readExternal( in );
        documentation = in.readUTF();
        imports = (List) in.readObject();
        functionImports = (List) in.readObject();
        attributes = (List) in.readObject();
        globals = (List) in.readObject();
        functions = (List) in.readObject();
        windowDeclarations = (Set) in.readObject();
        rules = (List) in.readObject();
        entryPointDeclarations = (Set) in.readObject();
        typeDeclarations = (List) in.readObject();
        enumDeclarations = (List) in.readObject();
    }

    public void writeExternal( ObjectOutput out ) throws IOException {
        super.writeExternal( out );
        out.writeUTF( documentation );
        out.writeObject(imports);
        out.writeObject( functionImports );
        out.writeObject( attributes );
        out.writeObject( globals );
        out.writeObject( functions );
        out.writeObject( windowDeclarations );
        out.writeObject( rules );
        out.writeObject( entryPointDeclarations );
        out.writeObject( typeDeclarations );
        out.writeObject( enumDeclarations );
    }

    public String getName() {
        return getNamespace() == null ? "" : getNamespace();
    }
    
    public void setName( String name ) {
        setNamespace(name);
    }

    public String getDocumentation() {
        return this.documentation;
    }

    public void addAllImports( final Collection importEntries ) {
        if ( this.imports == Collections.EMPTY_LIST ) {
            this.imports = new ArrayList();
        }
        this.imports.addAll( importEntries );
    }

    public void addImport( final ImportDescr importEntry ) {
        if ( this.imports == Collections.EMPTY_LIST ) {
            this.imports = new ArrayList();
        }
        this.imports.add( importEntry );
    }

    public List getImports() {
        return this.imports;
    }

    public void addFunctionImport( final FunctionImportDescr importFunction ) {
        if ( this.functionImports == Collections.EMPTY_LIST ) {
            this.functionImports = new ArrayList();
        }
        this.functionImports.add( importFunction );
    }

    public List getFunctionImports() {
        return this.functionImports;
    }

    public void addGlobal( final GlobalDescr global ) {
        if ( this.globals == Collections.EMPTY_LIST ) {
            this.globals = new ArrayList();
        }
        this.globals.add( global );
    }

    public List getGlobals() {
        return this.globals;
    }

    public void addAttribute( final AttributeDescr attribute ) {
        if ( this.attributes == Collections.EMPTY_LIST ) {
            this.attributes = new ArrayList();
        }
        this.attributes.add( attribute );
    }

    public List getAttributes() {
        return this.attributes;
    }

    public AttributeDescr getAttribute( String name ) {
        if ( name != null ) {
            for ( AttributeDescr attr : this.attributes ) {
                if ( name.equals( attr.getName() ) ) {
                    return attr;
                }
            }
        }
        return null;
    }

    public void addFunction( final FunctionDescr function ) {
        if ( this.functions == Collections.EMPTY_LIST ) {
            this.functions = new ArrayList( 1 );
        }
        this.functions.add( function );
    }

    public List getFunctions() {
        return this.functions;
    }

    public void addRule( final RuleDescr rule ) {
        if ( this.rules == Collections.EMPTY_LIST ) {
            this.rules = new ArrayList( 1 );
        }
        for ( final AttributeDescr at : attributes ) {
            // check if rule overrides the attribute
            if ( !rule.getAttributes().containsKey( at.getName() ) ) {
                // if not, use default value
                rule.addAttribute( at );
            }
        }
        this.rules.add( rule );
    }

    public List getRules() {
        return this.rules;
    }

    public void addTypeDeclaration( TypeDeclarationDescr declaration ) {
        if ( this.typeDeclarations == Collections.EMPTY_LIST ) {
            this.typeDeclarations = new ArrayList();
        }
        this.typeDeclarations.add( declaration );
    }

    public List getTypeDeclarations() {
        return this.typeDeclarations;
    }

    public void addEntryPointDeclaration( EntryPointDeclarationDescr epDescr ) {
        if ( this.entryPointDeclarations == Collections.EMPTY_SET ) {
            this.entryPointDeclarations = new HashSet();
        }
        this.entryPointDeclarations.add( epDescr );
    }

    public Set getEntryPointDeclarations() {
        return this.entryPointDeclarations;
    }

    public Set getWindowDeclarations() {
        return this.windowDeclarations;
    }
    
    public void addWindowDeclaration( WindowDeclarationDescr window ) {
        if ( this.windowDeclarations == Collections.EMPTY_SET ) {
            this.windowDeclarations = new HashSet();
        }
        this.windowDeclarations.add( window );
    }

    public void addEnumDeclaration( EnumDeclarationDescr declaration ) {
        if ( this.enumDeclarations == Collections.EMPTY_LIST ) {
            this.enumDeclarations = new ArrayList();
        }
        this.enumDeclarations.add( declaration );
    }

    public List getEnumDeclarations() {
        return this.enumDeclarations;
    }
    
    
    public List getClassAndEnumDeclarationDescrs() {
        List list = new ArrayList( getEnumDeclarations() );
        list.addAll( getTypeDeclarations() );
        return Collections.unmodifiableList( list );
    }

    public void removeObjectsGeneratedFromResource(Resource resource) {
        removeObjectsGeneratedFromResource(resource, imports);
        removeObjectsGeneratedFromResource(resource, functionImports);
        removeObjectsGeneratedFromResource(resource, attributes);
        removeObjectsGeneratedFromResource(resource, globals);
        removeObjectsGeneratedFromResource(resource, functions);
        removeObjectsGeneratedFromResource(resource, rules);
        removeObjectsGeneratedFromResource(resource, typeDeclarations);
        removeObjectsGeneratedFromResource(resource, entryPointDeclarations);
        removeObjectsGeneratedFromResource(resource, windowDeclarations);
        removeObjectsGeneratedFromResource(resource, enumDeclarations);
    }

    private  void removeObjectsGeneratedFromResource(Resource resource, Collection descrs) {
        Iterator i = descrs.iterator();
        while (i.hasNext()) {
            if (resource.equals(i.next().getResource())) {
                i.remove();
            }
        }
    }
}