Preparing to support multi user. Use the database to store accounts' info. LoginActivity is successfully refactored. Any other features are unavailable for now.

8 files changed, 337 insertions, 21 deletions

diff --git a/app/src/main/java/Account/Account.java b/app/src/main/java/Account/Account.java
new file mode 100644
index 00000000..95873c61
--- /dev/null
+++ b/app/src/main/java/Account/Account.java
@@ -0,0 +1,72 @@
+package Account;
+
+import androidx.annotation.NonNull;
+import androidx.room.ColumnInfo;
+import androidx.room.Entity;
+import androidx.room.PrimaryKey;
+
+@Entity(tableName = "accounts")
+public class Account {
+    @PrimaryKey
+    @NonNull
+    @ColumnInfo(name = "username")
+    private String username;
+    @ColumnInfo(name = "profile_image_url")
+    private String profileImageUrl;
+    @ColumnInfo(name = "banner_image_url")
+    private String bannerImageUrl;
+    @ColumnInfo(name = "karma")
+    private int karma;
+    @ColumnInfo(name = "access_token")
+    private String accessToken;
+    @ColumnInfo(name = "refresh_token")
+    private String refreshToken;
+    @ColumnInfo(name = "code")
+    private String code;
+    @ColumnInfo(name = "is_current_user")
+    private boolean isCurrentUser;
+
+    public Account(@NonNull String username, String accessToken, String refreshToken, String code,
+                   String profileImageUrl, String bannerImageUrl, int karma, boolean isCurrentUser) {
+        this.username = username;
+        this.accessToken = accessToken;
+        this.refreshToken = refreshToken;
+        this.code = code;
+        this.profileImageUrl = profileImageUrl;
+        this.bannerImageUrl = bannerImageUrl;
+        this.karma = karma;
+        this.isCurrentUser = isCurrentUser;
+    }
+
+    public String getUsername() {
+        return username;
+    }
+
+    public String getProfileImageUrl() {
+        return profileImageUrl;
+    }
+
+    public String getBannerImageUrl() {
+        return bannerImageUrl;
+    }
+
+    public int getKarma() {
+        return karma;
+    }
+
+    public String getAccessToken() {
+        return accessToken;
+    }
+
+    public String getRefreshToken() {
+        return refreshToken;
+    }
+
+    public String getCode() {
+        return code;
+    }
+
+    public boolean isCurrentUser() {
+        return isCurrentUser;
+    }
+}
diff --git a/app/src/main/java/Account/AccountDao.java b/app/src/main/java/Account/AccountDao.java
new file mode 100644
index 00000000..bd53b92b
--- /dev/null
+++ b/app/src/main/java/Account/AccountDao.java
@@ -0,0 +1,22 @@
+package Account;
+
+import androidx.lifecycle.LiveData;
+import androidx.room.Dao;
+import androidx.room.Insert;
+import androidx.room.OnConflictStrategy;
+import androidx.room.Query;
+
+@Dao
+public interface AccountDao {
+    @Insert(onConflict = OnConflictStrategy.REPLACE)
+    void insert(Account account);
+
+    @Query("DELETE FROM accounts")
+    void deleteAllAccounts();
+
+    @Query("SELECT * FROM accounts WHERE username = :userName COLLATE NOCASE LIMIT 1")
+    LiveData<Account> getAccountLiveData(String userName);
+
+    @Query("SELECT * FROM accounts WHERE username = :userName COLLATE NOCASE LIMIT 1")
+    Account getAccountData(String userName);
+}
diff --git a/app/src/main/java/Account/AccountRepository.java b/app/src/main/java/Account/AccountRepository.java
new file mode 100644
index 00000000..49bdfacf
--- /dev/null
+++ b/app/src/main/java/Account/AccountRepository.java
@@ -0,0 +1,40 @@
+package Account;
+
+import android.app.Application;
+import android.os.AsyncTask;
+
+import androidx.lifecycle.LiveData;
+
+public class AccountRepository {
+    private AccountDao mAccountDao;
+    private LiveData<Account> mAccountLiveData;
+
+    AccountRepository(Application application, String username) {
+        mAccountDao = AccountRoomDatabase.getDatabase(application).accountDao();
+
+        mAccountLiveData = mAccountDao.getAccountLiveData(username);
+    }
+
+    LiveData<Account> getAccountLiveData() {
+        return mAccountLiveData;
+    }
+
+    public void insert(Account Account) {
+        new InsertAsyncTask(mAccountDao).execute(Account);
+    }
+
+    private static class InsertAsyncTask extends AsyncTask<Account, Void, Void> {
+
+        private AccountDao mAsyncTaskDao;
+
+        InsertAsyncTask(AccountDao dao) {
+            mAsyncTaskDao = dao;
+        }
+
+        @Override
+        protected Void doInBackground(final Account... params) {
+            mAsyncTaskDao.insert(params[0]);
+            return null;
+        }
+    }
+}
diff --git a/app/src/main/java/Account/AccountRoomDatabase.java b/app/src/main/java/Account/AccountRoomDatabase.java
new file mode 100644
index 00000000..2f5d0120
--- /dev/null
+++ b/app/src/main/java/Account/AccountRoomDatabase.java
@@ -0,0 +1,27 @@
+package Account;
+
+import android.content.Context;
+
+import androidx.room.Database;
+import androidx.room.Room;
+import androidx.room.RoomDatabase;
+
+@Database(entities = {Account.class}, version = 1)
+public abstract class AccountRoomDatabase extends RoomDatabase {
+    private static AccountRoomDatabase INSTANCE;
+
+    public abstract AccountDao accountDao();
+
+    public static AccountRoomDatabase getDatabase(final Context context) {
+        if(INSTANCE == null) {
+            synchronized (AccountRoomDatabase.class) {
+                if(INSTANCE == null) {
+                    INSTANCE = Room.databaseBuilder(context.getApplicationContext(),
+                            AccountRoomDatabase.class, "accounts")
+                            .build();
+                }
+            }
+        }
+        return INSTANCE;
+    }
+}
diff --git a/app/src/main/java/Account/AccountViewModel.java b/app/src/main/java/Account/AccountViewModel.java
new file mode 100644
index 00000000..b649d9a7
--- /dev/null
+++ b/app/src/main/java/Account/AccountViewModel.java
@@ -0,0 +1,47 @@
+package Account;
+
+import android.app.Application;
+
+import androidx.annotation.NonNull;
+import androidx.lifecycle.AndroidViewModel;
+import androidx.lifecycle.LiveData;
+import androidx.lifecycle.ViewModel;
+import androidx.lifecycle.ViewModelProvider;
+
+public class AccountViewModel extends AndroidViewModel {
+    private AccountRepository mAccountRepository;
+    private LiveData<Account> mAccountLiveData;
+
+    public AccountViewModel(Application application, String id) {
+        super(application);
+        mAccountRepository = new AccountRepository(application, id);
+        mAccountLiveData = mAccountRepository.getAccountLiveData();
+    }
+
+    public LiveData<Account> getAccountLiveData() {
+        return mAccountLiveData;
+    }
+
+    public void insert(Account userData) {
+        mAccountRepository.insert(userData);
+    }
+
+    public static class Factory extends ViewModelProvider.NewInstanceFactory {
+
+        @NonNull
+        private final Application mApplication;
+
+        private final String userName;
+
+        public Factory(@NonNull Application application, String userName) {
+            mApplication = application;
+            this.userName = userName;
+        }
+
+        @Override
+        public <T extends ViewModel> T create(Class<T> modelClass) {
+            //noinspection unchecked
+            return (T) new AccountViewModel(mApplication, userName);
+        }
+    }
+}
diff --git a/app/src/main/java/ml/docilealligator/infinityforreddit/FetchMyInfo.java b/app/src/main/java/ml/docilealligator/infinityforreddit/FetchMyInfo.java
index 393349a1..63b80532 100644
--- a/app/src/main/java/ml/docilealligator/infinityforreddit/FetchMyInfo.java
+++ b/app/src/main/java/ml/docilealligator/infinityforreddit/FetchMyInfo.java
@@ -1,9 +1,10 @@
 package ml.docilealligator.infinityforreddit;
 
 import android.content.SharedPreferences;
