1 files changed, 20 insertions, 10 deletions
diff --git a/src/ryuslash/org/dvdroid/DVDroidActivity.java b/src/ryuslash/org/dvdroid/DVDroidActivity.java
index 20c1ad0..5a46286 100644
--- a/src/ryuslash/org/dvdroid/DVDroidActivity.java
+++ b/src/ryuslash/org/dvdroid/DVDroidActivity.java
@@ -4,6 +4,7 @@ import java.util.List;
 
 import android.app.ListActivity;
 import android.content.Context;
+import android.content.Intent;
 import android.os.Bundle;
 import android.view.View;
 import android.widget.ArrayAdapter;
@@ -35,17 +36,10 @@ public class DVDroidActivity extends ListActivity
 
     public void addDVD(View view)
     {
-        ArrayAdapter<DVD> adapter = (ArrayAdapter<DVD>)getListAdapter();
-        EditText editText = (EditText)findViewById(R.id.name_entry);
-        DVD dvd = datasource.createDVD(editText.getText().toString());
-        int duration = Toast.LENGTH_SHORT;
-        Toast toast = Toast.makeText(this, "Added " + editText.getText(),
-                                     duration);
+        Intent intent = new Intent(this, AddDVDActivity.class);
 
-        editText.setText("");
-        adapter.add(dvd);
-        adapter.notifyDataSetChanged();
-        toast.show();
+        datasource.close();
+        startActivityForResult(intent, 1);
     }
 
     @Override
@@ -61,4 +55,20 @@ public class DVDroidActivity extends ListActivity
         datasource.close();
         super.onPause();
     }
+
+    @Override
+    protected void onActivityResult(int requestCode,
+                                    int resultCode,
+                                    Intent data)
+    {
+        super.onActivityResult(requestCode, resultCode, data);
+        datasource.open();
+
+        ArrayAdapter<DVD> adapter = (ArrayAdapter<DVD>)getListAdapter();
+        List<DVD> dvds = datasource.getWishlist();
+
+        adapter.clear();
+        adapter.addAll(dvds);
+        adapter.notifyDataSetChanged();
+    }
 }