java/manifest/src/org/jetbrains/lang/manifest/parser/ManifestParser.java

/*
 * Copyright (c) 2007-2009, Osmorc Development Team
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright notice, this list
 *       of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright notice, this
 *       list of conditions and the following disclaimer in the documentation and/or other
 *       materials provided with the distribution.
 *     * Neither the name of 'Osmorc Development Team' nor the names of its contributors may be
 *       used to endorse or promote products derived from this software without specific
 *       prior written permission.
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
 * THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package org.jetbrains.lang.manifest.parser;

import com.intellij.lang.ASTNode;
import com.intellij.lang.PsiBuilder;
import com.intellij.lang.PsiParser;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.components.ServiceManager;
import com.intellij.psi.TokenType;
import com.intellij.psi.tree.IElementType;
import com.intellij.psi.tree.TokenSet;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.lang.manifest.ManifestBundle;
import org.jetbrains.lang.manifest.header.HeaderParser;
import org.jetbrains.lang.manifest.header.HeaderParserRepository;
import org.jetbrains.lang.manifest.header.impl.StandardHeaderParser;
import org.jetbrains.lang.manifest.psi.ManifestElementType;
import org.jetbrains.lang.manifest.psi.ManifestTokenType;

import static com.intellij.lang.PsiBuilderUtil.expect;
import static com.intellij.util.ObjectUtils.notNull;

/**
 * @author Robert F. Beeger (robert@beeger.net)
 */
public class ManifestParser implements PsiParser {
  public static final TokenSet HEADER_END_TOKENS = TokenSet.create(ManifestTokenType.SECTION_END, ManifestTokenType.HEADER_NAME);

  private final HeaderParserRepository myRepository;

  public ManifestParser() {
    myRepository = ServiceManager.getService(HeaderParserRepository.class);
  }

  @NotNull
  @Override
  public ASTNode parse(IElementType root, PsiBuilder builder) {
    builder.setDebugMode(ApplicationManager.getApplication().isUnitTestMode());

    PsiBuilder.Marker rootMarker = builder.mark();
    while (!builder.eof()) {
      parseSection(builder);
    }
    rootMarker.done(root);

    return builder.getTreeBuilt();
  }

  private void parseSection(PsiBuilder builder) {
    PsiBuilder.Marker section = builder.mark();

    while (!builder.eof()) {
      IElementType tokenType = builder.getTokenType();
      if (tokenType == ManifestTokenType.HEADER_NAME) {
        parseHeader(builder);
      }
      else if (tokenType == ManifestTokenType.SECTION_END) {
        builder.advanceLexer();
        break;
      }
      else {
        PsiBuilder.Marker marker = builder.mark();
        consumeHeaderValue(builder);
        marker.error(ManifestBundle.message("manifest.header.expected"));
      }
    }

    section.done(ManifestElementType.SECTION);
  }

  private void parseHeader(PsiBuilder builder) {
    PsiBuilder.Marker header = builder.mark();
    String headerName = builder.getTokenText();
    assert headerName != null : "[" + builder.getOriginalText() + "]@" + builder.getCurrentOffset();
    builder.advanceLexer();

    PsiBuilder.Marker errors = null;
    if (builder.getTokenType() == TokenType.BAD_CHARACTER) {
      errors = builder.mark();
      while (builder.getTokenType() == TokenType.BAD_CHARACTER) {
        builder.advanceLexer();
      }
      errors.error(ManifestBundle.message("manifest.unexpected.token"));
    }

    if (builder.getTokenType() == ManifestTokenType.COLON) {
      builder.advanceLexer();

      if (!expect(builder, ManifestTokenType.SIGNIFICANT_SPACE)) {
        builder.error(ManifestBundle.message("manifest.whitespace.expected"));
      }

      HeaderParser headerParser = notNull(myRepository.getHeaderParser(headerName), StandardHeaderParser.INSTANCE);
      headerParser.parse(builder);
    }
    else {
      PsiBuilder.Marker marker;
      if (errors == null) {
        marker = builder.mark();
      }
      else {
        marker = errors.precede();
        errors.drop();
      }
      consumeHeaderValue(builder);
      marker.error(ManifestBundle.message("manifest.colon.expected"));
    }

    header.done(ManifestElementType.HEADER);
  }

  private static void consumeHeaderValue(PsiBuilder builder) {
    while (!builder.eof() && !HEADER_END_TOKENS.contains(builder.getTokenType())) {
      builder.advanceLexer();
    }
  }
}