Login page updated to get OAuth2 access token
[pub/Android/ownCloud.git] / src / com / owncloud / android / ui / activity / AuthenticatorActivity.java
index 23ce0c8..f3d6096 100644 (file)
@@ -20,12 +20,20 @@ package com.owncloud.android.ui.activity;
 \r
 import java.net.MalformedURLException;\r
 import java.net.URL;\r
+import java.util.HashMap;\r
+\r
+import org.json.JSONException;\r
+import org.json.JSONObject;\r
 \r
 import com.owncloud.android.AccountUtils;\r
 import com.owncloud.android.authenticator.AccountAuthenticator;\r
 import com.owncloud.android.authenticator.AuthenticationRunnable;\r
 import com.owncloud.android.authenticator.OnAuthenticationResultListener;\r
 import com.owncloud.android.authenticator.OnConnectCheckListener;\r
+import com.owncloud.android.authenticator.oauth2.OAuth2GetCodeRunnable;\r
+import com.owncloud.android.authenticator.oauth2.OnOAuth2GetCodeResultListener;\r
+import com.owncloud.android.authenticator.oauth2.connection.ConnectorOAuth2;\r
+import com.owncloud.android.authenticator.oauth2.services.OAuth2GetTokenService;\r
 import com.owncloud.android.ui.dialog.SslValidatorDialog;\r
 import com.owncloud.android.ui.dialog.SslValidatorDialog.OnSslValidatorListener;\r
 import com.owncloud.android.network.OwnCloudClientUtils;\r
@@ -40,9 +48,12 @@ import android.accounts.AccountManager;
 import android.app.AlertDialog;\r
 import android.app.Dialog;\r
 import android.app.ProgressDialog;\r
+import android.content.BroadcastReceiver;\r
 import android.content.ContentResolver;\r
+import android.content.Context;\r
 import android.content.DialogInterface;\r
 import android.content.Intent;\r
+import android.content.IntentFilter;\r
 import android.content.SharedPreferences;\r
 import android.net.Uri;\r
 import android.os.Bundle;\r
@@ -54,6 +65,8 @@ import android.view.View;
 import android.view.View.OnClickListener;\r
 import android.view.View.OnFocusChangeListener;\r
 import android.view.Window;\r
+import android.widget.CheckBox;\r
+import android.widget.EditText;\r
 import android.widget.ImageView;\r
 import android.widget.TextView;\r
 import com.owncloud.android.R;\r
@@ -68,7 +81,7 @@ import eu.alefzero.webdav.WebdavClient;
  */\r
 public class AuthenticatorActivity extends AccountAuthenticatorActivity\r
         implements OnAuthenticationResultListener, OnConnectCheckListener, OnRemoteOperationListener, OnSslValidatorListener, \r
