Add OAuth 2.0 token web scripts copied from Chatter add-on

Author: wabson

share-oauth-repo/src/main/java/org/alfresco/integrations/salesforce/chatter/GetOAuthToken.java

@@ -0,0 +1,76 @@
+package org.alfresco.integrations.salesforce.chatter;
+
+import java.io.IOException;
+
+import org.alfresco.service.cmr.oauth2.OAuth2CredentialsStoreService;
+import org.alfresco.service.cmr.remotecredentials.OAuth2CredentialsInfo;
+import org.json.JSONException;
+import org.json.JSONStringer;
+import org.json.JSONWriter;
+import org.springframework.extensions.webscripts.AbstractWebScript;
+import org.springframework.extensions.webscripts.Format;
+import org.springframework.extensions.webscripts.Status;
+import org.springframework.extensions.webscripts.WebScriptException;
+import org.springframework.extensions.webscripts.WebScriptRequest;
+import org.springframework.extensions.webscripts.WebScriptResponse;
+
+/**
+ * Fetch an OAuth 2.0 ticket from the credentials store.
+ * 
+ * @author Will Abson
+ */
+public class GetOAuthToken extends AbstractWebScript
+{
+
+    // Services
+    private OAuth2CredentialsStoreService    oauth2CredentialsStoreService;
+    
+    public void setOauth2CredentialsStoreService(OAuth2CredentialsStoreService oauth2CredentialsStoreService)
+    {
+        this.oauth2CredentialsStoreService = oauth2CredentialsStoreService;
+    }
+
+    @Override
+    public void execute(WebScriptRequest req, WebScriptResponse resp)
+            throws IOException
+    {
+        String keyName = req.getServiceMatch().getTemplateVars().get("name");
+        
+        if (keyName == null || "".equals(keyName))
+        {
+            throw new WebScriptException("A key name must be specified");
+        }
+        
+        OAuth2CredentialsInfo credentialInfo = oauth2CredentialsStoreService.getPersonalOAuth2Credentials(keyName);
+        
+        if (credentialInfo != null)
+        {
+            try
+            {
+                resp.setContentType(Format.JSON.mimetype());
+                resp.setContentEncoding("UTF-8");
+                // Start object
+                JSONWriter jsonObj = new JSONStringer().object();
+                // Add string values
+                jsonObj.key("accessToken").value(credentialInfo.getOAuthAccessToken());
+                jsonObj.key("refreshToken").value(credentialInfo.getOAuthRefreshToken());
+                jsonObj.key("ticketExpiresAt").value(credentialInfo.getOAuthTicketExpiresAt());
+                jsonObj.key("ticketTokenIssuedAt").value(credentialInfo.getOAuthTicketIssuedAt());
+                // End object
+                jsonObj.endObject();
+                
+                // Write JSON to the response body
+                resp.getWriter().write(jsonObj.toString());
+            }
+            catch (JSONException e)
+            {
+                throw new WebScriptException("Error building JSON data", e);
+            }
+        }
+        else
+        {
+            throw new WebScriptException(Status.STATUS_NOT_FOUND, "Could not find credentials with name " + keyName);
+        }
+    }
+
+}

share-oauth-repo/src/main/java/org/alfresco/integrations/salesforce/chatter/SaveOAuthToken.java

@@ -0,0 +1,80 @@
+package org.alfresco.integrations.salesforce.chatter;
+
+import java.io.IOException;
+import java.util.Date;
+
+import org.alfresco.service.cmr.oauth2.OAuth2CredentialsStoreService;
+import org.alfresco.service.cmr.remoteticket.NoSuchSystemException;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.json.JSONTokener;
+import org.springframework.extensions.webscripts.AbstractWebScript;
+import org.springframework.extensions.webscripts.WebScriptException;
+import org.springframework.extensions.webscripts.WebScriptRequest;
+import org.springframework.extensions.webscripts.WebScriptResponse;
+
+/**
+ * Save an OAuth 2.0 ticket into the credentials store.
+ * 
+ * @author Will Abson
+ */
+public class SaveOAuthToken extends AbstractWebScript
+{
+
+    private static Log logger = LogFactory.getLog(SaveOAuthToken.class);
+    
+    // Services
+    private OAuth2CredentialsStoreService    oauth2CredentialsStoreService;
+    
+    public void setOauth2CredentialsStoreService(OAuth2CredentialsStoreService oauth2CredentialsStoreService)
+    {
+        this.oauth2CredentialsStoreService = oauth2CredentialsStoreService;
+    }
+
+    @Override
+    public void execute(WebScriptRequest req, WebScriptResponse arg1)
+            throws IOException
+    {
+        String jsonStr = req.getContent().getContent();
+        if (logger.isDebugEnabled())
+        {
+            logger.debug("Got JSON data: " + jsonStr);
+        }
+        
+        try
+        {
+            JSONObject reqJson = new JSONObject(new JSONTokener(jsonStr));
+
+            String remoteSystem = reqJson.getString("name"), 
+                    accessToken = reqJson.getString("accessToken"), 
+                    refreshToken = reqJson.getString("refreshToken");
+            
+            if (logger.isDebugEnabled())
+            {
+                logger.debug("Name: " + remoteSystem);
+                logger.debug("Access token: " + accessToken);
+                logger.debug("Refresh token: " + refreshToken);
+            }
+            
+            Date expiresIn = null; // TODO need to pick an arbitrary date?
+            
+            try
+            {
+                oauth2CredentialsStoreService.storePersonalOAuth2Credentials(remoteSystem, accessToken, refreshToken, expiresIn, new Date());
+            }
+            catch (NoSuchSystemException nsse)
+            {
+                throw nsse;
+            }
+            
+        }
+        catch (JSONException e)
+        {
+            throw new WebScriptException("A problem occurred parsing the request JSON", e);
+        }
+        
+    }
+
+}