/* $Id$ */ /** * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You 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. */ package org.apache.manifoldcf.crawler.connectors.dropbox; import org.apache.manifoldcf.core.common.*; import org.apache.manifoldcf.connectorcommon.common.*; import com.dropbox.client2.DropboxAPI; import com.dropbox.client2.exception.DropboxException; import java.io.IOException; import java.io.InputStream; import java.io.InterruptedIOException; import java.util.HashMap; import java.util.HashSet; import java.util.ArrayList; import java.util.List; import java.util.Locale; import java.util.Map; import java.util.Set; import java.util.Iterator; import java.util.Date; import org.apache.manifoldcf.crawler.system.Logging; import org.apache.manifoldcf.crawler.connectors.BaseRepositoryConnector; import org.apache.manifoldcf.agents.interfaces.ServiceInterruption; import org.apache.manifoldcf.core.interfaces.ConfigParams; import org.apache.manifoldcf.core.interfaces.ManifoldCFException; import org.apache.commons.lang.StringUtils; import org.apache.manifoldcf.agents.interfaces.RepositoryDocument; import org.apache.manifoldcf.core.interfaces.Specification; import org.apache.manifoldcf.core.interfaces.IHTTPOutput; import org.apache.manifoldcf.core.interfaces.IPasswordMapperActivity; import org.apache.manifoldcf.core.interfaces.IPostParameters; import org.apache.manifoldcf.core.interfaces.IThreadContext; import org.apache.manifoldcf.core.interfaces.SpecificationNode; import org.apache.manifoldcf.crawler.interfaces.IProcessActivity; import org.apache.manifoldcf.crawler.interfaces.ISeedingActivity; import org.apache.manifoldcf.crawler.interfaces.IExistingVersions; import org.apache.log4j.Logger; /** * * @author andrew */ public class DropboxRepositoryConnector extends BaseRepositoryConnector { protected final static String ACTIVITY_READ = "read document"; public final static String ACTIVITY_FETCH = "fetch"; protected static final String RELATIONSHIP_CHILD = "child"; /** Deny access token for default authority */ private final static String defaultAuthorityDenyToken = "DEAD_AUTHORITY"; // Nodes and attributes private static final String JOB_STARTPOINT_NODE_TYPE = "startpoint"; private static final String JOB_PATH_ATTRIBUTE = "path"; private static final String JOB_ACCESS_NODE_TYPE = "access"; private static final String JOB_TOKEN_ATTRIBUTE = "token"; // Tab properties private static final String DROPBOX_SERVER_TAB_PROPERTY = "DropboxRepositoryConnector.Server"; private static final String DROPBOX_PATH_TAB_PROPERTY = "DropboxRepositoryConnector.DropboxPath"; private static final String DROPBOX_SECURITY_TAB_PROPERTY = "DropboxRepositoryConnector.Security"; // Template names /** * Forward to the javascript to check the configuration parameters */ private static final String EDIT_CONFIG_HEADER_FORWARD = "editConfiguration.js"; /** * Server tab template */ private static final String EDIT_CONFIG_FORWARD_SERVER = "editConfiguration_Server.html"; /** * Forward to the HTML template to view the configuration parameters */ private static final String VIEW_CONFIG_FORWARD = "viewConfiguration.html"; /** * Forward to the javascript to check the specification parameters for the * job */ private static final String EDIT_SPEC_HEADER_FORWARD = "editSpecification.js"; /** * Forward to the template to edit the configuration parameters for the job */ private static final String EDIT_SPEC_FORWARD_DROPBOXPATH = "editSpecification_DropboxPath.html"; /** * Forward to the template to edit the configuration parameters for the job */ private static final String EDIT_SPEC_FORWARD_SECURITY = "editSpecification_Security.html"; /** * Forward to the template to view the specification parameters for the job */ private static final String VIEW_SPEC_FORWARD = "viewSpecification.html"; /** * Endpoint server name */ protected String server = "dropbox"; protected DropboxSession session = null; protected long lastSessionFetch = -1L; protected static final long timeToRelease = 300000L; protected String app_key = null; protected String app_secret = null; protected String key = null; protected String secret = null; public DropboxRepositoryConnector() { super(); } /** * Return the list of activities that this connector supports (i.e. writes * into the log). * * @return the list. */ @Override public String[] getActivitiesList() { return new String[]{ACTIVITY_FETCH, ACTIVITY_READ}; } /** * Get the bin name strings for a document identifier. The bin name * describes the queue to which the document will be assigned for throttling * purposes. Throttling controls the rate at which items in a given queue * are fetched; it does not say anything about the overall fetch rate, which * may operate on multiple queues or bins. For example, if you implement a * web crawler, a good choice of bin name would be the server name, since * that is likely to correspond to a real resource that will need real * throttle protection. * * @param documentIdentifier is the document identifier. * @return the set of bin names. If an empty array is returned, it is * equivalent to there being no request rate throttling available for this * identifier. */ @Override public String[] getBinNames(String documentIdentifier) { return new String[]{server}; } /** * Close the connection. Call this before discarding the connection. */ @Override public void disconnect() throws ManifoldCFException { if (session != null) { session.close(); session = null; lastSessionFetch = -1L; } app_key = null; app_secret= null; key = null; secret = null; } /** * This method create a new DROPBOX session for a DROPBOX repository, if the * repositoryId is not provided in the configuration, the connector will * retrieve all the repositories exposed for this endpoint the it will start * to use the first one. * * @param configParameters is the set of configuration parameters, which in * this case describe the target appliance, basic auth configuration, etc. * (This formerly came out of the ini file.) */ @Override public void connect(ConfigParams configParams) { super.connect(configParams); app_key=params.getParameter(DropboxConfig.APP_KEY_PARAM); app_secret=params.getObfuscatedParameter(DropboxConfig.APP_SECRET_PARAM); key = params.getParameter(DropboxConfig.KEY_PARAM); secret = params.getObfuscatedParameter(DropboxConfig.SECRET_PARAM); } /** * Test the connection. Returns a string describing the connection * integrity. * * @return the connection's status as a displayable string. */ @Override public String check() throws ManifoldCFException { try { checkConnection(); return super.check(); } catch (ServiceInterruption e) { return "Connection temporarily failed: " + e.getMessage(); } catch (ManifoldCFException e) { return "Connection failed: " + e.getMessage(); } } protected void checkConnection() throws ManifoldCFException, ServiceInterruption { getSession(); CheckConnectionThread t = new CheckConnectionThread(); try { t.start(); t.join(); Throwable thr = t.getException(); if (thr != null) { if (thr instanceof DropboxException) { throw (DropboxException) thr; } else if (thr instanceof RuntimeException) { throw (RuntimeException) thr; } else { throw (Error) thr; } } return; } catch (InterruptedException e) { t.interrupt(); throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } catch (DropboxException e) { Logging.connectors.warn("DROPBOX: Error checking repository: " + e.getMessage(), e); handleDropboxException(e); } } protected class CheckConnectionThread extends Thread { protected Throwable exception = null; public CheckConnectionThread() { super(); setDaemon(true); } public void run() { try { session.getRepositoryInfo(); } catch (Throwable e) { this.exception = e; } } public Throwable getException() { return exception; } } /** * Set up a session */ protected void getSession() throws ManifoldCFException, ServiceInterruption { if (session == null) { // Check for parameter validity if (StringUtils.isEmpty(app_key)) { throw new ManifoldCFException("Parameter " + DropboxConfig.APP_KEY_PARAM + " required but not set"); } if (StringUtils.isEmpty(app_secret)) { throw new ManifoldCFException("Parameter " + DropboxConfig.APP_SECRET_PARAM + " required but not set"); } if (StringUtils.isEmpty(key)) { throw new ManifoldCFException("Parameter " + DropboxConfig.KEY_PARAM + " required but not set"); } if (Logging.connectors.isDebugEnabled()) { Logging.connectors.debug("DROPBOX: Username = '" + key + "'"); } if (StringUtils.isEmpty(secret)) { throw new ManifoldCFException("Parameter " + DropboxConfig.SECRET_PARAM + " required but not set"); } Logging.connectors.debug("DROPBOX: Password exists"); // Create a session session = new DropboxSession(app_key, app_secret, key, secret); lastSessionFetch = System.currentTimeMillis(); } } @Override public void poll() throws ManifoldCFException { if (lastSessionFetch == -1L) { return; } long currentTime = System.currentTimeMillis(); if (currentTime >= lastSessionFetch + timeToRelease) { session.close(); session = null; lastSessionFetch = -1L; } } /** This method is called to assess whether to count this connector instance should * actually be counted as being connected. *@return true if the connector instance is actually connected. */ @Override public boolean isConnected() { return session != null; } /** * Get the maximum number of documents to amalgamate together into one * batch, for this connector. * * @return the maximum number. 0 indicates "unlimited". */ @Override public int getMaxDocumentRequest() { return 1; } /** * Return the list of relationship types that this connector recognizes. * * @return the list. */ @Override public String[] getRelationshipTypes() { return new String[]{RELATIONSHIP_CHILD}; } /** * Fill in a Server tab configuration parameter map for calling a Velocity * template. * * @param newMap is the map to fill in * @param parameters is the current set of configuration parameters */ private static void fillInServerConfigurationMap(Map<String, Object> newMap, IPasswordMapperActivity mapper, ConfigParams parameters) { String app_key = parameters.getParameter(DropboxConfig.APP_KEY_PARAM); String app_secret = parameters.getObfuscatedParameter(DropboxConfig.APP_SECRET_PARAM); String username = parameters.getParameter(DropboxConfig.KEY_PARAM); String password = parameters.getObfuscatedParameter(DropboxConfig.SECRET_PARAM); if (app_key == null) { app_key = StringUtils.EMPTY; } if (app_secret == null) { app_secret = StringUtils.EMPTY; } else { app_secret = mapper.mapPasswordToKey(app_secret); } if (username == null) { username = StringUtils.EMPTY; } if (password == null) { password = StringUtils.EMPTY; } else { password = mapper.mapPasswordToKey(password); } newMap.put("APP_KEY", app_key); newMap.put("APP_SECRET", app_secret); newMap.put("KEY", username); newMap.put("SECRET", password); } /** * View configuration. This method is called in the body section of the * connector's view configuration page. Its purpose is to present the * connection information to the user. The coder can presume that the HTML * that is output from this configuration will be within appropriate <html> * and <body> tags. * * @param threadContext is the local thread context. * @param out is the output to which any HTML should be sent. * @param parameters are the configuration parameters, as they currently * exist, for this connection being configured. */ @Override public void viewConfiguration(IThreadContext threadContext, IHTTPOutput out, Locale locale, ConfigParams parameters) throws ManifoldCFException, IOException { Map<String, Object> paramMap = new HashMap<String, Object>(); // Fill in map from each tab fillInServerConfigurationMap(paramMap, out, parameters); Messages.outputResourceWithVelocity(out,locale,VIEW_CONFIG_FORWARD,paramMap); } /** * * Output the configuration header section. This method is called in the * head section of the connector's configuration page. Its purpose is to add * the required tabs to the list, and to output any javascript methods that * might be needed by the configuration editing HTML. * * @param threadContext is the local thread context. * @param out is the output to which any HTML should be sent. * @param parameters are the configuration parameters, as they currently * exist, for this connection being configured. * @param tabsArray is an array of tab names. Add to this array any tab * names that are specific to the connector. */ @Override public void outputConfigurationHeader(IThreadContext threadContext, IHTTPOutput out, Locale locale, ConfigParams parameters, List<String> tabsArray) throws ManifoldCFException, IOException { // Add the Server tab tabsArray.add(Messages.getString(locale, DROPBOX_SERVER_TAB_PROPERTY)); // Map the parameters Map<String, Object> paramMap = new HashMap<String, Object>(); // Fill in the parameters from each tab fillInServerConfigurationMap(paramMap, out, parameters); // Output the Javascript - only one Velocity template for all tabs Messages.outputResourceWithVelocity(out,locale,EDIT_CONFIG_HEADER_FORWARD,paramMap); } @Override public void outputConfigurationBody(IThreadContext threadContext, IHTTPOutput out, Locale locale, ConfigParams parameters, String tabName) throws ManifoldCFException, IOException { // Call the Velocity templates for each tab // Server tab Map<String, Object> paramMap = new HashMap<String, Object>(); // Set the tab name paramMap.put("TabName", tabName); // Fill in the parameters fillInServerConfigurationMap(paramMap, out, parameters); Messages.outputResourceWithVelocity(out,locale,EDIT_CONFIG_FORWARD_SERVER,paramMap); } /** * Process a configuration post. This method is called at the start of the * connector's configuration page, whenever there is a possibility that form * data for a connection has been posted. Its purpose is to gather form * information and modify the configuration parameters accordingly. The name * of the posted form is "editconnection". * * @param threadContext is the local thread context. * @param variableContext is the set of variables available from the post, * including binary file post information. * @param parameters are the configuration parameters, as they currently * exist, for this connection being configured. * @return null if all is well, or a string error message if there is an * error that should prevent saving of the connection (and cause a * redirection to an error page). * */ @Override public String processConfigurationPost(IThreadContext threadContext, IPostParameters variableContext, ConfigParams parameters) throws ManifoldCFException { String app_key = variableContext.getParameter("app_key"); if (app_key != null) { parameters.setParameter(DropboxConfig.APP_KEY_PARAM, app_key); } String app_secret = variableContext.getParameter("app_secret"); if (app_secret != null) { parameters.setObfuscatedParameter(DropboxConfig.APP_SECRET_PARAM, variableContext.mapKeyToPassword(app_secret)); } String key = variableContext.getParameter("key"); if (key != null) { parameters.setParameter(DropboxConfig.KEY_PARAM, key); } String secret = variableContext.getParameter("secret"); if (secret != null) { parameters.setObfuscatedParameter(DropboxConfig.SECRET_PARAM, variableContext.mapKeyToPassword(secret)); } return null; } /** * Fill in specification Velocity parameter map for DROPBOXPath tab. */ private static void fillInDropboxPathSpecificationMap(Map<String, Object> newMap, Specification ds) { int i = 0; String DropboxPath = DropboxConfig.DROPBOX_PATH_PARAM_DEFAULT_VALUE; while (i < ds.getChildCount()) { SpecificationNode sn = ds.getChild(i); if (sn.getType().equals(JOB_STARTPOINT_NODE_TYPE)) { DropboxPath = sn.getAttributeValue(JOB_PATH_ATTRIBUTE); } i++; } newMap.put("DROPBOXPATH", DropboxPath); } /** * Fill in specification Velocity parameter map for Dropbox Security tab. */ private static void fillInDropboxSecuritySpecificationMap(Map<String, Object> newMap, Specification ds) { List<Map<String,String>> accessTokenList = new ArrayList<Map<String,String>>(); for (int i = 0; i < ds.getChildCount(); i++) { SpecificationNode sn = ds.getChild(i); if (sn.getType().equals(JOB_ACCESS_NODE_TYPE)) { String token = sn.getAttributeValue(JOB_TOKEN_ATTRIBUTE); Map<String,String> accessMap = new HashMap<String,String>(); accessMap.put("TOKEN",token); accessTokenList.add(accessMap); } } newMap.put("ACCESSTOKENS", accessTokenList); } /** View specification. * This method is called in the body section of a job's view page. Its purpose is to present the document * specification information to the user. The coder can presume that the HTML that is output from * this configuration will be within appropriate <html> and <body> tags. * The connector will be connected before this method can be called. *@param out is the output to which any HTML should be sent. *@param locale is the locale the output is preferred to be in. *@param ds is the current document specification for this job. *@param connectionSequenceNumber is the unique number of this connection within the job. */ @Override public void viewSpecification(IHTTPOutput out, Locale locale, Specification ds, int connectionSequenceNumber) throws ManifoldCFException, IOException { Map<String, Object> paramMap = new HashMap<String, Object>(); paramMap.put("SeqNum", Integer.toString(connectionSequenceNumber)); // Fill in the map with data from all tabs fillInDropboxPathSpecificationMap(paramMap, ds); fillInDropboxSecuritySpecificationMap(paramMap, ds); Messages.outputResourceWithVelocity(out,locale,VIEW_SPEC_FORWARD,paramMap); } /** Process a specification post. * This method is called at the start of job's edit or view page, whenever there is a possibility that form * data for a connection has been posted. Its purpose is to gather form information and modify the * document specification accordingly. The name of the posted form is always "editjob". * The connector will be connected before this method can be called. *@param variableContext contains the post data, including binary file-upload information. *@param locale is the locale the output is preferred to be in. *@param ds is the current document specification for this job. *@param connectionSequenceNumber is the unique number of this connection within the job. *@return null if all is well, or a string error message if there is an error that should prevent saving of * the job (and cause a redirection to an error page). */ @Override public String processSpecificationPost(IPostParameters variableContext, Locale locale, Specification ds, int connectionSequenceNumber) throws ManifoldCFException { String seqPrefix = "s"+connectionSequenceNumber+"_"; String dropboxPath = variableContext.getParameter(seqPrefix+"dropboxpath"); if (dropboxPath != null) { int i = 0; while (i < ds.getChildCount()) { SpecificationNode oldNode = ds.getChild(i); if (oldNode.getType().equals(JOB_STARTPOINT_NODE_TYPE)) { ds.removeChild(i); break; } i++; } SpecificationNode node = new SpecificationNode(JOB_STARTPOINT_NODE_TYPE); node.setAttribute(JOB_PATH_ATTRIBUTE, dropboxPath); ds.addChild(ds.getChildCount(), node); } String xc = variableContext.getParameter(seqPrefix+"tokencount"); if (xc != null) { // Delete all tokens first int i = 0; while (i < ds.getChildCount()) { SpecificationNode sn = ds.getChild(i); if (sn.getType().equals(JOB_ACCESS_NODE_TYPE)) ds.removeChild(i); else i++; } int accessCount = Integer.parseInt(xc); i = 0; while (i < accessCount) { String accessDescription = "_"+Integer.toString(i); String accessOpName = seqPrefix+"accessop"+accessDescription; xc = variableContext.getParameter(accessOpName); if (xc != null && xc.equals("Delete")) { // Next row i++; continue; } // Get the stuff we need String accessSpec = variableContext.getParameter(seqPrefix+"spectoken"+accessDescription); SpecificationNode node = new SpecificationNode(JOB_ACCESS_NODE_TYPE); node.setAttribute(JOB_TOKEN_ATTRIBUTE,accessSpec); ds.addChild(ds.getChildCount(),node); i++; } String op = variableContext.getParameter(seqPrefix+"accessop"); if (op != null && op.equals("Add")) { String accessspec = variableContext.getParameter(seqPrefix+"spectoken"); SpecificationNode node = new SpecificationNode(JOB_ACCESS_NODE_TYPE); node.setAttribute(JOB_TOKEN_ATTRIBUTE,accessspec); ds.addChild(ds.getChildCount(),node); } } return null; } /** Output the specification body section. * This method is called in the body section of a job page which has selected a repository connection of the * current type. Its purpose is to present the required form elements for editing. * The coder can presume that the HTML that is output from this configuration will be within appropriate * <html>, <body>, and <form> tags. The name of the form is always "editjob". * The connector will be connected before this method can be called. *@param out is the output to which any HTML should be sent. *@param locale is the locale the output is preferred to be in. *@param ds is the current document specification for this job. *@param connectionSequenceNumber is the unique number of this connection within the job. *@param actualSequenceNumber is the connection within the job that has currently been selected. *@param tabName is the current tab name. (actualSequenceNumber, tabName) form a unique tuple within * the job. */ @Override public void outputSpecificationBody(IHTTPOutput out, Locale locale, Specification ds, int connectionSequenceNumber, int actualSequenceNumber, String tabName) throws ManifoldCFException, IOException { // Output DROPBOXPath tab Map<String, Object> paramMap = new HashMap<String, Object>(); paramMap.put("TabName", tabName); paramMap.put("SeqNum", Integer.toString(connectionSequenceNumber)); paramMap.put("SelectedNum", Integer.toString(actualSequenceNumber)); fillInDropboxPathSpecificationMap(paramMap, ds); fillInDropboxSecuritySpecificationMap(paramMap, ds); Messages.outputResourceWithVelocity(out,locale,EDIT_SPEC_FORWARD_DROPBOXPATH,paramMap); Messages.outputResourceWithVelocity(out,locale,EDIT_SPEC_FORWARD_SECURITY,paramMap); } /** Output the specification header section. * This method is called in the head section of a job page which has selected a repository connection of the * current type. Its purpose is to add the required tabs to the list, and to output any javascript methods * that might be needed by the job editing HTML. * The connector will be connected before this method can be called. *@param out is the output to which any HTML should be sent. *@param locale is the locale the output is preferred to be in. *@param ds is the current document specification for this job. *@param connectionSequenceNumber is the unique number of this connection within the job. *@param tabsArray is an array of tab names. Add to this array any tab names that are specific to the connector. */ @Override public void outputSpecificationHeader(IHTTPOutput out, Locale locale, Specification ds, int connectionSequenceNumber, List<String> tabsArray) throws ManifoldCFException, IOException { tabsArray.add(Messages.getString(locale, DROPBOX_PATH_TAB_PROPERTY)); tabsArray.add(Messages.getString(locale, DROPBOX_SECURITY_TAB_PROPERTY)); Map<String, Object> paramMap = new HashMap<String, Object>(); paramMap.put("SeqNum", Integer.toString(connectionSequenceNumber)); // Fill in the specification header map, using data from all tabs. fillInDropboxPathSpecificationMap(paramMap, ds); fillInDropboxSecuritySpecificationMap(paramMap, ds); Messages.outputResourceWithVelocity(out,locale,EDIT_SPEC_HEADER_FORWARD,paramMap); } /** Queue "seed" documents. Seed documents are the starting places for crawling activity. Documents * are seeded when this method calls appropriate methods in the passed in ISeedingActivity object. * * This method can choose to find repository changes that happen only during the specified time interval. * The seeds recorded by this method will be viewed by the framework based on what the * getConnectorModel() method returns. * * It is not a big problem if the connector chooses to create more seeds than are * strictly necessary; it is merely a question of overall work required. * * The end time and seeding version string passed to this method may be interpreted for greatest efficiency. * For continuous crawling jobs, this method will * be called once, when the job starts, and at various periodic intervals as the job executes. * * When a job's specification is changed, the framework automatically resets the seeding version string to null. The * seeding version string may also be set to null on each job run, depending on the connector model returned by * getConnectorModel(). * * Note that it is always ok to send MORE documents rather than less to this method. * The connector will be connected before this method can be called. *@param activities is the interface this method should use to perform whatever framework actions are desired. *@param spec is a document specification (that comes from the job). *@param seedTime is the end of the time range of documents to consider, exclusive. *@param lastSeedVersionString is the last seeding version string for this job, or null if the job has no previous seeding version string. *@param jobMode is an integer describing how the job is being run, whether continuous or once-only. *@return an updated seeding version string, to be stored with the job. */ @Override public String addSeedDocuments(ISeedingActivity activities, Specification spec, String lastSeedVersion, long seedTime, int jobMode) throws ManifoldCFException, ServiceInterruption { String dropboxPath = StringUtils.EMPTY; int i = 0; while (i < spec.getChildCount()) { SpecificationNode sn = spec.getChild(i); if (sn.getType().equals(JOB_STARTPOINT_NODE_TYPE)) { dropboxPath = sn.getAttributeValue(JOB_PATH_ATTRIBUTE); break; } i++; } getSession(); GetSeedsThread t = new GetSeedsThread(dropboxPath); try { t.start(); boolean wasInterrupted = false; try { XThreadStringBuffer seedBuffer = t.getBuffer(); // Pick up the paths, and add them to the activities, before we join with the child thread. while (true) { // The only kind of exceptions this can throw are going to shut the process down. String docPath = seedBuffer.fetch(); if (docPath == null) break; // Add the pageID to the queue activities.addSeedDocument(docPath); } } catch (InterruptedException e) { wasInterrupted = true; throw e; } catch (ManifoldCFException e) { if (e.getErrorCode() == ManifoldCFException.INTERRUPTED) wasInterrupted = true; throw e; } finally { if (!wasInterrupted) t.finishUp(); } } catch (InterruptedException e) { t.interrupt(); throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } catch (DropboxException e) { Logging.connectors.warn("DROPBOX: Error adding seed documents: " + e.getMessage(), e); handleDropboxException(e); } return ""; } protected class GetSeedsThread extends Thread { protected Throwable exception = null; protected final String path; protected final XThreadStringBuffer seedBuffer; public GetSeedsThread(String path) { super(); this.path = path; this.seedBuffer = new XThreadStringBuffer(); setDaemon(true); } @Override public void run() { try { session.getSeeds(seedBuffer,path,25000); //upper limit on files to get supported by dropbox api in a single directory } catch (Throwable e) { this.exception = e; } finally { seedBuffer.signalDone(); } } public XThreadStringBuffer getBuffer() { return seedBuffer; } public void finishUp() throws InterruptedException, DropboxException { seedBuffer.abandon(); join(); Throwable thr = exception; if (thr != null) { if (thr instanceof DropboxException) throw (DropboxException) thr; else if (thr instanceof RuntimeException) throw (RuntimeException) thr; else if (thr instanceof Error) throw (Error) thr; else throw new RuntimeException("Unhandled exception of type: "+thr.getClass().getName(),thr); } } } /** Process a set of documents. * This is the method that should cause each document to be fetched, processed, and the results either added * to the queue of documents for the current job, and/or entered into the incremental ingestion manager. * The document specification allows this class to filter what is done based on the job. * The connector will be connected before this method can be called. *@param documentIdentifiers is the set of document identifiers to process. *@param statuses are the currently-stored document versions for each document in the set of document identifiers * passed in above. *@param activities is the interface this method should use to queue up new document references * and ingest documents. *@param jobMode is an integer describing how the job is being run, whether continuous or once-only. *@param usesDefaultAuthority will be true only if the authority in use for these documents is the default one. */ @Override public void processDocuments(String[] documentIdentifiers, IExistingVersions statuses, Specification spec, IProcessActivity activities, int jobMode, boolean usesDefaultAuthority) throws ManifoldCFException, ServiceInterruption { Logging.connectors.debug("DROPBOX: Inside processDocuments"); // Forced acls String[] acls = getAcls(spec); // Sort it, java.util.Arrays.sort(acls); for (String documentIdentifier : documentIdentifiers) { getSession(); String versionString; GetObjectThread objt = new GetObjectThread(documentIdentifier); objt.start(); try { objt.finishUp(); } catch (InterruptedException e) { objt.interrupt(); throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } catch (DropboxException e) { Logging.connectors.warn("DROPBOX: Error getting object: " + e.getMessage(), e); handleDropboxException(e); } DropboxAPI.Entry dropboxObject = objt.getResponse(); if (dropboxObject.isDir) { //a folder will always be processed versionString = StringUtils.EMPTY; // adding all the children + subdirs for a folder List<DropboxAPI.Entry> children = dropboxObject.contents; for (DropboxAPI.Entry child : children) { activities.addDocumentReference(child.path, documentIdentifier, RELATIONSHIP_CHILD); } activities.noDocument(documentIdentifier,versionString); continue; } if (dropboxObject.isDeleted) { activities.deleteDocument(documentIdentifier); continue; } if (StringUtils.isEmpty(dropboxObject.rev)) { //a document that doesn't contain versioning information will never be processed activities.deleteDocument(documentIdentifier); continue; } StringBuilder sb = new StringBuilder(); // Acls packList(sb,acls,'+'); if (acls.length > 0) { sb.append('+'); pack(sb,defaultAuthorityDenyToken,'+'); } else sb.append('-'); sb.append(dropboxObject.rev); versionString = sb.toString(); if (!activities.checkDocumentNeedsReindexing(documentIdentifier,versionString)) continue; long startTime = System.currentTimeMillis(); String errorCode = null; String errorDesc = null; Long fileSize = null; String nodeId = documentIdentifier; String version = versionString; try { // Length in bytes long fileLength = dropboxObject.bytes; if (!activities.checkLengthIndexable(fileLength)) { errorCode = activities.EXCLUDED_LENGTH; errorDesc = "Document excluded because of length ("+fileLength+")"; activities.noDocument(documentIdentifier,versionString); continue; } //documentURI String documentURI = dropboxObject.path; if (!activities.checkURLIndexable(documentURI)) { errorCode = activities.EXCLUDED_URL; errorDesc = "Document excluded because of URL ('"+documentURI+"')"; activities.noDocument(documentIdentifier,versionString); continue; } //Modified date Date modifiedDate; if (dropboxObject.modified != null) modifiedDate = com.dropbox.client2.RESTUtility.parseDate(dropboxObject.modified); else modifiedDate = null; if (!activities.checkDateIndexable(modifiedDate)) { errorCode = activities.EXCLUDED_DATE; errorDesc = "Document excluded because of date ("+modifiedDate+")"; activities.noDocument(documentIdentifier,versionString); continue; } // Mime type String mimeType = dropboxObject.mimeType; if (!activities.checkMimeTypeIndexable(mimeType)) { errorCode = activities.EXCLUDED_MIMETYPE; errorDesc = "Document excluded because of mime type ('"+mimeType+"')"; activities.noDocument(documentIdentifier,versionString); continue; } // content ingestion RepositoryDocument rd = new RepositoryDocument(); if (acls.length > 0) { rd.setSecurityACL(RepositoryDocument.SECURITY_TYPE_DOCUMENT,acls); String[] denyAclArray = new String[]{defaultAuthorityDenyToken}; rd.setSecurityDenyACL(RepositoryDocument.SECURITY_TYPE_DOCUMENT,denyAclArray); } if (dropboxObject.path != null) rd.setFileName(dropboxObject.path); if (dropboxObject.mimeType != null) rd.setMimeType(dropboxObject.mimeType); if (dropboxObject.modified != null) rd.setModifiedDate(modifiedDate); // There doesn't appear to be a created date... rd.addField("Modified", dropboxObject.modified); rd.addField("Size", dropboxObject.size); rd.addField("Path", dropboxObject.path); rd.addField("Root", dropboxObject.root); rd.addField("ClientMtime", dropboxObject.clientMtime); rd.addField("mimeType", dropboxObject.mimeType); rd.addField("rev", dropboxObject.rev); getSession(); BackgroundStreamThread t = new BackgroundStreamThread(nodeId); t.start(); try { boolean wasInterrupted = false; try { InputStream is = t.getSafeInputStream(); try { rd.setBinary(is, fileLength); activities.ingestDocumentWithException(nodeId, version, documentURI, rd); // No errors. Record the fact that we made it. errorCode = "OK"; fileSize = new Long(fileLength); } finally { is.close(); } } catch (java.net.SocketTimeoutException e) { throw e; } catch (InterruptedIOException e) { wasInterrupted = true; throw e; } catch (ManifoldCFException e) { if (e.getErrorCode() == ManifoldCFException.INTERRUPTED) wasInterrupted = true; throw e; } finally { if (!wasInterrupted) // This does a join t.finishUp(); } } catch (InterruptedException e) { // We were interrupted out of the join, most likely. Before we abandon the thread, // send a courtesy interrupt. t.interrupt(); throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } catch (java.net.SocketTimeoutException e) { errorCode = e.getClass().getSimpleName().toUpperCase(Locale.ROOT); errorDesc = e.getMessage(); handleIOException(e); } catch (InterruptedIOException e) { t.interrupt(); throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } catch (IOException e) { errorCode = e.getClass().getSimpleName().toUpperCase(Locale.ROOT); errorDesc = e.getMessage(); handleIOException(e); } catch (DropboxException e) { Logging.connectors.warn("DROPBOX: Error getting stream: " + e.getMessage(), e); errorCode = e.getClass().getSimpleName().toUpperCase(Locale.ROOT); errorDesc = e.getMessage(); handleDropboxException(e); } } catch (ManifoldCFException e) { if (e.getErrorCode() == ManifoldCFException.INTERRUPTED) errorCode = null; throw e; } finally { if (errorCode != null) activities.recordActivity(new Long(startTime), ACTIVITY_READ, fileSize, nodeId, errorCode, errorDesc, null); } } } protected class GetObjectThread extends Thread { protected final String nodeId; protected Throwable exception = null; protected DropboxAPI.Entry response = null; public GetObjectThread(String nodeId) { super(); setDaemon(true); this.nodeId = nodeId; } public void run() { try { response = session.getObject(nodeId); } catch (Throwable e) { this.exception = e; } } public void finishUp() throws InterruptedException, DropboxException { join(); Throwable thr = exception; if (thr != null) { if (thr instanceof DropboxException) throw (DropboxException) thr; else if (thr instanceof RuntimeException) throw (RuntimeException) thr; else if (thr instanceof Error) throw (Error) thr; else throw new RuntimeException("Unhandled exception of type: "+thr.getClass().getName(),thr); } } public DropboxAPI.Entry getResponse() { return response; } public Throwable getException() { return exception; } } protected class BackgroundStreamThread extends Thread { protected final String nodeId; protected boolean abortThread = false; protected Throwable responseException = null; protected InputStream sourceStream = null; protected XThreadInputStream threadStream = null; public BackgroundStreamThread(String nodeId) { super(); setDaemon(true); this.nodeId = nodeId; } public void run() { try { try { synchronized (this) { if (!abortThread) { sourceStream = session.getDropboxInputStream(nodeId); threadStream = new XThreadInputStream(sourceStream); this.notifyAll(); } } if (threadStream != null) { // Stuff the content until we are done threadStream.stuffQueue(); } } finally { if (sourceStream != null) sourceStream.close(); } } catch (Throwable e) { responseException = e; } } public InputStream getSafeInputStream() throws InterruptedException, IOException, DropboxException { // Must wait until stream is created, or until we note an exception was thrown. while (true) { synchronized (this) { if (responseException != null) throw new IllegalStateException("Check for response before getting stream"); checkException(responseException); if (threadStream != null) return threadStream; wait(); } } } public void finishUp() throws InterruptedException, IOException, DropboxException { // This will be called during the finally // block in the case where all is well (and // the stream completed) and in the case where // there were exceptions. synchronized (this) { if (threadStream != null) { threadStream.abort(); } abortThread = true; } join(); checkException(responseException); } protected synchronized void checkException(Throwable exception) throws IOException, DropboxException { if (exception != null) { Throwable e = exception; if (e instanceof DropboxException) throw (DropboxException)e; else if (e instanceof IOException) throw (IOException)e; else if (e instanceof RuntimeException) throw (RuntimeException)e; else if (e instanceof Error) throw (Error)e; else throw new RuntimeException("Unhandled exception of type: "+e.getClass().getName(),e); } } } /** Grab forced acl out of document specification. *@param spec is the document specification. *@return the acls. */ protected static String[] getAcls(Specification spec) { Set<String> map = new HashSet<String>(); for (int i = 0; i < spec.getChildCount(); i++) { SpecificationNode sn = spec.getChild(i); if (sn.getType().equals(JOB_ACCESS_NODE_TYPE)) { String token = sn.getAttributeValue(JOB_TOKEN_ATTRIBUTE); map.add(token); } } String[] rval = new String[map.size()]; Iterator<String> iter = map.iterator(); int i = 0; while (iter.hasNext()) { rval[i++] = (String)iter.next(); } return rval; } /** Handle a dropbox exception. */ protected static void handleDropboxException(DropboxException e) throws ManifoldCFException, ServiceInterruption { // Right now I don't know enough, so throw Service Interruptions long currentTime = System.currentTimeMillis(); throw new ServiceInterruption("Dropbox exception: "+e.getMessage(), e, currentTime + 300000L, currentTime + 3 * 60 * 60000L,-1,false); } /** Handle an IO exception. */ protected static void handleIOException(IOException e) throws ManifoldCFException, ServiceInterruption { if (!(e instanceof java.net.SocketTimeoutException) && (e instanceof InterruptedIOException)) { throw new ManifoldCFException("Interrupted: " + e.getMessage(), e, ManifoldCFException.INTERRUPTED); } long currentTime = System.currentTimeMillis(); throw new ServiceInterruption("IO exception: "+e.getMessage(), e, currentTime + 300000L, currentTime + 3 * 60 * 60000L,-1,false); } }