| 1 | |
package org.jbehave.core.io.google; |
| 2 | |
|
| 3 | |
import java.io.IOException; |
| 4 | |
import java.io.InputStream; |
| 5 | |
import java.net.MalformedURLException; |
| 6 | |
import java.net.URL; |
| 7 | |
import java.util.List; |
| 8 | |
|
| 9 | |
import org.jbehave.core.io.odf.LoadOdtFromURL; |
| 10 | |
|
| 11 | |
import com.google.gdata.client.DocumentQuery; |
| 12 | |
import com.google.gdata.client.docs.DocsService; |
| 13 | |
import com.google.gdata.data.MediaContent; |
| 14 | |
import com.google.gdata.data.docs.DocumentListEntry; |
| 15 | |
import com.google.gdata.data.docs.DocumentListFeed; |
| 16 | |
import com.google.gdata.data.media.MediaSource; |
| 17 | |
import com.google.gdata.util.AuthenticationException; |
| 18 | |
import com.google.gdata.util.ServiceException; |
| 19 | |
|
| 20 | |
public class LoadOdtFromGoogle extends LoadOdtFromURL { |
| 21 | |
|
| 22 | 1 | private static final DocsService DEFAULT_DOCS_SERVICE = new DocsService("jbehave"); |
| 23 | |
private static final String DEFAULT_FEED_URI = "https://docs.google.com/feeds/default/private/full/"; |
| 24 | |
private final DocsService service; |
| 25 | |
private final String feedURI; |
| 26 | |
|
| 27 | |
public LoadOdtFromGoogle(String username, String password) { |
| 28 | 1 | this(username, password, DEFAULT_FEED_URI); |
| 29 | 0 | } |
| 30 | |
|
| 31 | |
public LoadOdtFromGoogle(String username, String password, String feedURI) { |
| 32 | 1 | this(username, password, feedURI, DEFAULT_DOCS_SERVICE); |
| 33 | 0 | } |
| 34 | |
|
| 35 | 3 | public LoadOdtFromGoogle(String username, String password, String feedURI, DocsService service) { |
| 36 | 3 | this.service = service; |
| 37 | 3 | this.feedURI = feedURI; |
| 38 | |
try { |
| 39 | 3 | service.setUserCredentials(username, password); |
| 40 | 1 | } catch (AuthenticationException e) { |
| 41 | 1 | throw new GoogleAccessFailed(username, e); |
| 42 | 2 | } |
| 43 | 2 | } |
| 44 | |
|
| 45 | |
protected InputStream resourceAsStream(String title) throws IOException, MalformedURLException { |
| 46 | |
try { |
| 47 | 2 | return documentAsStream(exportURL(title)); |
| 48 | 0 | } catch (ServiceException e) { |
| 49 | 0 | throw new IOException(e); |
| 50 | |
} |
| 51 | |
} |
| 52 | |
|
| 53 | |
private String exportURL(String title) throws IOException, ServiceException, MalformedURLException { |
| 54 | 2 | DocumentQuery query = documentQuery(title); |
| 55 | 2 | List<DocumentListEntry> entries = service.getFeed(query, DocumentListFeed.class).getEntries(); |
| 56 | 1 | if (entries.isEmpty()) { |
| 57 | 0 | throw new GoogleDocumentNotFound(title); |
| 58 | |
} |
| 59 | 1 | return ((MediaContent) entries.get(0).getContent()).getUri() + "&exportFormat=odt"; |
| 60 | |
} |
| 61 | |
|
| 62 | |
DocumentQuery documentQuery(String title) throws MalformedURLException { |
| 63 | 1 | DocumentQuery query = new DocumentQuery(new URL(feedURI)); |
| 64 | 1 | query.setTitleQuery(title); |
| 65 | 1 | query.setTitleExact(true); |
| 66 | 1 | query.setMaxResults(1); |
| 67 | 1 | return query; |
| 68 | |
} |
| 69 | |
|
| 70 | |
private InputStream documentAsStream(String url) throws IOException, MalformedURLException { |
| 71 | |
try { |
| 72 | 1 | MediaSource ms = service.getMedia(mediaContent(url)); |
| 73 | 1 | return ms.getInputStream(); |
| 74 | 0 | } catch (ServiceException e) { |
| 75 | 0 | throw new GoogleMediaExportFailed(url, e); |
| 76 | |
} |
| 77 | |
} |
| 78 | |
|
| 79 | |
MediaContent mediaContent(String url) { |
| 80 | 0 | MediaContent mc = new MediaContent(); |
| 81 | 0 | mc.setUri(url); |
| 82 | 0 | return mc; |
| 83 | |
} |
| 84 | |
|
| 85 | |
@SuppressWarnings("serial") |
| 86 | |
public static class GoogleAccessFailed extends RuntimeException { |
| 87 | |
|
| 88 | |
public GoogleAccessFailed(String username, Throwable cause) { |
| 89 | 1 | super("Google access failed for user " + username, cause); |
| 90 | 1 | } |
| 91 | |
|
| 92 | |
} |
| 93 | |
|
| 94 | |
@SuppressWarnings("serial") |
| 95 | |
public static class GoogleDocumentNotFound extends RuntimeException { |
| 96 | |
|
| 97 | |
public GoogleDocumentNotFound(String title) { |
| 98 | 0 | super("Failed to find Google document from " + title); |
| 99 | 0 | } |
| 100 | |
|
| 101 | |
} |
| 102 | |
|
| 103 | |
@SuppressWarnings("serial") |
| 104 | |
public static class GoogleMediaExportFailed extends RuntimeException { |
| 105 | |
|
| 106 | |
public GoogleMediaExportFailed(String url, Throwable cause) { |
| 107 | 0 | super("Failed to export Google media from " + url, cause); |
| 108 | 0 | } |
| 109 | |
|
| 110 | |
} |
| 111 | |
} |