-import androidx.annotation.NonNull;
 import android.util.Log;
 
+import androidx.annotation.NonNull;
+
 import retrofit2.Call;
 import retrofit2.Callback;
 import retrofit2.Retrofit;
@@ -39,4 +40,28 @@ class FetchMyInfo {
             }
         });
     }
+
+    static void fetchMyInfo(final Retrofit retrofit, String accessToken,
+                            final FetchUserMyListener fetchUserMyListener) {
+        RedditAPI api = retrofit.create(RedditAPI.class);
+
+        Call<String> userInfo = api.getMyInfo(RedditUtils.getOAuthHeader(accessToken));
+        userInfo.enqueue(new Callback<String>() {
+            @Override
+            public void onResponse(@NonNull Call<String> call, @NonNull retrofit2.Response<String> response) {
+                if(response.isSuccessful()) {
+                    fetchUserMyListener.onFetchMyInfoSuccess(response.body());
+                } else {
+                    Log.i("call failed", response.message());
+                    fetchUserMyListener.onFetchMyInfoFail();
+                }
+            }
+
+            @Override
+            public void onFailure(@NonNull Call<String> call, @NonNull Throwable t) {
+                Log.i("call failed", t.getMessage());
+                fetchUserMyListener.onFetchMyInfoFail();
+            }
+        });
+    }
 }
