001    /*
002     * Copyright 2010-2013 JetBrains s.r.o.
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package org.jetbrains.jet.lang.descriptors.impl;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
021    import org.jetbrains.jet.lang.descriptors.NamespaceDescriptor;
022    import org.jetbrains.jet.lang.descriptors.PropertyDescriptor;
023    import org.jetbrains.jet.lang.descriptors.SimpleFunctionDescriptor;
024    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
025    import org.jetbrains.jet.lang.resolve.DescriptorUtils;
026    import org.jetbrains.jet.lang.resolve.name.FqName;
027    import org.jetbrains.jet.lang.resolve.name.Name;
028    import org.jetbrains.jet.lang.resolve.scopes.WritableScope;
029    
030    import java.util.List;
031    
032    public class NamespaceDescriptorImpl extends AbstractNamespaceDescriptorImpl {
033    
034        private WritableScope memberScope;
035    
036        public NamespaceDescriptorImpl(@NotNull NamespaceDescriptorParent containingDeclaration,
037                                       @NotNull List<AnnotationDescriptor> annotations,
038                                       @NotNull Name name) {
039            super(containingDeclaration, annotations, name);
040        }
041    
042        public void initialize(@NotNull WritableScope memberScope) {
043            if (this.memberScope != null) {
044                throw new IllegalStateException("Namespace member scope reinitialize");
045            }
046    
047            this.memberScope = memberScope;
048        }
049    
050        @Override
051        @NotNull
052        public WritableScope getMemberScope() {
053            return memberScope;
054        }
055    
056        @Override
057        public void addNamespace(@NotNull NamespaceDescriptor namespaceDescriptor) {
058            getMemberScope().addNamespace(namespaceDescriptor);
059        }
060    
061        @NotNull
062        @Override
063        public FqName getFqName() {
064            return DescriptorUtils.getFQName(this).toSafe();
065        }
066    
067        private NamespaceLikeBuilder builder = null;
068        public NamespaceLikeBuilder getBuilder() {
069            if (builder == null) {
070                builder = new NamespaceLikeBuilder() {
071                    @Override
072                    public void addClassifierDescriptor(@NotNull MutableClassDescriptorLite classDescriptor) {
073                        getMemberScope().addClassifierDescriptor(classDescriptor);
074                    }
075    
076                    @Override
077                    public void addFunctionDescriptor(@NotNull SimpleFunctionDescriptor functionDescriptor) {
078                        getMemberScope().addFunctionDescriptor(functionDescriptor);
079                    }
080    
081                    @Override
082                    public void addPropertyDescriptor(@NotNull PropertyDescriptor propertyDescriptor) {
083                        getMemberScope().addPropertyDescriptor(propertyDescriptor);
084                    }
085    
086                    @NotNull
087                    @Override
088                    public DeclarationDescriptor getOwnerForChildren() {
089                        return NamespaceDescriptorImpl.this;
090                    }
091    
092                    @Override
093                    public ClassObjectStatus setClassObjectDescriptor(@NotNull MutableClassDescriptorLite classObjectDescriptor) {
094                        throw new IllegalStateException("Must be guaranteed not to happen by the parser");
095                    }
096                };
097            }
098    
099            return builder;
100        }
101    }