package org.mariotaku.ktextension

import android.os.Bundle
import android.os.Parcelable

inline fun Bundle(copyFrom: Bundle? = null, action: Bundle.() -> Unit): Bundle {
    val bundle = Bundle()
    if (copyFrom != null) {
        bundle.putAll(copyFrom)
    }
    action(bundle)
    return bundle
}

operator fun Bundle.set(key: String, value: Boolean) {
    return putBoolean(key, value)
}

operator fun Bundle.set(key: String, value: Int) {
    return putInt(key, value)
}

operator fun Bundle.set(key: String, value: Long) {
    return putLong(key, value)
}

operator fun Bundle.set(key: String, value: String?) {
    return putString(key, value)
}

operator fun Bundle.set(key: String, value: IntArray?) {
    return putIntArray(key, value)
}

operator fun Bundle.set(key: String, value: Parcelable?) {
    return putParcelable(key, value)
}

operator fun Bundle.set(key: String, value: Array<out Parcelable>?) {
    return putParcelableArray(key, value)
}

operator fun Bundle.set(key: String, value: Array<String>?) {
    return putStringArray(key, value)
}

inline fun <reified T> Bundle.getTypedArray(key: String): Array<T> {
    val parcelable = getParcelableArray(key)
    return Array(parcelable.size) { parcelable[it] as T }
}

inline fun <reified T> Bundle.getNullableTypedArray(key: String): Array<T>? {
    val parcelable = getParcelableArray(key) ?: return null
    return Array(parcelable.size) { parcelable[it] as T }
}