1 files changed, 104 insertions, 0 deletions
diff --git a/src/org/apg/DataSource.java b/src/org/apg/DataSource.java
new file mode 100644
index 000000000..340afa9f8
--- /dev/null
+++ b/src/org/apg/DataSource.java
@@ -0,0 +1,104 @@
+package org.apg;
+
+import java.io.ByteArrayInputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.InputStream;
+
+import org.apg.Apg.GeneralException;
+import org.apg.R;
+
+import android.content.Context;
+import android.net.Uri;
+import android.os.Environment;
+
+public class DataSource {
+    private Uri mContentUri = null;
+    private String mText = null;
+    private byte[] mData = null;
+
+    public DataSource() {
+
+    }
+
+    public void setUri(Uri uri) {
+        mContentUri = uri;
+        mText = null;
+        mData = null;
+    }
+
+    public void setUri(String uri) {
+        if (uri.startsWith("/")) {
+            setUri(Uri.parse("file://" + uri));
+        } else {
+            setUri(Uri.parse(uri));
+        }
+    }
+
+    public void setText(String text) {
+        mText = text;
+        mData = null;
+        mContentUri = null;
+    }
+
+    public void setData(byte[] data) {
+        mData = data;
+        mText = null;
+        mContentUri = null;
+    }
+
+    public boolean isText() {
+        return mText != null;
+    }
+
+    public boolean isBinary() {
+        return mData != null || mContentUri != null;
+    }
+
+    public InputData getInputData(Context context, boolean withSize) throws GeneralException,
+            FileNotFoundException, IOException {
+        InputStream in = null;
+        long size = 0;
+
+        if (mContentUri != null) {
+            if (mContentUri.getScheme().equals("file")) {
+                // get the rest after "file://"
+                String path = Uri.decode(mContentUri.toString().substring(7));
+                if (path.startsWith(Environment.getExternalStorageDirectory().getAbsolutePath())) {
+                    if (!Environment.getExternalStorageState().equals(Environment.MEDIA_MOUNTED)) {
+                        throw new GeneralException(
+                                context.getString(R.string.error_externalStorageNotReady));
+                    }
+                }
+                in = new FileInputStream(path);
+                File file = new File(path);
+                if (withSize) {
+                    size = file.length();
+                }
+            } else {
+                in = context.getContentResolver().openInputStream(mContentUri);
+                if (withSize) {
+                    InputStream tmp = context.getContentResolver().openInputStream(mContentUri);
+                    size = Apg.getLengthOfStream(tmp);
+                    tmp.close();
+                }
+            }
+        } else if (mText != null || mData != null) {
+            byte[] bytes = null;
+            if (mData != null) {
+                bytes = mData;
+            } else {
+                bytes = mText.getBytes();
+            }
+            in = new ByteArrayInputStream(bytes);
+            if (withSize) {
+                size = bytes.length;
+            }
+        }
+
+        return new InputData(in, size);
+    }
+
+}