java/languages.antlr/src/org/netbeans/modules/languages/antlr/AntlrTokenId.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.netbeans.modules.languages.antlr;

import java.util.Collection;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.Map;
import org.netbeans.api.editor.mimelookup.MimeRegistration;
import org.netbeans.api.lexer.InputAttributes;
import org.netbeans.api.lexer.Language;
import org.netbeans.api.lexer.LanguagePath;
import org.netbeans.api.lexer.Token;
import org.netbeans.api.lexer.TokenId;
import org.netbeans.spi.lexer.LanguageEmbedding;
import org.netbeans.spi.lexer.LanguageHierarchy;
import org.netbeans.spi.lexer.LanguageProvider;
import org.netbeans.spi.lexer.Lexer;
import org.netbeans.spi.lexer.LexerRestartInfo;
import org.openide.filesystems.FileObject;
import org.openide.util.Lookup;

/**
 *
 * @author lkishalmi
 */
public enum AntlrTokenId implements TokenId {

    ACTION("action"),
    COMMENT("comment"),
    ERROR("error"),
    KEYWORD("keyword"),
    NUMBER("number"),
    PUNCTUATION("punctuation"),
    REGEXP_CHARS("regexp-chars"),
    RULE("rule"),
    STRING("string"),
    TOKEN("token"),
    WHITESPACE("whitespace");

    private final String category;

    AntlrTokenId(String category) {
        this.category = category;
    }

    @Override
    public String primaryCategory() {
        return category;
    }

    public static final String MIME_TYPE = "text/x-antlr4";

    private static final Language<AntlrTokenId> language =
            new LanguageHierarchy<AntlrTokenId>() {

                @Override
                protected String mimeType() {
                    return AntlrTokenId.MIME_TYPE;
                }

                @Override
                protected Collection<AntlrTokenId> createTokenIds() {
                    return EnumSet.allOf(AntlrTokenId.class);
                }

                @Override
                protected Lexer<AntlrTokenId> createLexer(LexerRestartInfo<AntlrTokenId> info) {
                    FileObject fo =(FileObject) info.getAttributeValue(FileObject.class);
                    if (fo != null) {
                        return new AntlrLexer(info);
                    } else {
                        return null;
                    }
                }

                @Override
                protected LanguageEmbedding<? extends TokenId> embedding(Token<AntlrTokenId> token,
                        LanguagePath languagePath, InputAttributes inputAttributes) {
                    switch (token.id()) {
                        case ACTION:
                            // No dependency on the Ruby module:
                            //Language rubyLanguage = RubyTokenId.language();
                            Language<? extends TokenId> javaLanguage = null;

                            @SuppressWarnings("unchecked") Collection<LanguageProvider> providers = (Collection<LanguageProvider>) Lookup.getDefault().lookupAll(LanguageProvider.class);
                            for (LanguageProvider provider : providers) {
                                javaLanguage = (Language<? extends TokenId>) provider.findLanguage("text/x-java"); //NOI18N
                                if (javaLanguage != null) {
                                    break;
                                }
                            }

                            return javaLanguage != null ? LanguageEmbedding.create(javaLanguage, 0, 0, false) : null;
                        default:
                            return null;
                    }
                }
            }.language();

    public static Language<AntlrTokenId> language() {
        return language;
    }



}