-        OnFocusChangeListener, OnClickListener {\r
+        OnFocusChangeListener, OnClickListener, OnOAuth2GetCodeResultListener {\r
 \r
     private static final int DIALOG_LOGIN_PROGRESS = 0;\r
     private static final int DIALOG_SSL_VALIDATOR = 1;\r
@@ -94,6 +107,30 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
     public static final String PARAM_USERNAME = "param_Username";\r
     public static final String PARAM_HOSTNAME = "param_Hostname";\r
 \r
+    // oAuth2 variables.\r
+    private static final int OAUTH2_LOGIN_PROGRESS = 3;\r
+    private static final String OAUTH2_STATUS_TEXT = "OAUTH2_STATUS_TEXT";\r
+    private static final String OAUTH2_STATUS_ICON = "OAUTH2_STATUS_ICON";\r
+    private static final String OAUTH2_CODE_RESULT = "CODE_RESULT";\r
+    private static final String OAUTH2_BASE_URL = "BASE_URL"; \r
+    private static final String OAUTH2_IS_CHECKED = "OAUTH2_IS_CHECKED";    \r
+    private Thread mOAuth2GetCodeThread;\r
+    private OAuth2GetCodeRunnable mOAuth2GetCodeRunnable;     \r
+    private String oAuth2BaseUrl;\r
+    private TokenReceiver tokenReceiver;\r
+    private JSONObject codeResponseJson; \r
+    private int mOAuth2StatusText, mOAuth2StatusIcon;    \r
+    \r
+    public ConnectorOAuth2 connectorOAuth2;\r
+    \r
+    // Variables used to save the on the state the contents of all fields.\r
+    private static final String HOST_URL_TEXT = "HOST_URL_TEXT";\r
+    private static final String OAUTH2_URL_TEXT = "OAUTH2_URL_TEXT";\r
+    private static final String ACCOUNT_USERNAME = "ACCOUNT_USERNAME";\r
+    private static final String ACCOUNT_PASSWORD = "ACCOUNT_PASSWORD";\r
+\r
+    // END of oAuth2 variables.\r
+    \r
     @Override\r
     protected void onCreate(Bundle savedInstanceState) {\r
         super.onCreate(savedInstanceState);\r
@@ -103,6 +140,8 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
         ImageView iv2 = (ImageView) findViewById(R.id.viewPassword);\r
         TextView tv = (TextView) findViewById(R.id.host_URL);\r
         TextView tv2 = (TextView) findViewById(R.id.account_password);\r
+        // New textview to oAuth2 URL.\r
+        TextView tv3 = (TextView) findViewById(R.id.oAuth_URL);\r
 \r
         if (savedInstanceState != null) {\r
             mStatusIcon = savedInstanceState.getInt(STATUS_ICON);\r
@@ -114,7 +153,35 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
             if (!mStatusCorrect)\r
                 iv.setVisibility(View.VISIBLE);\r
             else\r
-                iv.setVisibility(View.INVISIBLE);\r
+                iv.setVisibility(View.INVISIBLE);            \r
+            \r
+            // Getting the state of oAuth2 components.\r
+            mOAuth2StatusIcon = savedInstanceState.getInt(OAUTH2_STATUS_ICON);\r
+            mOAuth2StatusText = savedInstanceState.getInt(OAUTH2_STATUS_TEXT);\r
+                // We set this to true if the rotation happens when the user is validating oAuth2 user_code.\r
+            changeViewByOAuth2Check(savedInstanceState.getBoolean(OAUTH2_IS_CHECKED));\r
+            oAuth2BaseUrl = savedInstanceState.getString(OAUTH2_BASE_URL);\r
+                // We store a JSon object with all the data returned from oAuth2 server when we get user_code.\r
+                // Is better than store variable by variable. We use String object to serialize from/to it.\r
+            try {\r
+                if (savedInstanceState.containsKey(OAUTH2_CODE_RESULT)) {\r
+                    codeResponseJson = new JSONObject(savedInstanceState.getString(OAUTH2_CODE_RESULT));\r
+                }\r
+            } catch (JSONException e) {\r
+                Log.e(TAG, "onCreate->JSONException: " + e.toString());\r
+            }\r
+            // END of getting the state of oAuth2 components.\r
+            \r
+            // Getting contents of each field.\r
+            EditText hostUrl = (EditText)findViewById(R.id.host_URL);\r
+            hostUrl.setText(savedInstanceState.getString(HOST_URL_TEXT), TextView.BufferType.EDITABLE);\r
+            EditText oauth2Url = (EditText)findViewById(R.id.oAuth_URL);\r
+            oauth2Url.setText(savedInstanceState.getString(OAUTH2_URL_TEXT), TextView.BufferType.EDITABLE);\r
+            EditText accountUsername = (EditText)findViewById(R.id.account_username);\r
+            accountUsername.setText(savedInstanceState.getString(ACCOUNT_USERNAME), TextView.BufferType.EDITABLE);\r
+            EditText accountPassword = (EditText)findViewById(R.id.account_password);\r
+            accountPassword.setText(savedInstanceState.getString(ACCOUNT_PASSWORD), TextView.BufferType.EDITABLE);\r
+            // END of getting contents of each field\r
 \r
         } else {\r
             mStatusText = mStatusIcon = 0;\r
@@ -125,6 +192,10 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
         iv2.setOnClickListener(this);\r
         tv.setOnFocusChangeListener(this);\r
         tv2.setOnFocusChangeListener(this);\r
+        // Setting the listener for oAuth2 URL TextView.\r
+        tv3.setOnFocusChangeListener(this);\r
+        \r
+        super.onCreate(savedInstanceState);\r
     }\r
 \r
     @Override\r
@@ -132,6 +203,24 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
         outState.putInt(STATUS_ICON, mStatusIcon);\r
         outState.putInt(STATUS_TEXT, mStatusText);\r
         outState.putBoolean(STATUS_CORRECT, mStatusCorrect);\r
+        \r
+        // Saving the state of oAuth2 components.\r
+        outState.putInt(OAUTH2_STATUS_ICON, mOAuth2StatusIcon);\r
+        outState.putInt(OAUTH2_STATUS_TEXT, mOAuth2StatusText);\r
+        CheckBox oAuth2Check = (CheckBox) findViewById(R.id.oauth_onOff_check);\r
+        outState.putBoolean(OAUTH2_IS_CHECKED, oAuth2Check.isChecked());\r
+        if (codeResponseJson != null){\r
+            outState.putString(OAUTH2_CODE_RESULT, codeResponseJson.toString());\r
+        }\r
+        outState.putString(OAUTH2_BASE_URL, oAuth2BaseUrl);\r
+        // END of saving the state of oAuth2 components.\r
+        \r
+        // Saving contents of each field.\r
+        outState.putString(HOST_URL_TEXT,((TextView) findViewById(R.id.host_URL)).getText().toString().trim());\r
+        outState.putString(OAUTH2_URL_TEXT,((TextView) findViewById(R.id.oAuth_URL)).getText().toString().trim());\r
+        outState.putString(ACCOUNT_USERNAME,((TextView) findViewById(R.id.account_username)).getText().toString().trim());\r
+        outState.putString(ACCOUNT_PASSWORD,((TextView) findViewById(R.id.account_password)).getText().toString().trim());\r
+        \r
         super.onSaveInstanceState(outState);\r
     }\r
 \r
@@ -159,6 +248,37 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
             dialog = working_dialog;\r
             break;\r
         }\r
+        // oAuth2 dialog. We show here to the user the URL and user_code that the user must validate in a web browser.\r
+        case OAUTH2_LOGIN_PROGRESS: {\r
+            ProgressDialog working_dialog = new ProgressDialog(this);\r
+            try {\r
+                working_dialog.setMessage(String.format(getString(R.string.oauth_code_validation_message), \r
+                        codeResponseJson.getString(OAuth2GetCodeRunnable.CODE_VERIFICATION_URL), \r
+                        codeResponseJson.getString(OAuth2GetCodeRunnable.CODE_USER_CODE)));\r
+            } catch (JSONException e) {\r
+                Log.e(TAG, "onCreateDialog->JSONException: " + e.toString());\r
+            }\r
+            working_dialog.setIndeterminate(true);\r
+            working_dialog.setCancelable(true);\r
+            working_dialog\r
+            .setOnCancelListener(new DialogInterface.OnCancelListener() {\r
+                @Override\r
+                public void onCancel(DialogInterface dialog) {\r
+                    Log.i(TAG, "Login canceled");\r
+                    if (mOAuth2GetCodeThread != null) {\r
+                        mOAuth2GetCodeThread.interrupt();\r
+                        finish();\r
+                    } \r
+                    if (tokenReceiver != null) {\r
+                        unregisterReceiver(tokenReceiver);\r
+                        tokenReceiver = null;\r
+                        finish();\r
+                    }\r
+                }\r
+            });\r
+            dialog = working_dialog;\r
+            break;\r
+        }\r
         case DIALOG_SSL_VALIDATOR: {\r
             dialog = SslValidatorDialog.newInstance(this, mLastSslUntrustedServerResult, this);\r
             break;\r
@@ -196,6 +316,25 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
             Log.e(TAG, "Incorrect dialog called with id = " + id);\r
         }\r
     }\r
