#!/usr/bin/python3

# checkpoint:
# 1. export blueprint package list (copy)
# 2. blueprint description edit/update
# 3. hostname setting
# 4. user setting

import crypt
import re

import composerlib
import testlib


@testlib.nondestructive
class TestCustom(composerlib.ComposerCase):

    def testExport(self):
        b = self.browser
        m = self.machine

        self.login_and_go("/composer", superuser=True)
        b.wait_present("#main")

        # go to blueprint openssh-server
        b.click("#openssh-server-name")

        # export package list
        actions_drop_down_sel = ".cmpsr-header__actions #dropdownKebab"
        with b.wait_timeout(300):
            b.click(actions_drop_down_sel)
        b.wait_attr(actions_drop_down_sel, "aria-expanded", "true")
        b.click("a:contains('Export')")
        b.wait_present("#cmpsr-modal-export")
        with b.wait_timeout(300):
            b.wait_in_text("#textInput2-modal-markup", "openssh-server")
        packages_list = b.text("#textInput2-modal-markup")
        b.click("#cmpsr-modal-export button[data-btn=copy-export]")
        b.click("#cmpsr-modal-export button[data-btn=close-export]")
        b.wait_not_present("#cmpsr-modal-export")

        list_from_backend = m.execute("""
            composer-cli blueprints depsolve openssh-server | tail -n +2
            """).split()
        # remove tailing ".x86_64" or ".noarch"(6 chars) or ".aarch64"(7 chars)
        # and "*:" in libpcap-14:1.9.1-2.fc31.x86_64
        regex = re.compile("-[0-9]*:")

        def remove(x):
            if x.endswith("aarch64"):
                return regex.sub("-", x[:-8])
            else:
                return regex.sub("-", x[:-7])
        format_list = map(remove, list_from_backend)

        self.assertEqual(packages_list, "\n".join(format_list))

        # collect code coverage result
        self.check_coverage()

    def testDescription(self):
        b = self.browser
        m = self.machine

        self.login_and_go("/composer", superuser=True)
        b.wait_present("#main")

        # go to blueprint openssh-server
        b.click("#openssh-server-name")

        # update description (cancel first)
        updated_description = "edit ssh server image description"
        with b.wait_timeout(300):
            b.click("a:contains('ssh server image')")
        b.wait_present("#cmpsr-modal-edit-description")
        b.set_input_text("#textInput-modal-markup", updated_description)
        b.click("#cmpsr-modal-edit-description button:contains('Cancel')")
        b.wait_not_present("#cmpsr-modal-edit-description")
        b.wait_present("a:contains('ssh server image')")

        # update description
        actions_drop_down_sel = ".cmpsr-header__actions #dropdownKebab"
        b.click(actions_drop_down_sel)
        b.wait_attr(actions_drop_down_sel, "aria-expanded", "true")
        b.click("a:contains('Edit description')")
        b.wait_present("#cmpsr-modal-edit-description")
        b.set_input_text("#textInput-modal-markup", updated_description)
        b.click("#cmpsr-modal-edit-description button:contains('Save')")
        b.wait_not_present("#cmpsr-modal-edit-description")
        b.wait_present("a:contains('{}')".format(updated_description))
        # backend got updated as well
        desc = m.execute("""
            composer-cli blueprints show openssh-server | grep description | \
            awk -F '"' '{print $2}'
            """).rstrip()
        self.assertEqual(desc, updated_description)

        # collect code coverage result
        self.check_coverage()

    def testHostname(self):
        b = self.browser
        m = self.machine

        self.login_and_go("/composer", superuser=True)
        b.wait_present("#main")

        b.click("#openssh-server-name")
        with b.wait_timeout(300):
            b.click("#blueprint-tabs-tab-customizations")
        # set hostname to "openssh-server"
        b.click("button[aria-describedby=Hostname-help2]")
        b.set_input_text("input[aria-label=Hostname]", "hostname-openssh-server")
        b.click(".form-control-pf-save")
        b.wait_present("span:contains(hostname-openssh-server)")
        hostname = m.execute("""
            composer-cli blueprints show openssh-server | grep hostname | awk -F '"' '{print $2}'
            """).rstrip()
        b.wait_present("span:contains({})".format(hostname))

        # update hostname
        b.click(".form-control-pf-value")
        b.set_input_text("input[aria-label=Hostname]", "update-openssh-server")
        b.key_press("\r")
        b.wait_present("span:contains(update-openssh-server)")
        hostname = m.execute("""
            composer-cli blueprints show openssh-server | grep hostname | awk -F '"' '{print $2}'
            """).rstrip()
        b.wait_present("span:contains({})".format(hostname))

        # clicking x button will not update
        b.click(".form-control-pf-value")
        b.set_input_text("input[aria-label=Hostname]", "no-update-openssh-server")
        b.click(".form-control-pf-cancel")
        b.wait_present("span:contains(update-openssh-server)")
        hostname = m.execute("""
            composer-cli blueprints show openssh-server | grep hostname | awk -F '"' '{print $2}'
            """).rstrip()
        b.wait_present("span:contains({})".format(hostname))

        # invalid character, save button is disabled
        b.click(".form-control-pf-value")
        b.set_input_text("input[aria-label=Hostname]", "?")
        b.wait_attr(".form-control-pf-save", "disabled", "")

        # collect code coverage result
        self.check_coverage()

    def testUserAccount(self):
        b = self.browser
        m = self.machine

        # HACK: cockpit-composer may loose keystrokes, see issue #944
        def set_input_value(sel, val):
            w = ""
            for k in val:
                w += k
                while b.val(sel) != w:
                    b.focus(sel)
                    b.key_press(k)

        self.login_and_go("/composer", superuser=True)
        b.wait_present("#main")

        b.click("#openssh-server-name")
        with b.wait_timeout(300):
            b.click("#blueprint-tabs-tab-customizations")

        # add a new user
        b.click("button:contains('Create User Account')")
        b.wait_present("#cmpsr-modal-user-account")
        set_input_value("#textInput2-modal-user", "x y")
        b.wait_val("#textInput1-modal-user", "xy")
        b.click("input[type=checkbox]")
        set_input_value("#textInput1-modal-password", "fooBAR!@#123")
        set_input_value("#textInput2-modal-password", "fooBAR!@#123")
        ssh_key = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDUOtNJdBEXyKxBB898rdT54ULjMGuO6v4" \
                  "jLXmRsdRhR5Id/lKNc9hsdioPWUePgYlqML2iSV72vKQoVhkyYkpcsjr3zvBny9+5xej3+T" \
                  "BLoEMAm2hmllKPmxYJDU8jQJ7wJuRrOVOnk0iSNF+FcY/yaQ0owSF02Nphx47j2KWc0IjGG" \
                  "lt4fl0fmHJuZBA2afN/4IYIIsEWZziDewVtaEjWV3InMRLllfdqGMllhFR+ed2hQz9PN2Qc" \
                  "apmEvUR4UCy/mJXrke5htyFyHi8ECfyMMyYeHwbWLFQIve4CWix9qtksvKjcetnxT+WWrut" \
                  "dr3c9cfIj/c0v/Zg/c4zETxtp cockpit-test"
        b.set_input_text("#textInput5-modal-user", ssh_key)
        b.click("#cmpsr-modal-user-account button:contains('Create')")
        b.wait_not_present("#cmpsr-modal-user-account")
        b.wait_text("tr td[data-label='Full name']", "x y")
        b.wait_text("tr td[data-label='User name']", "xy")
        b.wait_present("tr td[data-label='Server administrator'] .fa-check")
        b.wait_present("tr td[data-label=Password] .fa-check")
        b.wait_in_text("tr td[data-label='SSH key']", "cockpit-test")

        # check backend to have correct password configured
        passwd_str = m.execute("""
            composer-cli blueprints show openssh-server | grep password | awk -F '"' '{print $2}'
            """).rstrip()
        salt = passwd_str[:passwd_str.rfind("$")]
        passwd_crypt = crypt.crypt("fooBAR!@#123", salt)
        self.assertEqual(passwd_str, passwd_crypt)
        # check backend to have correct ssh key configured
        ssh_key_backend = m.execute("""
            composer-cli blueprints show openssh-server | grep key | awk -F '"' '{print $2}'
            """).rstrip()
        self.assertEqual(ssh_key_backend, ssh_key)

        # update password
        new_password = "c456rty$%^RTY"
        b.click("button[aria-label='Edit User Account xy']")
        b.wait_present("#cmpsr-modal-user-account")
        b.click("button:contains('Set New Password')")
        set_input_value("#textInput1-modal-password", new_password)
        set_input_value("#textInput2-modal-password", new_password)
        b.click("#cmpsr-modal-user-account button:contains('Update')")
        b.wait_not_present("#cmpsr-modal-user-account")
        b.wait_present("tr td[data-label=Password] .fa-check")
        passwd_str = m.execute("""
            composer-cli blueprints show openssh-server | grep password | awk '{print $3}'
            """).strip('"\n')
        salt = passwd_str[:passwd_str.rfind("$")]
        passwd_crypt = crypt.crypt(new_password, salt)
        self.assertEqual(passwd_str, passwd_crypt)

        # remove password
        b.click("button[aria-label='Edit User Account xy']")
        b.wait_present("#cmpsr-modal-user-account")
        b.click("button:contains('Remove Password')")
        b.click("#cmpsr-modal-user-account button:contains('Update')")
        b.wait_not_present("#cmpsr-modal-user-account")
        b.wait_present("tr td[data-label=Password]")
        b.wait_not_present("tr td[data-label=Password] .fa-check")

        # duplicated user
        b.click("button:contains('Create User Account')")
        b.wait_present("#cmpsr-modal-user-account")
        set_input_value("#textInput2-modal-user", "x y")
        b.wait_val("#textInput1-modal-user", "xy")
        b.wait_in_text("#textInput1-modal-user-help1", "This user name already exists.")
        b.click("#cmpsr-modal-user-account button:contains('Cancel')")
        b.wait_not_present("#cmpsr-modal-user-account")

        # password checking error
        b.click("button:contains('Create User Account')")
        b.wait_present("#cmpsr-modal-user-account")
        set_input_value("#textInput2-modal-user", "admin user")
        b.wait_val("#textInput1-modal-user", "auser")
        b.click("input[type=checkbox]")
        set_input_value("#textInput1-modal-password", "aaa")
        set_input_value("#textInput2-modal-password", "bbb")
        b.wait_in_text("#textInput2-modal-password-help",
                       "The values entered for password do not match.")
        b.click("#cmpsr-modal-user-account .close")
        b.wait_not_present("#cmpsr-modal-user-account")

        # delete user
        delete_drop_down_sel = "button[aria-label='User Account Actions xy']"
        b.click(delete_drop_down_sel)
        b.wait_attr(delete_drop_down_sel, "aria-expanded", "true")
        b.click("a:contains('Delete User Account')")
        b.wait_not_present("table")

        # collect code coverage result
        self.check_coverage()


if __name__ == '__main__':
    testlib.test_main()