001    /*
002     * Copyright 2010-2016 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.kotlin.js.translate.reference;
018    
019    import org.jetbrains.kotlin.js.backend.ast.JsExpression;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.kotlin.descriptors.ClassDescriptor;
022    import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
023    import org.jetbrains.kotlin.js.translate.context.TranslationContext;
024    import org.jetbrains.kotlin.js.translate.general.AbstractTranslator;
025    import org.jetbrains.kotlin.js.translate.intrinsic.objects.ObjectIntrinsic;
026    import org.jetbrains.kotlin.psi.KtReferenceExpression;
027    import org.jetbrains.kotlin.psi.KtSimpleNameExpression;
028    
029    import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getDescriptorForReferenceExpression;
030    
031    public class CompanionObjectIntrinsicAccessTranslator extends AbstractTranslator implements AccessTranslator {
032        @NotNull
033        /*package*/ static CompanionObjectIntrinsicAccessTranslator newInstance(
034                @NotNull KtSimpleNameExpression expression,
035                @NotNull TranslationContext context
036        ) {
037            DeclarationDescriptor referenceDescriptor = getDescriptorForReferenceExpression(context.bindingContext(), expression);
038            assert referenceDescriptor != null : "JetSimpleName expression must reference a descriptor " + expression.getText();
039            return new CompanionObjectIntrinsicAccessTranslator(referenceDescriptor, context);
040        }
041    
042        /*package*/ static boolean isCompanionObjectReference(
043                @NotNull KtReferenceExpression expression,
044                @NotNull TranslationContext context
045        ) {
046            DeclarationDescriptor descriptor = getDescriptorForReferenceExpression(context.bindingContext(), expression);
047            return descriptor instanceof ClassDescriptor && context.intrinsics().getObjectIntrinsic((ClassDescriptor) descriptor).exists();
048        }
049    
050        @NotNull
051        private final JsExpression referenceToCompanionObject;
052    
053        private CompanionObjectIntrinsicAccessTranslator(@NotNull DeclarationDescriptor descriptor, @NotNull TranslationContext context) {
054            super(context);
055            this.referenceToCompanionObject = generateReferenceToCompanionObject(descriptor, context);
056        }
057    
058        @NotNull
059        private static JsExpression generateReferenceToCompanionObject(
060                @NotNull DeclarationDescriptor descriptor,
061                @NotNull TranslationContext context
062        ) {
063            ObjectIntrinsic objectIntrinsic = context.intrinsics().getObjectIntrinsic((ClassDescriptor) descriptor);
064            return objectIntrinsic.apply(context);
065        }
066    
067        @Override
068        @NotNull
069        public JsExpression translateAsGet() {
070            return referenceToCompanionObject;
071        }
072    
073        @Override
074        @NotNull
075        public JsExpression translateAsSet(@NotNull JsExpression toSetTo) {
076            throw new IllegalStateException("companion object can't be set");
077        }
078    
079        @NotNull
080        @Override
081        public AccessTranslator getCached() {
082            return this;
083        }
084    }