+    \r
+    @Override\r
+    protected void onResume() {\r
+        Log.d(TAG, "onResume() start");\r
+        // Registering token receiver. We must listening to the service that is pooling to the oAuth server for a token.\r
+        if (tokenReceiver == null) {\r
+            IntentFilter tokenFilter = new IntentFilter(OAuth2GetTokenService.TOKEN_RECEIVED_MESSAGE);                \r
+            tokenReceiver = new TokenReceiver();\r
+            this.registerReceiver(tokenReceiver,tokenFilter);\r
+        }\r
+        super.onResume();\r
+    }\r
+\r
+    @Override\r
+    protected void onPause() {\r
+        Log.d(TAG, "onPause() start");\r
+        super.onPause();\r
+    }    \r
+    \r
 \r
     public void onAuthenticationResult(boolean success, String message) {\r
         if (success) {\r
@@ -240,6 +379,8 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
                     AccountAuthenticator.ACCOUNT_TYPE);\r
             intent.putExtra(AccountManager.KEY_USERDATA, username);\r
 \r
+            accManager.setUserData(account, AccountAuthenticator.KEY_OC_URL,\r
+                    url.toString());\r
             accManager.setUserData(account,\r
                     AccountAuthenticator.KEY_OC_VERSION, mConnChkRunnable\r
                             .getDiscoveredVersion().toString());\r
@@ -452,6 +593,28 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
                 v.setInputType(input_type);\r
                 iv.setVisibility(View.INVISIBLE);\r
             }\r
