1 files changed, 78 insertions, 0 deletions
diff --git a/mediapicker/src/main/kotlin/ca/allanwang/kau/imagepicker/MediaModel.kt b/mediapicker/src/main/kotlin/ca/allanwang/kau/imagepicker/MediaModel.kt
new file mode 100644
index 0000000..c384d48
--- /dev/null
+++ b/mediapicker/src/main/kotlin/ca/allanwang/kau/imagepicker/MediaModel.kt
@@ -0,0 +1,78 @@
+package ca.allanwang.kau.imagepicker
+
+import android.database.Cursor
+import android.database.SQLException
+import android.net.Uri
+import android.os.Parcel
+import android.os.Parcelable
+import android.provider.MediaStore
+import android.support.annotation.NonNull
+import java.io.File
+
+
+/**
+ * Created by Allan Wang on 2017-07-14.
+ */
+
+data class MediaModel(
+        val data: String, val mimeType: String, val size: Long, val dateModified: Long, val displayName: String?
+) : Parcelable {
+
+    @Throws(SQLException::class)
+    constructor(@NonNull cursor: Cursor) : this(
+            cursor.getString(0),
+            cursor.getString(1),
+            cursor.getLong(2),
+            cursor.getLong(3),
+            cursor.getString(4)
+    )
+
+    constructor(parcel: Parcel) : this(
+            parcel.readString(),
+            parcel.readString(),
+            parcel.readLong(),
+            parcel.readLong(),
+            parcel.readString())
+
+    override fun writeToParcel(parcel: Parcel, flags: Int) {
+        parcel.writeString(this.data)
+        parcel.writeString(this.mimeType)
+        parcel.writeLong(this.size)
+        parcel.writeLong(this.dateModified)
+        parcel.writeString(this.displayName)
+    }
+
+    val isGif
+        get() = mimeType.endsWith("gif")
+
+    val isImage
+        get() = mimeType.endsWith("image")
+
+    val isVideo
+        get() = mimeType.endsWith("video")
+
+    val uri: Uri by lazy { Uri.fromFile(File(data)) }
+
+    override fun describeContents(): Int {
+        return 0
+    }
+
+    companion object CREATOR : Parcelable.Creator<MediaModel> {
+        val projection = arrayOf(
+                MediaStore.MediaColumns.DATA,
+                MediaStore.MediaColumns.MIME_TYPE,
+                MediaStore.MediaColumns.SIZE,
+                MediaStore.MediaColumns.DATE_MODIFIED,
+                MediaStore.MediaColumns.DISPLAY_NAME
+        )
+
+        override fun createFromParcel(parcel: Parcel): MediaModel {
+            return MediaModel(parcel)
+        }
+
+        override fun newArray(size: Int): Array<MediaModel?> {
+            return arrayOfNulls(size)
+        }
+    }
+
+}
+\ No newline at end of file