72 lines
2.2 KiB
Java
72 lines
2.2 KiB
Java
/******************************************************************************
|
|
* Copyright (c) 2002 - 2006 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;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.net.URLConnection;
|
|
import java.util.Set;
|
|
|
|
import com.ibm.wala.cast.js.html.jericho.JerichoHtmlParser;
|
|
import com.ibm.wala.util.debug.Assertions;
|
|
|
|
public class WebUtil {
|
|
|
|
public static final String preamble = "preamble.js";
|
|
|
|
private static IHtmlParserFactory factory = new IHtmlParserFactory() {
|
|
public IHtmlParser getParser() {
|
|
return new JerichoHtmlParser();
|
|
}
|
|
};
|
|
|
|
public static void setFactory(IHtmlParserFactory factory) {
|
|
WebUtil.factory = factory;
|
|
}
|
|
|
|
public static Set<MappedSourceModule> extractScriptFromHTML(String url) {
|
|
try {
|
|
if (! url.startsWith("file://")) {
|
|
url = "file://" + url;
|
|
}
|
|
return extractScriptFromHTML(new URL(url));
|
|
} catch (MalformedURLException e) {
|
|
Assertions.UNREACHABLE( e.toString() );
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static Set<MappedSourceModule> extractScriptFromHTML(URL url) {
|
|
try {
|
|
JSSourceExtractor extractor = new DefaultSourceExtractor();
|
|
return extractor.extractSources(url, factory.getParser(), new IdentityUrlResolver());
|
|
} catch (IOException e) {
|
|
throw new RuntimeException("trouble with " + url, e);
|
|
}
|
|
}
|
|
|
|
public static void main(String[] args) throws MalformedURLException {
|
|
System.err.println(extractScriptFromHTML(new URL(args[0])));
|
|
}
|
|
|
|
public static InputStream getStream(URL url) throws IOException {
|
|
URLConnection conn = url.openConnection();
|
|
conn.setDefaultUseCaches(false);
|
|
conn.setUseCaches(false);
|
|
|
|
return conn.getInputStream();
|
|
}
|
|
}
|
|
|
|
|