diff --git a/app/src/main/java/ml/docilealligator/infinityforreddit/LoginActivity.java b/app/src/main/java/ml/docilealligator/infinityforreddit/LoginActivity.java
index 53b0e143..20fa44fa 100644
--- a/app/src/main/java/ml/docilealligator/infinityforreddit/LoginActivity.java
+++ b/app/src/main/java/ml/docilealligator/infinityforreddit/LoginActivity.java
@@ -12,6 +12,9 @@ import android.webkit.WebView;
 import android.webkit.WebViewClient;
 import android.widget.Toast;
 
+import androidx.annotation.NonNull;
+import androidx.appcompat.app.AppCompatActivity;
+
 import org.json.JSONException;
 import org.json.JSONObject;
 
@@ -21,8 +24,7 @@ import java.util.Map;
 import javax.inject.Inject;
 import javax.inject.Named;
 
-import androidx.annotation.NonNull;
-import androidx.appcompat.app.AppCompatActivity;
+import Account.AccountRoomDatabase;
 import retrofit2.Call;
 import retrofit2.Callback;
 import retrofit2.Response;
@@ -36,6 +38,10 @@ public class LoginActivity extends AppCompatActivity {
     @Named("no_oauth")
     Retrofit mRetrofit;
 
+    @Inject
+    @Named("oauth")
+    Retrofit mOauthRetrofit;
+
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
@@ -85,41 +91,69 @@ public class LoginActivity extends AppCompatActivity {
                             public void onResponse(@NonNull Call<String> call, @NonNull Response<String> response) {
                                 if(response.isSuccessful()) {
                                     try {
-                                        JSONObject responseJSON = new JSONObject(response.body());
+                                        String accountResponse = response.body();
+                                        if(accountResponse == null) {
+                                            //Handle error
+                                            return;
+                                        }
+
+                                        JSONObject responseJSON = new JSONObject(accountResponse);
                                         String accessToken = responseJSON.getString(RedditUtils.ACCESS_TOKEN_KEY);
                                         String refreshToken = responseJSON.getString(RedditUtils.REFRESH_TOKEN_KEY);
 
-                                        editor.putString(SharedPreferencesUtils.ACCESS_TOKEN_KEY, accessToken);
-                                        editor.putString(SharedPreferencesUtils.REFRESH_TOKEN_KEY, refreshToken);
-                                        editor.apply();
-
-                                        Intent resultIntent = new Intent();
-                                        setResult(Activity.RESULT_OK, resultIntent);
-                                        finish();
+                                        FetchMyInfo.fetchMyInfo(mOauthRetrofit, accessToken, new FetchMyInfo.FetchUserMyListener() {
+                                            @Override
+                                            public void onFetchMyInfoSuccess(String response) {
+                                                ParseMyInfo.parseMyInfo(response, new ParseMyInfo.ParseMyInfoListener() {
+                                                    @Override
+                                                    public void onParseMyInfoSuccess(String name, String profileImageUrl, String bannerImageUrl, int karma) {
+                                                        new ParseAndInsertAccount(name, accessToken, refreshToken, profileImageUrl, bannerImageUrl,
+                                                                karma, authCode, AccountRoomDatabase.getDatabase(LoginActivity.this).accountDao(),
+                                                                () -> {
+                                                                    Intent resultIntent = new Intent();
+                                                                    setResult(Activity.RESULT_OK, resultIntent);
+                                                                    finish();
+                                                                }).execute();
+                                                    }
+
+                                                    @Override
+                                                    public void onParseMyInfoFail() {
+                                                        Toast.makeText(LoginActivity.this, R.string.parse_user_info_error, Toast.LENGTH_SHORT).show();
+                                                        finish();
+                                                    }
+                                                });
+                                            }
+
+                                            @Override
+                                            public void onFetchMyInfoFail() {
+                                                Toast.makeText(LoginActivity.this, R.string.cannot_fetch_user_info, Toast.LENGTH_SHORT).show();
+                                                finish();
+                                            }
+                                        });
                                     } catch (JSONException e) {
                                         e.printStackTrace();
-                                        Toast.makeText(LoginActivity.this, "Error occurred when parsing the JSON response", Toast.LENGTH_SHORT).show();
+                                        Toast.makeText(LoginActivity.this, R.string.parse_json_response_error, Toast.LENGTH_SHORT).show();
                                         finish();
                                     }
                                 } else {
-                                    Toast.makeText(LoginActivity.this, "Error Retrieving the token", Toast.LENGTH_SHORT).show();
+                                    Toast.makeText(LoginActivity.this, R.string.retrieve_token_error, Toast.LENGTH_SHORT).show();
                                     finish();
                                 }
                             }
 
                             @Override
-                            public void onFailure(Call<String> call, Throwable t) {
-                                Toast.makeText(LoginActivity.this, "Error Retrieving the token", Toast.LENGTH_SHORT).show();
+                            public void onFailure(@NonNull Call<String> call, @NonNull Throwable t) {
+                                Toast.makeText(LoginActivity.this, R.string.retrieve_token_error, Toast.LENGTH_SHORT).show();
                                 finish();
                             }
                         });
                     } else {
-                        Toast.makeText(LoginActivity.this, "Something went wrong. Try again later.", Toast.LENGTH_SHORT).show();
+                        Toast.makeText(LoginActivity.this, R.string.something_went_wrong, Toast.LENGTH_SHORT).show();
                         finish();
                     }
 
                 } else if (url.contains("error=access_denied")) {
-                    Toast.makeText(LoginActivity.this, "Access denied", Toast.LENGTH_SHORT).show();
+                    Toast.makeText(LoginActivity.this, R.string.access_denied, Toast.LENGTH_SHORT).show();
                     finish();
                 } else {
                     view.loadUrl(url);
@@ -142,10 +176,9 @@ public class LoginActivity extends AppCompatActivity {
 
     @Override
     public boolean onOptionsItemSelected(MenuItem item) {
-        switch (item.getItemId()) {
-            case android.R.id.home:
-                finish();
-                return true;
+        if (item.getItemId() == android.R.id.home) {
+            finish();
+            return true;
         }
         return false;
     }
diff --git a/app/src/main/java/ml/docilealligator/infinityforreddit/ParseAndInsertAccount.java b/app/src/main/java/ml/docilealligator/infinityforreddit/ParseAndInsertAccount.java
new file mode 100644
index 00000000..e86fbada
--- /dev/null
+++ b/app/src/main/java/ml/docilealligator/infinityforreddit/ParseAndInsertAccount.java
@@ -0,0 +1,50 @@
+package ml.docilealligator.infinityforreddit;
+
+import android.os.AsyncTask;
+
+import Account.Account;
+import Account.AccountDao;
+
+class ParseAndInsertAccount extends AsyncTask<Void, Void, Void> {
+
+    interface ParseAndInsertAccountListener {
+        void success();
+    }
+
+    private String username;
+    private String accessToken;
+    private String refreshToken;
+    private String profileImageUrl;
+    private String bannerImageUrl;
+    private int karma;
+    private String code;
+    private AccountDao accountDao;
+    private ParseAndInsertAccountListener parseAndInsertAccountListener;
+
+    ParseAndInsertAccount(String username, String accessToken, String refreshToken, String profileImageUrl, String bannerImageUrl,
+                          int karma, String code, AccountDao accountDao,
+                          ParseAndInsertAccountListener parseAndInsertAccountListener) {
+        this.username = username;
+        this.accessToken = accessToken;
+        this.refreshToken = refreshToken;
+        this.profileImageUrl = profileImageUrl;
+        this.bannerImageUrl = bannerImageUrl;
+        this.karma = karma;
+        this.code = code;
+        this.accountDao = accountDao;
+        this.parseAndInsertAccountListener = parseAndInsertAccountListener;
+    }
+
+    @Override
+    protected Void doInBackground(Void... voids) {
+        Account account = new Account(username, accessToken, refreshToken, code, profileImageUrl,
+                bannerImageUrl, karma, true);
+        accountDao.insert(account);
+        return null;
+    }
+
+    @Override
+    protected void onPostExecute(Void aVoid) {
+        parseAndInsertAccountListener.success();
+    }
+}