+        // If the focusChange occurs on the oAuth2 URL field, we do this.\r
+        } else if (view.getId() == R.id.oAuth_URL) {\r
+            if (!hasFocus) {\r
+                TextView tv3 = ((TextView) findViewById(R.id.oAuth_URL));\r
+                // We get the URL of oAuth2 server.\r
+                oAuth2BaseUrl = tv3.getText().toString().trim();\r
+                if (oAuth2BaseUrl.length() != 0) {\r
+                    // We start a thread to get user_code from the oAuth2 server.\r
+                    setOAuth2ResultIconAndText(R.drawable.progress_small, R.string.oauth_login_connection);\r
+                    mOAuth2GetCodeRunnable = new OAuth2GetCodeRunnable(oAuth2BaseUrl, this);\r
+                    mOAuth2GetCodeRunnable.setListener(this, mHandler);\r
+                    mOAuth2GetCodeThread = new Thread(mOAuth2GetCodeRunnable);\r
+                    mOAuth2GetCodeThread.start();\r
+                } else {\r
+                    findViewById(R.id.refreshButton).setVisibility(\r
+                            View.INVISIBLE);\r
+                    setOAuth2ResultIconAndText(0, 0);\r
+                }\r
+            } else {\r
+                // avoids that the 'connect' button can be clicked if the test was previously passed\r
+                findViewById(R.id.buttonOK).setEnabled(false); \r
+            }\r
         }\r
     }\r
 \r
@@ -481,6 +644,151 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
             view.setInputType(input_type);\r
         }\r
     }\r
