Move several classes to other packages.

1 files changed, 48 insertions, 0 deletions

diff --git a/app/src/main/java/SubredditDatabase/SubredditViewModel.java b/app/src/main/java/SubredditDatabase/SubredditViewModel.java
new file mode 100644
index 00000000..10135401
--- /dev/null
+++ b/app/src/main/java/SubredditDatabase/SubredditViewModel.java
@@ -0,0 +1,48 @@
+package SubredditDatabase;
+
+import android.app.Application;
+import android.arch.lifecycle.AndroidViewModel;
+import android.arch.lifecycle.LiveData;
+import android.arch.lifecycle.ViewModel;
+import android.arch.lifecycle.ViewModelProvider;
+import android.support.annotation.NonNull;
+
+public class SubredditViewModel extends AndroidViewModel {
+    private SubredditRepository mSubredditRepository;
+    private LiveData<SubredditData> mSubredditLiveData;
+
+    SubredditViewModel(Application application, String id, boolean isId) {
+        super(application);
+        mSubredditRepository = new SubredditRepository(application, id, isId);
+        mSubredditLiveData = mSubredditRepository.getSubredditLiveData();
+    }
+
+    public LiveData<SubredditData> getSubredditLiveData() {
+        return mSubredditLiveData;
+    }
+
+    public void insert(SubredditData subredditData) {
+        mSubredditRepository.insert(subredditData);
+    }
+
+    public static class Factory extends ViewModelProvider.NewInstanceFactory {
+
+        @NonNull
+        private final Application mApplication;
+
+        private final String value;
+        private final boolean isId;
+
+        public Factory(@NonNull Application application, String value, boolean isId) {
+            mApplication = application;
+            this.value = value;
+            this.isId = isId;
+        }
+
+        @Override
+        public <T extends ViewModel> T create(Class<T> modelClass) {
+            //noinspection unchecked
+            return (T) new SubredditViewModel(mApplication, value, isId);
+        }
+    }
+}