Package

Source Code of DocumentList

/* Copyright (c) 2008 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import javax.activation.MimetypesFileTypeMap;

import com.google.gdata.client.DocumentQuery;
import com.google.gdata.client.GoogleService;
import com.google.gdata.client.Query;
import com.google.gdata.client.GoogleAuthTokenFactory.UserToken;
import com.google.gdata.client.docs.DocsService;
import com.google.gdata.data.Link;
import com.google.gdata.data.MediaContent;
import com.google.gdata.data.PlainTextConstruct;
import com.google.gdata.data.acl.AclEntry;
import com.google.gdata.data.acl.AclFeed;
import com.google.gdata.data.acl.AclRole;
import com.google.gdata.data.acl.AclScope;
import com.google.gdata.data.docs.DocumentEntry;
import com.google.gdata.data.docs.DocumentListEntry;
import com.google.gdata.data.docs.DocumentListFeed;
import com.google.gdata.data.docs.FolderEntry;
import com.google.gdata.data.docs.PresentationEntry;
import com.google.gdata.data.docs.RevisionFeed;
import com.google.gdata.data.docs.SpreadsheetEntry;
import com.google.gdata.data.media.MediaSource;
import com.google.gdata.util.AuthenticationException;
import com.google.gdata.util.ServiceException;

/**
* An application that serves as a sample to show how the GoogleDocsUpload List
* Service can be used to search your documents, upload and download files,
* change sharing permission, file documents in folders, and view revisions
* history.
*/
public class DocumentList {
  public DocsService service;
  public GoogleService spreadsheetsService;

  public static final String DEFAULT_AUTH_PROTOCOL = "https";
  public static final String DEFAULT_AUTH_HOST = "docs.google.com";

  public static final String DEFAULT_PROTOCOL = "http";
  public static final String DEFAULT_HOST = "docs.google.com";

  public static final String SPREADSHEETS_SERVICE_NAME = "wise";
  public static final String SPREADSHEETS_HOST = "spreadsheets.google.com";

  private final String URL_FEED = "/feeds";
  private final String URL_DOWNLOAD = "/download";
  private final String URL_DOCLIST_FEED = "/private/full";

  private final String URL_DEFAULT = "/default";
  private final String URL_FOLDERS = "/contents";
  private final String URL_ACL = "/acl";
  private final String URL_REVISIONS = "/revisions";

  private final String URL_CATEGORY_DOCUMENT = "/-/document";
  private final String URL_CATEGORY_SPREADSHEET = "/-/spreadsheet";
  private final String URL_CATEGORY_PDF = "/-/pdf";
  private final String URL_CATEGORY_PRESENTATION = "/-/presentation";
  private final String URL_CATEGORY_STARRED = "/-/starred";
  private final String URL_CATEGORY_TRASHED = "/-/trashed";
  private final String URL_CATEGORY_FOLDER = "/-/folder";
  private final String URL_CATEGORY_EXPORT = "/Export";

  private final String PARAMETER_SHOW_FOLDERS = "showfolders=true";