+    \r
+    @Override protected void onDestroy() {       \r
+        // We must stop the service thats it's pooling to oAuth2 server for a token.\r
+        Intent tokenService = new Intent(this, OAuth2GetTokenService.class);\r
+        stopService(tokenService);\r
+        \r
+        // We stop listening the result of the pooling service.\r
+        if (tokenReceiver != null) {\r
+            unregisterReceiver(tokenReceiver);\r
+            tokenReceiver = null;\r
+            finish();\r
+        }\r
+\r
+        super.onDestroy();\r
+    }    \r
+    \r
+    // Controlling the oAuth2 checkbox on the activity: hide and show widgets.\r
+    public void onOff_check_Click(View view) {\r
+        CheckBox oAuth2Check = (CheckBox)view;      \r
+        changeViewByOAuth2Check(oAuth2Check.isChecked());\r
+\r
+    }\r
+    \r
+    public void changeViewByOAuth2Check(Boolean checked) {\r
+        \r
+        EditText oAuth2Url = (EditText) findViewById(R.id.oAuth_URL);\r
+        EditText accountUsername = (EditText) findViewById(R.id.account_username);\r
+        EditText accountPassword = (EditText) findViewById(R.id.account_password);\r
+        ImageView viewPassword = (ImageView) findViewById(R.id.viewPassword); \r
+        ImageView auth2ActionIndicator = (ImageView) findViewById(R.id.auth2_action_indicator); \r
+        TextView oauth2StatusText = (TextView) findViewById(R.id.oauth2_status_text);         \r
+\r
+        if (checked) {\r
+            oAuth2Url.setVisibility(View.VISIBLE);\r
+            accountUsername.setVisibility(View.GONE);\r
+            accountPassword.setVisibility(View.GONE);\r
+            viewPassword.setVisibility(View.GONE);\r
+            auth2ActionIndicator.setVisibility(View.INVISIBLE);\r
+            oauth2StatusText.setVisibility(View.INVISIBLE);\r
+        } else {\r
+            oAuth2Url.setVisibility(View.GONE);\r
+            accountUsername.setVisibility(View.VISIBLE);\r
+            accountPassword.setVisibility(View.VISIBLE);\r
+            viewPassword.setVisibility(View.INVISIBLE);\r
+            auth2ActionIndicator.setVisibility(View.GONE);\r
+            oauth2StatusText.setVisibility(View.GONE);\r
+        }     \r
+\r
+    }    \r
+    \r
+    // Controlling the oAuth2 result of server connection.\r
+    private void setOAuth2ResultIconAndText(int drawable_id, int text_id) {\r
+        ImageView iv = (ImageView) findViewById(R.id.auth2_action_indicator);\r
+        TextView tv = (TextView) findViewById(R.id.oauth2_status_text);\r
+\r
+        if (drawable_id == 0 && text_id == 0) {\r
+            iv.setVisibility(View.INVISIBLE);\r
+            tv.setVisibility(View.INVISIBLE);\r
+        } else {\r
+            iv.setImageResource(drawable_id);\r
+            tv.setText(text_id);\r
+            iv.setVisibility(View.VISIBLE);\r
+            tv.setVisibility(View.VISIBLE);\r
+        }\r
+    }     \r
+    \r
+    // Results from the first call to oAuth2 server : getting the user_code and verification_url.\r
+    @Override\r
+    public void onOAuth2GetCodeResult(ResultOAuthType type, JSONObject responseJson) {\r
+        if ((type == ResultOAuthType.OK_SSL)||(type == ResultOAuthType.OK_NO_SSL)) {\r
+            codeResponseJson = responseJson;\r
+            startOAuth2Authentication();\r
+        } else if (type == ResultOAuthType.HOST_NOT_AVAILABLE) {\r
+            setOAuth2ResultIconAndText(R.drawable.common_error, R.string.oauth_connection_url_unavailable);\r
+        }\r
+    }\r
+\r
+    // If the results of getting the user_code and verification_url are OK, we get the received data and we start\r
+    // the pooling service to oAuth2 server to get a valid token.\r
+    private void startOAuth2Authentication () {\r
+        String deviceCode = null;\r
+        String verificationUrl = null;\r
+        String userCode = null;\r
+        int expiresIn = -1;\r
+        int interval = -1;\r
+\r
+        Log.d(TAG, "ResponseOAuth2->" + codeResponseJson.toString());\r
+\r
+        try {\r
+            // We get data that we must show to the user or we will use internally.\r
+            verificationUrl = codeResponseJson.getString(OAuth2GetCodeRunnable.CODE_VERIFICATION_URL);\r
+            userCode = codeResponseJson.getString(OAuth2GetCodeRunnable.CODE_USER_CODE);\r
+            expiresIn = codeResponseJson.getInt(OAuth2GetCodeRunnable.CODE_EXPIRES_IN);                \r
+\r
+            // And we get data that we must use to get a token.\r
+            deviceCode = codeResponseJson.getString(OAuth2GetCodeRunnable.CODE_DEVICE_CODE);\r
+            interval = codeResponseJson.getInt(OAuth2GetCodeRunnable.CODE_INTERVAL);\r
+\r
+        } catch (JSONException e) {\r
+            Log.e(TAG, "Exception accesing data in Json object" + e.toString());\r
+        }\r
+\r
+        // Updating status widget to OK.\r
+        setOAuth2ResultIconAndText(R.drawable.ic_ok, R.string.auth_connection_established);\r
+        \r
+        // Showing the dialog with instructions for the user.\r
+        showDialog(OAUTH2_LOGIN_PROGRESS);\r
+\r
+        // Loggin all the data.\r
+        Log.d(TAG, "verificationUrl->" + verificationUrl);\r
+        Log.d(TAG, "userCode->" + userCode);\r
+        Log.d(TAG, "deviceCode->" + deviceCode);\r
+        Log.d(TAG, "expiresIn->" + expiresIn);\r
+        Log.d(TAG, "interval->" + interval);\r
+\r
+        // Starting the pooling service.\r
+        try {\r
+            Intent tokenService = new Intent(this, OAuth2GetTokenService.class);\r
+            tokenService.putExtra(OAuth2GetTokenService.TOKEN_BASE_URI, oAuth2BaseUrl);\r
+            tokenService.putExtra(OAuth2GetTokenService.TOKEN_DEVICE_CODE, deviceCode);\r
+            tokenService.putExtra(OAuth2GetTokenService.TOKEN_INTERVAL, interval);\r
+\r
+            startService(tokenService);\r
+        }\r
+        catch (Exception e) {\r
+            Log.e(TAG, "tokenService creation problem :", e);\r
+        }\r
+    }   \r
+\r
+    // We get data from the oAuth2 token service with this broadcast receiver.\r
+    private class TokenReceiver extends BroadcastReceiver {\r
+        /**\r
+         * The token is received.\r
+         *  @author\r
+         * {@link BroadcastReceiver} to enable oAuth2 token receiving.\r
+         */\r
+        @Override\r
+        public void onReceive(Context context, Intent intent) {\r
+            @SuppressWarnings("unchecked")\r
+            HashMap<String, String> tokenResponse = (HashMap<String, String>)intent.getExtras().get(OAuth2GetTokenService.TOKEN_RECEIVED_DATA);\r
+            Log.d(TAG, "TokenReceiver->" + tokenResponse.get(OAuth2GetTokenService.TOKEN_ACCESS_TOKEN));\r
+            dismissDialog(OAUTH2_LOGIN_PROGRESS);\r
+\r
+        }\r
+    }\r
 \r
        @Override\r
        public void onRemoteOperationFinish(RemoteOperation operation, RemoteOperationResult result) {\r
@@ -584,5 +892,5 @@ public class AuthenticatorActivity extends AccountAuthenticatorActivity
     public void onFailedSavingCertificate() {\r
         showDialog(DIALOG_CERT_NOT_SAVED);\r
     }\r
-    \r
+\r
 }\r