com.skillw.asahi.util.AsahiDataMap.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Pouvoir Show documentation
Bukkit Script Engine Plugin.
There is a newer version: 1.6.7-beta-6
package com.skillw.asahi.util

import com.skillw.asahi.api.AsahiAPI.analysis
import com.skillw.asahi.api.member.context.AsahiContext
import taboolib.common5.Coerce
import kotlin.reflect.KProperty

/**
 * @className DataMap
 *
 * 专用于存储数据的Map
 *
 * @author Glom
 * @date 2022/8/14 7:45 Copyright 2022 user.
 */
open class AsahiDataMap(val analysis: Boolean = false) : AsahiContext by AsahiContext.create() {

    operator fun getValue(thisRef: Any?, property: KProperty<*>): Any? {
        return (thisRef as AsahiDataMap)[property.name]
    }

    operator fun setValue(thisRef: Any?, property: KProperty<*>, value: Any) {
        (thisRef as AsahiDataMap)[property.name] = value
    }

    /**
     * 获取任意对象
     *
     * @param key String 键
     * @param default T? 默认值
     * @return T 类型值11
     * @receiver DataMap
     */
    inline fun  get(key: String, default: T? = null): T {
        return getAs(key) ?: default ?: error(
            "get ${T::class.simpleName} error"
        )
    }

    inline fun  getAs(key: String): T? {
        return when (T::class) {
            String::class -> getStringInner(key) as? T?
            Int::class -> getIntInner(key) as? T?
            Long::class -> getLongInner(key) as? T?
            Float::class -> getFloatInner(key) as? T?
            Double::class -> getDoubleInner(key) as? T?
            Boolean::class -> getBooleanInner(key) as? T?
            Byte::class -> getByteInner(key) as? T?
            Short::class -> getShortInner(key) as? T?
            List::class -> getListInner(key) as? T?
            else -> get(key) as? T?
        }
    }

    /**
     * 获取任意对象
     *
     * @param key String 键
     * @param default T? 默认值
     * @return T 类型值11
     * @receiver DataMap
     */
    inline fun  getOrDefault(key: String, defaultFunc: () -> T? = { null }): T {
        return getAs(key) ?: defaultFunc() ?: error(
            "get ${T::class.simpleName} error"
        )
    }

    private fun Any.analysisCast(): Any {
        return if (this is String && analysis) analysis(this@AsahiDataMap) else this
    }

    /**
     * 根据键获取any
     *
     * @param key
     * @param processData
     * @return
     */
    fun getAny(key: String): Any {
        return get(key)
    }

    /**
     * 根据键获取list
     *
     * @param key
     * @param processData
     * @return
     */
    fun getList(key: String): List {
        return get>(key)
    }

    /**
     * 根据键获取short
     *
     * @param key
     * @param processData
     * @return
     */
    fun getShort(key: String): Short {
        return get(key)
    }

    /**
     * 根据键获取byte
     *
     * @param key
     * @param processData
     * @return
     */
    fun getByte(key: String): Byte {
        return get(key)
    }

    /**
     * 根据键获取boolean
     *
     * @param key
     * @param processData
     * @return
     */
    fun getBoolean(key: String): Boolean {
        return get(key)
    }

    /**
     * 根据键获取double
     *
     * @param key
     * @param processData
     * @return
     */
    fun getDouble(key: String): Double {
        return get(key)
    }

    /**
     * 根据键获取float
     *
     * @param key
     * @param processData
     * @return
     */
    fun getFloat(key: String): Float {
        return get(key)
    }

    /**
     * 根据键获取long
     *
     * @param key
     * @param processData
     * @return
     */
    fun getLong(key: String): Long {
        return get(key)
    }

    /**
     * 根据键获取int
     *
     * @param key
     * @param processData
     * @return
     */
    fun getInt(key: String): Int {
        return get(key)
    }

    /**
     * 根据键获取string
     *
     * @param key
     * @param processData
     * @return
     */
    fun getString(key: String): String {
        return get(key)
    }

    companion object {

        /**
         * 根据键获取any
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getAnyInner(key: String): Any? {
            return get(key)
        }

        /**
         * 根据键获取list
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getListInner(key: String): List? {
            return (get(key) as? List?)?.map { it.analysisCast() }
        }

        /**
         * 根据键获取short
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getShortInner(key: String): Short? {
            return Coerce.toShort((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取byte
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getByteInner(key: String): Byte? {
            return Coerce.toByte((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取boolean
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getBooleanInner(key: String): Boolean? {
            return Coerce.toBoolean((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取double
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getDoubleInner(key: String): Double? {
            return Coerce.toDouble((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取float
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getFloatInner(key: String): Float? {
            return Coerce.toFloat((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取long
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getLongInner(key: String): Long? {
            return Coerce.toLong((get(key) ?: return null).analysisCast())
        }

        /**
         * 根据键获取int
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getIntInner(key: String): Int? {
            return get(key)?.let { Coerce.toInteger(it.analysisCast()) }
        }

        /**
         * 根据键获取string
         *
         * @param key
         * @param processData
         * @return
         */
        fun AsahiDataMap.getStringInner(key: String): String? {
            return get(key)?.run { analysisCast().toString() }
        }
    }
}