org.moskito.control.plugins.pagespeed.PagespeedTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of moskito-control-plugins Show documentation
The newest version!
package org.moskito.control.plugins.pagespeed;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import net.anotheria.util.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * TODO comment this class
 *
 * @author lrosenberg
 * @since 05.08.20 17:27
 */
public class PagespeedTask {

	/**
	 * Log.
	 */
	private static Logger log = LoggerFactory.getLogger(PagespeedTask.class);
	/**
	 * Config for this task.
	 */
	private PagespeedPluginTargetConfig targetConfig;
	/**
	 * Api key.
	 */
	private String apiKey;

	public PagespeedTask (String apiKey, PagespeedPluginTargetConfig target){
		this.apiKey = apiKey;
		targetConfig = target;
	}

	public Map execute() throws IOException {
		String url = "https://www.googleapis.com/pagespeedonline/v5/runPagespeed?";
		if (apiKey!=null && apiKey.length()>0){
			url += "key="+apiKey+"&";
		}
		url += "strategy="+targetConfig.getStrategy()+"&";
		url += "url="+targetConfig.getUrl();

		log.debug("Retrieving URL "+url);

		String content = PagespeedHttpHelper.getURLContent(url);
		// System.out.println("Content: "+content);
		Map result = parse(content);
		if (log.isDebugEnabled())
			log.debug("Response "+result);
		return result;
	}

	private static Map parse(String content){
		Gson gson = new GsonBuilder().setPrettyPrinting().create();
		HashMap ret = new HashMap<>();

		Map hashMap = gson.fromJson(content, HashMap.class);
		for (String r : Constants.ELEMENTS){
			try {
				String value = retrieve(hashMap, r);
				ret.put(r, value);
			}catch(Exception any){
				log.warn("Couldn't retrieve +r+ from "+hashMap);
			}
		}

		//add metrics.
		Map metrics = (Map)((Map)((Map)hashMap.get("lighthouseResult")).get("audits")).get("metrics");
		metrics = (Map)metrics.get("details");
		List items = (List)metrics.get("items");
		metrics = (Map)items.get(0);

		for (String m : Constants.METRICS){
			String value = ""+metrics.get(m);
			ret.put(m, value);
		}
		return ret;
	}

	private static String retrieve(Map source, String what){
		//System.out.println("retrieving "+what+" from "+source);
		String tokens[] = StringUtils.tokenize(what, '.');
		Map current = source;
		for (int i =0; i