  @SuppressWarnings("unused")
  private String applicationName;
  @SuppressWarnings("unused")
  private String authProtocol;
  @SuppressWarnings("unused")
  private String authHost;
  private String protocol;
  private String host;
  @SuppressWarnings("unused")
  private String username;
  @SuppressWarnings("unused"
  private String password;
  @SuppressWarnings("unused")
  private String authSubToken;

  private final Map<String, String> DOWNLOAD_DOCUMENT_FORMATS;
  {
    DOWNLOAD_DOCUMENT_FORMATS = new HashMap<String, String>();
    DOWNLOAD_DOCUMENT_FORMATS.put("doc", "doc");
    DOWNLOAD_DOCUMENT_FORMATS.put("txt", "txt");
    DOWNLOAD_DOCUMENT_FORMATS.put("odt", "odt");
    DOWNLOAD_DOCUMENT_FORMATS.put("pdf", "pdf");
    DOWNLOAD_DOCUMENT_FORMATS.put("png", "png");
    DOWNLOAD_DOCUMENT_FORMATS.put("rtf", "rtf");
    DOWNLOAD_DOCUMENT_FORMATS.put("html", "html");
    DOWNLOAD_DOCUMENT_FORMATS.put("zip", "zip");
  }

  private final Map<String, String> DOWNLOAD_PRESENTATION_FORMATS;
  {
    DOWNLOAD_PRESENTATION_FORMATS = new HashMap<String, String>();
    DOWNLOAD_PRESENTATION_FORMATS.put("pdf", "pdf");
    DOWNLOAD_PRESENTATION_FORMATS.put("png", "png");
    DOWNLOAD_PRESENTATION_FORMATS.put("ppt", "ppt");
    DOWNLOAD_PRESENTATION_FORMATS.put("swf", "swf");
    DOWNLOAD_PRESENTATION_FORMATS.put("txt", "txt");
  }

  private final Map<String, String> DOWNLOAD_SPREADSHEET_FORMATS;
  {
    DOWNLOAD_SPREADSHEET_FORMATS = new HashMap<String, String>();
    DOWNLOAD_SPREADSHEET_FORMATS.put("xls", "xls");
    DOWNLOAD_SPREADSHEET_FORMATS.put("ods", "ods");
    DOWNLOAD_SPREADSHEET_FORMATS.put("pdf", "pdf");
    DOWNLOAD_SPREADSHEET_FORMATS.put("csv", "csv");
    DOWNLOAD_SPREADSHEET_FORMATS.put("tsv", "tsv");
    DOWNLOAD_SPREADSHEET_FORMATS.put("html", "html");
  }

  /**
   * Constructor.
   *
   * @param applicationName name of the application.
   * @throws DocumentListException the document list exception
   */
  public DocumentList(String applicationName) throws DocumentListException {
    this(applicationName, DEFAULT_AUTH_PROTOCOL, DEFAULT_AUTH_HOST, DEFAULT_PROTOCOL, DEFAULT_HOST);
  }

  /**
   * Constructor.
   *
   * @param applicationName name of the application
   * @param authProtocol the protocol to use for authentication
   * @param authHost the host to use for authentication
   * @param protocol the protocol to use for the http calls.
   * @param host the host that contains the feeds
   * @throws DocumentListException the document list exception
   */
  public DocumentList(String applicationName, String authProtocol, String authHost, String protocol, String host) throws DocumentListException {
    if (authProtocol == null || authHost == null || protocol == null || host == null) {
      throw new DocumentListException("null passed in required parameters");
    }

    service = new DocsService(applicationName);

    // Creating a spreadsheets service is necessary for downloading
    // spreadsheets
    spreadsheetsService = new GoogleService(SPREADSHEETS_SERVICE_NAME, applicationName);

    this.applicationName = applicationName;
    this.authProtocol = authProtocol;
    this.authHost = authHost;
    this.protocol = protocol;
    this.host = host;
  }

  /**
   * Set user credentials based on a username and password.
   *
   * @param user username to log in with.
   * @param pass password for the user logging in.
   * @throws AuthenticationException the authentication exception
   * @throws DocumentListException the document list exception
   */
  public void login(String user, String pass) throws AuthenticationException, DocumentListException {
    if (user == null || pass == null) {
      throw new DocumentListException("null login credentials");
    }

    this.username = user;
    this.password = pass;
    this.authSubToken = "";
    service.setUserCredentials(user, pass);
    spreadsheetsService.setUserCredentials(user, pass);
  }

  /**
   * Allow a user to login using an AuthSub token.
   *
   * @param token the token to be used when logging in.
   * @throws AuthenticationException the authentication exception
   * @throws DocumentListException the document list exception
   */
  public void loginWithAuthSubToken(String token) throws AuthenticationException, DocumentListException {
    if (token == null) {
      throw new DocumentListException("null login credentials");
    }

    this.authSubToken = token;
    this.username = "";
    this.password = "";

    service.setAuthSubToken(token);
    spreadsheetsService.setAuthSubToken(token);
  }

  /**
   * Create a new item in the DocList.
   *
   * @param title the title of the document to be created.
   * @param type the type of the document to be created. One of "spreadsheet",
   * "presentation", or "document".
   * @return the document list entry
   * @throws MalformedURLException the malformed url exception
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry createNew(String title, String type) throws MalformedURLException, IOException, ServiceException, DocumentListException {
    if (title == null || type == null) {
      throw new DocumentListException("null title or type");
    }

    DocumentListEntry newEntry = null;
    if (type.equals("document")) {
      newEntry = new DocumentEntry();
    } else if (type.equals("presentation")) {
      newEntry = new PresentationEntry();
    } else if (type.equals("spreadsheet")) {
      newEntry = new SpreadsheetEntry();
    } else if (type.equals("folder")) {
      newEntry = new FolderEntry();
    }

    newEntry.setTitle(new PlainTextConstruct(title));
    return service.insert(buildUrl(URL_DEFAULT + URL_DOCLIST_FEED), newEntry);
  }

  /**
   * Creates the new sub folder.
   *
   * @param title the title
   * @param folderResourceId the folder resource id
   * @return the document list entry
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry createNewSubFolder(String title, String folderResourceId) throws IOException, ServiceException, DocumentListException {
    DocumentListEntry newEntry = new FolderEntry();
    newEntry.setTitle(new PlainTextConstruct(title));
    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderResourceId + URL_FOLDERS);
    return service.insert(url, newEntry);
  }

  /**
   * Gets a feed containing the documents.
   *
   * @param category
   *            what types of documents to list: "all": lists all the doc
   *            objects (documents, spreadsheets, presentations) "folders":
   *            lists all doc objects including folders. "documents": lists
   *            only documents. "spreadsheets": lists only spreadsheets.
   *            "pdfs": lists only pdfs. "presentations": lists only
   *            presentations. "starred": lists only starred objects.
   *            "trashed": lists trashed objects.
   *
   * @throws IOException
   * @throws MalformedURLException
   * @throws ServiceException
   * @throws DocumentListException
   */
  public DocumentListFeed getDocsListFeed(String category) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (category == null) {
      throw new DocumentListException("null category");
    }

    URL url;

    if (category.equals("all")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED);
    } else if (category.equals("folders")) {
      String[] parameters = { PARAMETER_SHOW_FOLDERS };
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_FOLDER, parameters);
    } else if (category.equals("documents")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_DOCUMENT);
    } else if (category.equals("spreadsheets")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_SPREADSHEET);
    } else if (category.equals("pdfs")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_PDF);
    } else if (category.equals("presentations")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_PRESENTATION);
    } else if (category.equals("starred")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_STARRED);
    } else if (category.equals("trashed")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_TRASHED);
    } else {
      return null;
    }

    DocumentQuery query = new DocumentQuery(url);
    // if (!category.equals("folders")) {
    // query.setMaxResults(101);
    // }

    return service.getFeed(query, DocumentListFeed.class);
  }

  public DocumentListFeed getDocsListFeed(Link link) throws MalformedURLException, IOException, ServiceException, DocumentListException {
    if (link != null) {
      return service.getFeed(new URL(link.getHref()), DocumentListFeed.class);
    }
    return null;
  }

  /**
   * Gets the entry for the provided object id.
   *
   * @param resourceId the resource id of the object to fetch an entry for.
   * @return the docs list entry
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry getDocsListEntry(String resourceId) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }
    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId);

    return service.getEntry(url, DocumentListEntry.class);
  }

  /**
   * Gets the feed for all the objects contained in a folder.
   *
   * @param folderResourceId the resource id of the folder to return the feed for the
   * contents.
   * @return the folder docs list feed
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListFeed getFolderDocsListFeed(String folderResourceId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (folderResourceId == null) {
      throw new DocumentListException("null folderResourceId");
    }
    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderResourceId + URL_FOLDERS);
    return service.getFeed(url, DocumentListFeed.class);
  }

  /**
   * Gets the feed for all the folders contained in a folder.
   *
   * @param folderResourceId the resource id of the folder to return the feed for the
   * contents.
   * @return the sub folders
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListFeed getSubFolders(String folderResourceId) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (folderResourceId == null) {
      throw new DocumentListException("null folderResourceId");
    }
    String[] parameters = { PARAMETER_SHOW_FOLDERS };
    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderResourceId + URL_FOLDERS + URL_CATEGORY_FOLDER, parameters);
    return service.getFeed(url, DocumentListFeed.class);
  }

  /**
   * Gets a feed containing the documents.
   *
   * @param resourceId the resource id of the object to fetch revisions for.
   * @return the revisions feed
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public RevisionFeed getRevisionsFeed(String resourceId) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId + URL_REVISIONS);

    return service.getFeed(url, RevisionFeed.class);
  }

  /**
   * Search the documents, and return a feed of docs that match.
   *
   * @param searchParameters parameters to be used in searching criteria.
   * @return the document list feed
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListFeed search(Map<String, String> searchParameters) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    return search(searchParameters, null);
  }

  /**
   * Search the documents, and return a feed of docs that match.
   *
   * @param searchParameters
   *            parameters to be used in searching criteria. accepted
   *            parameters are: "q": Typical search query "alt": "author":
   *            "updated-min": Lower bound on the last time a document'
   *            content was changed. "updated-max": Upper bound on the last
   *            time a document' content was changed. "edited-min": Lower
   *            bound on the last time a document was edited by the current
   *            user. This value corresponds to the app:edited value in the
   *            Atom entry, which represents changes to the document's content
   *            or metadata. "edited-max": Upper bound on the last time a
   *            document was edited by the current user. This value
   *            corresponds to the app:edited value in the Atom entry, which
   *            represents changes to the document's content or metadata.
   *            "title": Specifies the search terms for the title of a
   *            document. This parameter used without title-exact will only
   *            submit partial queries, not exact queries. "title-exact":
   *            Specifies whether the title query should be taken as an exact
   *            string. Meaningless without title. Possible values are true
   *            and false. "opened-min": Bounds on the last time a document
   *            was opened by the current user. Use the RFC 3339 timestamp
   *            format. For example: 2005-08-09T10:57:00-08:00 "opened-max":
   *            Bounds on the last time a document was opened by the current
   *            user. Use the RFC 3339 timestamp format. For example:
   *            2005-08-09T10:57:00-08:00 "owner": Searches for documents with
   *            a specific owner. Use the email address of the owner.
   *            "writer": Searches for documents which can be written to by
   *            specific users. Use a single email address or a comma
   *            separated list of email addresses. "reader": Searches for
   *            documents which can be read by specific users. Use a single
   *            email address or a comma separated list of email addresses.
   *            "showfolders": Specifies whether the query should return
   *            folders as well as documents. Possible values are true and
   *            false.
   * @param category
   *            define the category to search. (documents, spreadsheets,
   *            presentations, starred, trashed, folders)
   *
   * @throws IOException
   * @throws MalformedURLException
   * @throws ServiceException
   * @throws DocumentListException
   */
  public DocumentListFeed search(Map<String, String> searchParameters, String category) throws IOException, MalformedURLException,
      ServiceException, DocumentListException {
    if (searchParameters == null) {
      throw new DocumentListException("searchParameters null");
    }

    URL url;

    if (category == null || category.equals("")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED);
    } else if (category.equals("documents")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_DOCUMENT);
    } else if (category.equals("spreadsheets")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_SPREADSHEET);
    } else if (category.equals("presentations")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_PRESENTATION);
    } else if (category.equals("starred")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_STARRED);
    } else if (category.equals("trashed")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_TRASHED);
    } else if (category.equals("folders")) {
      url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + URL_CATEGORY_FOLDER);
    } else {
      throw new DocumentListException("invaild category");
    }

    Query qry = new Query(url);

    for (String key : searchParameters.keySet()) {
      qry.setStringCustomParameter(key, searchParameters.get(key));
    }

    return service.query(qry, DocumentListFeed.class);
  }

  /**
   * Upload a file.
   *
   * @param filepath path to uploaded file.
   * @param title title to use for uploaded file.
   * @param convert the convert
   * @param hidden the hidden
   * @return the document list entry
   * @throws IOException when an error occurs in communication with the Doclist
   * service.
   * @throws ServiceException when the request causes an error in the Doclist service.
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry uploadFile(String filepath, String title, boolean convert, boolean hidden) throws IOException, ServiceException,
      DocumentListException {
    if (filepath == null || title == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    File file = new File(filepath);

    DocumentEntry newDocument = new DocumentEntry();
   
    newDocument.setFile(file, getMimeType(file));
   
    newDocument.setTitle(new PlainTextConstruct(title));
    newDocument.setHidden(hidden);

    String url = URL_DEFAULT + URL_DOCLIST_FEED;
    if (!convert) {
      url += "?convert=false";
    }
    return service.insert(buildUrl(url), newDocument);
  }

  /**
   * Upload a file to a folder.
   *
   * @param filepath path to uploaded file.
   * @param title title to use for uploaded file.
   * @param folderResourceId the folder resource id
   * @param convert the convert
   * @param hidden the hidden
   * @return the document list entry
   * @throws IOException when an error occurs in communication with the Doclist
   * service.
   * @throws ServiceException when the request causes an error in the Doclist service.
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry uploadFileToFolder(String filepath, String title, String folderResourceId, boolean convert, boolean hidden)
      throws IOException, ServiceException, DocumentListException {
    if (filepath == null || title == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    File file = new File(filepath);

    DocumentEntry newDocument = new DocumentEntry();
   
        newDocument.setFile(file, getMimeType(file));

    newDocument.setTitle(new PlainTextConstruct(title));
    newDocument.setHidden(hidden);

    String url = URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderResourceId + URL_FOLDERS;
    if (!convert) {
      url += "?convert=false";
    }

    return service.insert(buildUrl(url), newDocument);
  }

  public DocumentListEntry updateFile(String filepath, String title, DocumentListEntry entry, boolean hidden) throws IOException, ServiceException,
      DocumentListException {
    if (filepath == null || title == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    File file = new File(filepath);

        entry.setFile(file, getMimeType(file));
       
    entry.setHidden(hidden);
    return entry.updateMedia(true);
  }
 
  public String getMimeType(File file) {
    String mimeType = "";
        try {
          mimeType = DocumentListEntry.MediaType.fromFileName(file.getName()).getMimeType();
        } catch (IllegalArgumentException e) {
        }

        if (mimeType == "") {
          mimeType = new MimetypesFileTypeMap().getContentType(file);
        }
       
        return mimeType;   
  }

  /**
   * Trash an object.
   *
   * @param resourceId the resource id of object to be trashed.
   * @param delete true to delete the permanently, false to move it to the trash.
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void trashObject(String resourceId, boolean delete) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }

    String feedUrl = URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId;
    if (delete) {
      feedUrl += "?delete=true";
    }

    service.delete(buildUrl(feedUrl), getDocsListEntry(resourceId).getEtag());
  }

  /**
   * Remove an object from a folder.
   *
   * @param resourceId the resource id of an object to be removed from the folder.
   * @param folderResourceId the resource id of the folder to remove the object from.
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void removeFromFolder(String resourceId, String folderResourceId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (resourceId == null || folderResourceId == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderResourceId + URL_FOLDERS + "/" + resourceId);
    service.delete(url, getDocsListEntry(resourceId).getEtag());
  }

  /**
   * Downloads a file.
   *
   * @param exportUrl the full url of the export link to download the file from.
   * @param filepath path and name of the object to be saved as.
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void downloadFile(URL exportUrl, String filepath) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (exportUrl == null || filepath == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    MediaContent mc = new MediaContent();
    mc.setUri(exportUrl.toString());
    MediaSource ms = service.getMedia(mc);

    InputStream inStream = null;
    FileOutputStream outStream = null;

    try {
      inStream = ms.getInputStream();
      outStream = new FileOutputStream(filepath);

      int c;
      while ((c = inStream.read()) != -1) {
        outStream.write(c);
      }
    } finally {
      if (inStream != null) {
        inStream.close();
      }
      if (outStream != null) {
        outStream.flush();
        outStream.close();
      }
    }
  }

  /**
   * Downloads a spreadsheet file.
   *
   * @param resourceId the resource id of the object to be downloaded.
   * @param filepath path and name of the object to be saved as.
   * @param format format to download the file to. The following file types are
   * supported: spreadsheets: "ods", "pdf", "xls", "csv", "html",
   * "tsv"
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void downloadSpreadsheet(String resourceId, String filepath, String format) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (resourceId == null || filepath == null || format == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    UserToken docsToken = (UserToken) service.getAuthTokenFactory().getAuthToken();
    UserToken spreadsheetsToken = (UserToken) spreadsheetsService.getAuthTokenFactory().getAuthToken();
    service.setUserToken(spreadsheetsToken.getValue());

    HashMap<String, String> parameters = new HashMap<String, String>();
    parameters.put("key", resourceId.substring(resourceId.lastIndexOf(':') + 1));
    parameters.put("exportFormat", format);

    // If exporting to .csv or .tsv, add the gid parameter to specify which
    // sheet to export
    if (format.equals(DOWNLOAD_SPREADSHEET_FORMATS.get("csv")) || format.equals(DOWNLOAD_SPREADSHEET_FORMATS.get("tsv"))) {
      parameters.put("gid", "0"); // download only the first sheet
    }

    URL url = buildUrl(SPREADSHEETS_HOST, URL_DOWNLOAD + "/spreadsheets" + URL_CATEGORY_EXPORT, parameters);

    downloadFile(url, filepath);

    // Restore docs token for our DocList client
    service.setUserToken(docsToken.getValue());
  }

  /**
   * Downloads a document.
   *
   * @param resourceId the resource id of the object to be downloaded.
   * @param filepath path and name of the object to be saved as.
   * @param format format to download the file to. The following file types are
   * supported: documents: "doc", "txt", "odt", "png", "pdf",
   * "rtf", "html"
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void downloadDocument(String resourceId, String filepath, String format) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (resourceId == null || filepath == null || format == null) {
      throw new DocumentListException("null passed in for required parameters");
    }
    String[] parameters = { "docID=" + resourceId, "exportFormat=" + format };
    URL url = buildUrl(URL_DOWNLOAD + "/documents" + URL_CATEGORY_EXPORT, parameters);

    downloadFile(url, filepath);
  }

  /**
   * Downloads a presentation.
   *
   * @param resourceId the resource id of the object to be downloaded.
   * @param filepath path and name of the object to be saved as.
   * @param format format to download the file to. The following file types are
   * supported: presentations: "pdf", "ppt", "png", "swf", "txt"
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void downloadPresentation(String resourceId, String filepath, String format) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (resourceId == null || filepath == null || format == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    String[] parameters = { "docID=" + resourceId, "exportFormat=" + format };
    URL url = buildUrl(URL_DOWNLOAD + "/presentations" + URL_CATEGORY_EXPORT, parameters);

    downloadFile(url, filepath);
  }

  /**
   * Moves a object to a folder.
   *
   * @param resourceId the resource id of the object to be moved to the folder.
   * @param folderId the id of the folder to move the object to.
   * @return the document list entry
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public DocumentListEntry moveObjectToFolder(String resourceId, String folderId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (resourceId == null || folderId == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    DocumentListEntry doc = new DocumentListEntry();
    doc.setId(buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId).toString());

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + folderId + URL_FOLDERS);
    return service.insert(url, doc);
  }

  /**
   * Gets the access control list for a object.
   *
   * @param resourceId the resource id of the object to retrieve the ACL for.
   * @return the acl feed
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public AclFeed getAclFeed(String resourceId) throws IOException, MalformedURLException, ServiceException, DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId + URL_ACL);

    return service.getFeed(url, AclFeed.class);
  }

  /**
   * Add an ACL role to an object.
   *
   * @param role the role of the ACL to be added to the object.
   * @param scope the scope for the ACL.
   * @param resourceId the resource id of the object to set the ACL for.
   * @return the acl entry
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public AclEntry addAclRole(AclRole role, AclScope scope, String resourceId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (role == null || scope == null || resourceId == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    AclEntry entry = new AclEntry();
    entry.setRole(role);
    entry.setScope(scope);
    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId + URL_ACL);

    return service.insert(url, entry);
  }

  /**
   * Change the ACL role of a file.
   *
   * @param role the new role of the ACL to be updated.
   * @param scope the new scope for the ACL.
   * @param resourceId the resource id of the object to be updated.
   * @return the acl entry
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public AclEntry changeAclRole(AclRole role, AclScope scope, String resourceId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (role == null || scope == null || resourceId == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId + URL_ACL);

    return service.update(url, scope, role);
  }

  /**
   * Remove an ACL role from a object.
   *
   * @param scope scope of the ACL to be removed.
   * @param email email address to remove the role of.
   * @param resourceId the resource id of the object to remove the role from.
   * @throws IOException Signals that an I/O exception has occurred.
   * @throws MalformedURLException the malformed url exception
   * @throws ServiceException the service exception
   * @throws DocumentListException the document list exception
   */
  public void removeAclRole(String scope, String email, String resourceId) throws IOException, MalformedURLException, ServiceException,
      DocumentListException {
    if (scope == null || email == null || resourceId == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    URL url = buildUrl(URL_DEFAULT + URL_DOCLIST_FEED + "/" + resourceId + URL_ACL + "/" + scope + "%3A" + email);

    service.delete(url);
  }

  /**
   * Returns the format code based on a file extension, and object id.
   *
   * @param resourceId the resource id of the object you want the format for.
   * @param ext extension of the file you want the format for.
   * @return the download format
   * @throws DocumentListException the document list exception
   */
  public String getDownloadFormat(String resourceId, String ext) throws DocumentListException {
    if (resourceId == null || ext == null) {
      throw new DocumentListException("null passed in for required parameters");
    }

    if (resourceId.indexOf("document") == 0) {
      if (DOWNLOAD_DOCUMENT_FORMATS.containsKey(ext)) {
        return DOWNLOAD_DOCUMENT_FORMATS.get(ext);
      }
    } else if (resourceId.indexOf("presentation") == 0) {
      if (DOWNLOAD_PRESENTATION_FORMATS.containsKey(ext)) {
        return DOWNLOAD_PRESENTATION_FORMATS.get(ext);
      }
    } else if (resourceId.indexOf("spreadsheet") == 0) {
      if (DOWNLOAD_SPREADSHEET_FORMATS.containsKey(ext)) {
        return DOWNLOAD_SPREADSHEET_FORMATS.get(ext);
      }
    }
    throw new DocumentListException("invalid document type");
  }

  /**
   * Gets the suffix of the resourceId. If the resourceId is
   * "document:dh3bw3j_0f7xmjhd8", "dh3bw3j_0f7xmjhd8" will be returned.
   *
   * @param resourceId the resource id to extract the suffix from.
   * @return the resource id suffix
   * @throws DocumentListException the document list exception
   */
  public String getResourceIdSuffix(String resourceId) throws DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }

    if (resourceId.indexOf("%3A") != -1) {
      return resourceId.substring(resourceId.lastIndexOf("%3A") + 3);
    } else if (resourceId.indexOf(":") != -1) {
      return resourceId.substring(resourceId.lastIndexOf(":") + 1);
    }
    throw new DocumentListException("Bad resourceId");
  }

  /**
   * Gets the prefix of the resourceId. If the resourceId is
   * "document:dh3bw3j_0f7xmjhd8", "document" will be returned.
   *
   * @param resourceId the resource id to extract the suffix from.
   * @return the resource id prefix
   * @throws DocumentListException the document list exception
   */
  public String getResourceIdPrefix(String resourceId) throws DocumentListException {
    if (resourceId == null) {
      throw new DocumentListException("null resourceId");
    }

    if (resourceId.indexOf("%3A") != -1) {
      return resourceId.substring(0, resourceId.indexOf("%3A"));
    } else if (resourceId.indexOf(":") != -1) {
      return resourceId.substring(0, resourceId.indexOf(":"));
    } else {
      throw new DocumentListException("Bad resourceId");
    }
  }

  /**
   * Builds a URL from a patch.
   *
   * @param path the path to add to the protocol/host
   * @return the URL
   * @throws MalformedURLException the malformed url exception
   * @throws DocumentListException the document list exception
   */
  private URL buildUrl(String path) throws MalformedURLException, DocumentListException {
    if (path == null) {
      throw new DocumentListException("null path");
    }

    return buildUrl(path, null);
  }

  /**
   * Builds a URL with parameters.
   *
   * @param path the path to add to the protocol/host
   * @param parameters parameters to be added to the URL.
   * @return the uRL
   * @throws MalformedURLException the malformed url exception
   * @throws DocumentListException the document list exception
   */
  private URL buildUrl(String path, String[] parameters) throws MalformedURLException, DocumentListException {
    if (path == null) {
      throw new DocumentListException("null path");
    }

    return buildUrl(host, path, parameters);
  }

  /**
   * Builds a URL with parameters.
   *
   * @param domain the domain of the server
   * @param path the path to add to the protocol/host
   * @param parameters parameters to be added to the URL.
   * @return the uRL
   * @throws MalformedURLException the malformed url exception
   * @throws DocumentListException the document list exception
   */
  private URL buildUrl(String domain, String path, String[] parameters) throws MalformedURLException, DocumentListException {
    if (path == null) {
      throw new DocumentListException("null path");
    }

    StringBuffer url = new StringBuffer();
    url.append(protocol + "://" + domain + URL_FEED + path);

    if (parameters != null && parameters.length > 0) {
      url.append("?");
      for (int i = 0; i < parameters.length; i++) {
        url.append(parameters[i]);
        if (i != (parameters.length - 1)) {
          url.append("&");
        }
      }
    }

    return new URL(url.toString());
  }

  /**
   * Builds a URL with parameters.
   *
   * @param domain the domain of the server
   * @param path the path to add to the protocol/host
   * @param parameters parameters to be added to the URL as key value pairs.
   * @return the uRL
   * @throws MalformedURLException the malformed url exception
   * @throws DocumentListException the document list exception
   */
  private URL buildUrl(String domain, String path, Map<String, String> parameters) throws MalformedURLException, DocumentListException {
    if (path == null) {
      throw new DocumentListException("null path");
    }

    StringBuffer url = new StringBuffer();
    url.append(protocol + "://" + domain + URL_FEED + path);

    if (parameters != null && parameters.size() > 0) {
      Set<Map.Entry<String, String>> params = parameters.entrySet();
      Iterator<Map.Entry<String, String>> itr = params.iterator();

      url.append("?");
      while (itr.hasNext()) {
        Map.Entry<String, String> entry = itr.next();
        url.append(entry.getKey() + "=" + entry.getValue());
        if (itr.hasNext()) {
          url.append("&");
        }
      }
    }

    return new URL(url.toString());
  }

}
TOP

Related Classes of DocumentList

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.