from views.base_element import BaseEditBox, BaseButton, BaseElement
from views.sign_in_view import SignInView


class PassphraseInput(BaseEditBox):

    def __init__(self, driver):
        super(PassphraseInput, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.EditText")


class EnterSeedPhraseButton(BaseButton):

    def __init__(self, driver):
        super(EnterSeedPhraseButton, self).__init__(driver)
        self.locator = self.Locator.accessibility_id("enter-seed-phrase-button")


class ReencryptYourKeyButton(BaseButton):

    def __init__(self, driver):
        super(ReencryptYourKeyButton, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='Re-encrypt your keys']")


class ConfirmRecoverAccess(BaseButton):

    def __init__(self, driver):
        super(ConfirmRecoverAccess, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//android.widget.TextView[@text='RECOVER ACCESS']")

class ContinueCustomSeedPhraseButton(BaseButton):

    def __init__(self, driver):
        super(ContinueCustomSeedPhraseButton, self).__init__(driver)
        self.locator = self.Locator.accessibility_id("continue-custom-seed-phrase")

class CancelCustomSeedPhraseButton(BaseButton):

    def __init__(self, driver):
        super(CancelCustomSeedPhraseButton, self).__init__(driver)
        self.locator = self.Locator.accessibility_id("cancel-custom-seed-phrase")


class RequiredField(BaseElement):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.text_selector("Required field")


class InvalidRecoveryPhrase(BaseElement):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.text_selector("Recovery phrase is invalid")


class TooShortPassword(BaseElement):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.text_selector("Password is too short")


class MisspelledWords(BaseElement):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.text_selector("Some words might be misspelled")


class Warnings(BaseElement):
    def __init__(self, driver):
        super().__init__(driver)
        self.required_field = RequiredField(driver)
        self.invalid_recovery_phrase = InvalidRecoveryPhrase(driver)
        self.too_short_password = TooShortPassword(driver)
        self.misspelled_words = MisspelledWords(driver)


class ConfirmPhraseButton(BaseButton):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.id("android:id/button1")

    def navigate(self):
        from views.home_view import HomeView
        return HomeView(self.driver)


class CancelPhraseButton(BaseButton):
    def __init__(self, driver):
        super().__init__(driver)
        self.locator = self.Locator.id("android:id/button2")


class RecoverAccessView(SignInView):

    def __init__(self, driver):
        super(RecoverAccessView, self).__init__(driver)
        self.driver = driver

        self.passphrase_input = PassphraseInput(self.driver)
        self.enter_seed_phrase_button = EnterSeedPhraseButton(self.driver)
        self.confirm_recover_access = ConfirmRecoverAccess(self.driver)
        self.reencrypt_your_key_button = ReencryptYourKeyButton(self.driver)
        self.warnings = Warnings(self.driver)
        self.confirm_phrase_button = ConfirmPhraseButton(self.driver)
        self.cancel_button = CancelPhraseButton(self.driver)
        self.continue_custom_seed_phrase_button = ContinueCustomSeedPhraseButton(self.driver)
        self.cancel_custom_seed_phrase_button = CancelCustomSeedPhraseButton(self.driver)