com.ibm.wala.cast.js.html.jericho.JerichoHtmlParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of com.ibm.wala.cast.js Show documentation
T. J. Watson Libraries for Analysis
There is a newer version: 1.6.7
/*
 * Copyright (c) 2002 - 2011 IBM Corporation.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 */
package com.ibm.wala.cast.js.html.jericho;

import com.ibm.wala.cast.ir.translator.TranslatorToCAst;
import com.ibm.wala.cast.js.html.IHtmlCallback;
import com.ibm.wala.cast.js.html.IHtmlParser;
import com.ibm.wala.core.util.warnings.Warning;
import com.ibm.wala.util.collections.HashSetFactory;
import java.io.IOException;
import java.io.Reader;
import java.net.URL;
import java.util.List;
import java.util.Set;
import net.htmlparser.jericho.Config;
import net.htmlparser.jericho.Element;
import net.htmlparser.jericho.Logger;
import net.htmlparser.jericho.LoggerProvider;
import net.htmlparser.jericho.Source;

/**
 * @author danielk Uses the Jericho parser to go over the HTML
 */
public class JerichoHtmlParser implements IHtmlParser {
  static Set warnings = HashSetFactory.make();

  static {
    class CAstLoggerProvider implements LoggerProvider {
      @Override
      public Logger getLogger(String arg0) {
        class CAstLogger implements Logger {

          @Override
          public void debug(String arg0) {
            // TODO Auto-generated method stub

          }

          @Override
          public void error(final String arg0) {
            warnings.add(
                new Warning() {
                  @Override
                  public String getMsg() {
                    return arg0;
                  }
                });
          }

          @Override
          public void info(String arg0) {
            // TODO Auto-generated method stub

          }

          @Override
          public boolean isDebugEnabled() {
            return true;
          }

          @Override
          public boolean isErrorEnabled() {
            return true;
          }

          @Override
          public boolean isInfoEnabled() {
            return true;
          }

          @Override
          public boolean isWarnEnabled() {
            return true;
          }

          @Override
          public void warn(String arg0) {
            // TODO Auto-generated method stub

          }
        }

        return new CAstLogger();
      }
    }

    Config.LoggerProvider = new CAstLoggerProvider();
  }

  @Override
  public void parse(URL url, Reader reader, IHtmlCallback callback, String fileName)
      throws TranslatorToCAst.Error {
    warnings.clear();
    Parser parser = new Parser(callback, fileName);
    Source src;
    try {
      src = new Source(reader);
      src.setLogger(Config.LoggerProvider.getLogger(fileName));
      List childElements = src.getChildElements();
      for (Element e : childElements) {
        parser.parse(e);
      }
      if (!warnings.isEmpty()) {
        throw new TranslatorToCAst.Error(warnings);
      }
    } catch (IOException e) {
      System.err.println("Error parsing file: " + e.getMessage());
    }
  }

  /**
   * @author danielk Inner class does the actual traversal of the HTML using recursion
   */
  private static class Parser {
    private final IHtmlCallback handler;
    private final String fileName;

    public Parser(IHtmlCallback handler, String fileName) {
      this.handler = handler;
      this.fileName = fileName;
    }

    private void parse(Element root) {
      JerichoTag tag = new JerichoTag(root, fileName);
      handler.handleStartTag(tag);
      handler.handleText(tag.getElementPosition(), tag.getBodyText().snd);
      List childElements = root.getChildElements();
      for (Element child : childElements) {
        parse(child);
      }
      handler.handleEndTag(tag);
    }
  }
}