/* Pickerのタイプ */
internal enum class PickerType(val res: String) {
    DIAL("android:id/radial_picker"),
    CALENDAR("android:id/date_picker_day_picker"),
    DRUM("android:id/numberpicker_input")
}

internal fun getDatePickerType(
    uiDev: UiDevice = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation())
): PickerType {
    return if(hasObjByRes("android:id/datePicker")) {
        val _isDrum = uiDev.hasObject(By.res(PickerType.DRUM.res))
        val _isCalendar = uiDev.hasObject(By.res(PickerType.CALENDAR.res))
        when {
            _isDrum -> PickerType.DRUM
            _isCalendar -> PickerType.CALENDAR
            else -> throw UiObjectNotFoundException("Unknown picker !")
        }
    }
    else
        throw UiObjectNotFoundException("Picker not found !")
}

    private fun _SetSystemDate(year: Int, month: Int, day: Int) {
        require(MIN_YEAR <= year && year <= MAX_YEAR) {
            "${MIN_YEAR} <= year(${year}) <= ${MAX_YEAR}"
        }

        val _api = Build.VERSION.SDK_INT
        val _setDateMesg = PageMesgs.getValue(_api).setdate
        val _automaticMesg = PageMesgs.getValue(_api).auto
        val _okMesg = PageMesgs.getValue(_api).ok
        val _cancelMesg = PageMesgs.getValue(_api).cancel

        // 「Date & time」設定ページを開く
        showDateTimePage(context)

        // 時刻の自動(Network)設定を無効化
        findObjByText(_setDateMesg, uiDev).apply {
            if (! isEnabled)
                findObjByTextAndClick(_automaticMesg, uiDev)
            recycle()
        }

        // 年月日の変更
        findObjByTextAndClick(_setDateMesg, uiDev)      // Pickerを開く
        val _picker = getDatePickerType(uiDev)
        Log.i(TAG, "PickerType(API${Build.VERSION.SDK_INT}) = ${_picker}")
        try {
            when(_picker) {
                PickerType.DRUM ->
                    setDateWithDrum(year, month, day, uiDev)
                PickerType.CALENDAR ->
                    setDateWithCalendar(year, month, day, uiDev)
                else -> {}
            }
            findObjByTextAndClick(_okMesg, uiDev)       // OKボタン
        } catch (e: UiObjectNotFoundException) {
            findObjByTextAndClick(_cancelMesg, uiDev)   // CANCELボタン
            throw UiObjectNotFoundException("Date:${year}/${month + 1}/${day}")
        }

        // 「Date & time」設定ページを閉じる
        findObjByText(_setDateMesg, uiDev).recycle()
        uiDev.pressBack()       // バックボタン押下
    }

    private fun _SetAutomatic() {
        val _api = Build.VERSION.SDK_INT
        val _setDateMesg = PageMesgs.getValue(_api).setdate
        val _automaticMesg = PageMesgs.getValue(_api).auto

        // 「Date & time」設定ページを開く
        showDateTimePage(context)

        // 時刻の自動(Network)設定を有効化
        findObjByText(_setDateMesg, uiDev).apply {
            if (isEnabled)
                findObjByTextAndClick(_automaticMesg, uiDev)
            recycle()
        }

        // 「Date & time」設定ページを閉じる
        findObjByText(_setDateMesg, uiDev).recycle()
        uiDev.pressBack()       // バックボタン押下
    }

/* Settingのページで用いられるメッセージ */
internal data class MesgData(
    val auto: String,
    val setdate: String,
    val settime: String,
    val ok: String,
    val cancel: String
)
internal val PageMesgs = mapOf(
    23 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    24 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    25 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    26 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    27 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    28 to MesgData("Automatic date & time", "Set date", "Set time", "OK", "CANCEL"),
    29 to MesgData("Use network-provided time", "Date", "Time", "OK", "Cancel"),
    30 to MesgData("Use network-provided time", "Date", "Time", "OK", "Cancel"),
    31 to MesgData("Set time automatically", "Date", "Time", "OK", "Cancel"),
    32 to MesgData("Set time automatically", "Date", "Time", "OK", "Cancel"),
    33 to MesgData("Set time automatically", "Date", "Time", "OK", "Cancel"),
    34 to MesgData("Set time automatically", "Date", "Time", "OK", "Cancel")
)

/**
 * Settingsアプリの「Time & Date」ページを開きます。
 *
 * @param context コンテキスト
 */
internal fun showDateTimePage(
    context: Context = InstrumentationRegistry.getInstrumentation().context
) {
    val intent = Intent(Settings.ACTION_DATE_SETTINGS)
    intent.flags = Intent.FLAG_ACTIVITY_NEW_TASK
    context.startActivity(intent)
}

/**
 * システム年月日を設定するルールを提供します.
 * テストが終わると自動的にAutomaticモード（Networkに経由による日付合わせ）に変更されます.
 *
 * @param year 年
 * @param month 月（0,1,2,...,11）
 * @param day 日
 */
@TargetApi(23)
class SysDateTestRule(
    var year: Int = -1, var month: Int = -1, var day: Int = -1
) : TestRule {

    val context = InstrumentationRegistry.getInstrumentation().context
    val uiDev = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation())

    companion object {
        private const val MIN_YEAR = 2015
        private const val MAX_YEAR = 2030
    }

    // ----- ルール本体 -----------------------------------------------
    override fun apply(base: Statement, description: Description): Statement {
        return object : Statement() {
            override fun evaluate() {
                try {
                    setSysDate()        // 前処理
                    base.evaluate()     // テストの実施
                } finally {
                    setAutomatic()      // 後処理
                }
            }
        }
    }

    /**
     * システム年月日をネット経由で自動的に設定します.
     */
    fun setAutomatic() {
        Log.i(TAG, "setAutomatic")
        _SetAutomatic()
        year = -1
        month = -1
        day = -1
    }

    /**
     * システム年月日を指定値に設定します
     * @param year 年
     * @param month 月（0,1,2,...,11）
     * @param day 日
     */
    fun setSysDate(
        year: Int = this.year, month: Int = this.month, day: Int = this.day
    ) {
        Log.i(TAG, "setSysDate ${year}/${month}/${day}")
        if (year >= 0 && month >= 0 && day >= 0) {
            _SetSystemDate(year, month, day)
            this.year = year
            this.month = month
            this.day = day
        }
    }
	
    // ----------------------------------------------------------------
    private fun _SetAutomatic() { ... }
    private fun _SetSystemDate(year: Int, month: Int, day: Int) { ... }
}