h2weibo.utils.StatusImageExtractor.java Source code

Introduction

Here is the source code for h2weibo.utils.StatusImageExtractor.java

Source

/*
 * (The MIT License)
 *
 * Copyright (c) 2011 Rakuraku Jyo <jyo.rakuraku@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the 'Software'), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is furnished to
 * do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 */

package h2weibo.utils;

import org.apache.log4j.Logger;
import twitter4j.internal.org.json.JSONException;
import twitter4j.internal.org.json.JSONObject;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.HashMap;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class StatusImageExtractor {
    private static final Logger log = Logger.getLogger(StatusImageExtractor.class.getName());
    private HashMap<String, String> simplePatterns = new HashMap<String, String>();
    private HashMap<String, String[]> jsonPatterns = new HashMap<String, String[]>();

    public StatusImageExtractor() {
        simplePatterns.put(" *http://instagr.am/p/(\\w+)/", "http://instagr.am/p/_KEY_/media/");
        simplePatterns.put(" *http://twitpic.com/(\\w+)", "http://twitpic.com/show/large/_KEY_");
        simplePatterns.put(" *http://img.ly/(\\w+)", "http://img.ly/show/large/_KEY_");
        simplePatterns.put(" *http://yfrog.com/(\\w+)", "http://yfrog.com/_KEY_:iphone");
        simplePatterns.put(" *http://campl.us/(\\w+)", "http://campl.us/_KEY_:iphone");
        simplePatterns.put(" *http://([^ ]+\\.(png|jpg|jpeg))", "http://_KEY_");

        jsonPatterns.put("http://dribbble.com/shots/(\\w+)",
                new String[] { "http://api.dribbble.com/shots/_KEY_", "image_url" });
        jsonPatterns.put("http://drbl.in/(\\w+)",
                new String[] { "http://api.dribbble.com/shots/_KEY_", "image_url" });
    }

    public byte[] extract(StringBuffer input) {
        if (input == null)
            return null;

        for (String key : simplePatterns.keySet()) {
            Pattern p = Pattern.compile(key);
            Matcher m = p.matcher(input);

            if (m.find()) {
                String mediaUrl = simplePatterns.get(key);
                mediaUrl = mediaUrl.replaceAll("_KEY_", m.group(1));

                try {
                    byte[] bytes = downloadUrl(mediaUrl);
                    input.delete(m.start(), m.end()); // -1 for the space before
                    return bytes;
                } catch (IOException e) {
                    log.error("Not able to download image", e);
                }
            }
        }

        for (String key : jsonPatterns.keySet()) {
            Pattern p = Pattern.compile(key);
            Matcher m = p.matcher(input);

            if (m.find()) {
                String jsonUrl = jsonPatterns.get(key)[0];
                jsonUrl = jsonUrl.replaceAll("_KEY_", m.group(1));

                try {
                    byte[] jsonData = downloadUrl(jsonUrl);
                    JSONObject obj = new JSONObject(new String(jsonData));
                    String imageUrl = (String) obj.get(jsonPatterns.get(key)[1]);

                    return downloadUrl(imageUrl);
                } catch (IOException e) {
                    log.error("Not able to download image", e);
                } catch (JSONException e) {
                    log.error("Not able to parse json", e);
                }
            }
        }

        return null;
    }

    private byte[] downloadUrl(String mediaUrl) throws IOException {
        URL url = new URL(mediaUrl);
        InputStream in = url.openStream();
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        for (int b; (b = in.read()) != -1;) {
            out.write(b);
        }
        byte[] bytes = out.toByteArray();
        out.close();
        in.close();

        return bytes;
    }
}