uninstall.py

#
# Copyright (C) 2020 IBM. All Rights Reserved.
#
# See LICENSE.txt file in the root directory
# of this source tree for licensing information.
#

# pylint: disable=no-name-in-module,import-error
import getpass
import io
import json
import os
import sys
from distutils.dir_util import remove_tree

from clai.datasource.config_storage import ConfigStorage
from clai.datasource.stats_tracker import StatsTracker
from clai.server.agent_datasource import AgentDatasource
from clai.tools.anonymizer import Anonymizer
from clai.tools.console_helper import print_complete, print_error
from clai.tools.file_util import get_rc_file, get_setup_file, get_rc_files, is_rw_with_EBCDIC


def remove(path):
    print("cleaning %s" % path)
    try:
        remove_tree(path)
    except OSError:
        print("folder not found")


def remove_system_folder(path: str = None):
    if path is not None:
        # path to the clai dir in a local user installation
        default_system_destdir = "/".join(path.split("/")[0:-1])
    else:
        default_system_destdir = os.path.join(
            os.path.expanduser("/opt/local/share"),
            "clai",
        )

    remove(default_system_destdir)


def clai_installed(rc_file_path):
    path = os.path.expanduser(rc_file_path)
    if os.path.isfile(path):
        line_to_search = "export CLAI_PATH="
        print("searching %s" % line_to_search)
        lines = io.open(path, "r",
                        encoding="utf-8",
                        errors="ignore").readlines()
        lines_found = list(filter(lambda line: line_to_search in line, lines))

        if lines_found:
            my_path = lines_found[0].replace(line_to_search, "").replace("\n", "").strip()
            return my_path

    return None


def is_root_user():
    return os.geteuid() == 0


# pylint: disable=bare-except
def read_users(bin_path):
    try:
        with open(bin_path + "/usersInstalled.json") as file:
            users = json.load(file)
            return users
    except:
        return []


def unregister_the_user(bin_path):
    users = read_users(bin_path)
    user_path = os.path.expanduser(get_rc_file())
    if user_path in users:
        users.remove(user_path)

    with open(bin_path + "/usersInstalled.json", "w") as json_file:
        json.dump(users, json_file)

    return users


def stat_uninstall(bin_path):
    agent_datasource = AgentDatasource(config_storage=ConfigStorage(alternate_path=f"{bin_path}/configPlugins.json"))
    report_enable = agent_datasource.get_report_enable()
    stats_tracker = StatsTracker(sync=True, anonymizer=Anonymizer(alternate_path=f"{bin_path}/anonymize.json"))
    stats_tracker.report_enable = report_enable
    login = getpass.getuser()
    stats_tracker.log_uninstall(login)
    print("record uninstall")


def remove_setup_file(rc_file_path):
    path = os.path.expanduser(rc_file_path)
    os.remove(path)


def remove_between(rc_file_path, start, end):
    path = os.path.expanduser(rc_file_path)
    if os.path.isfile(path):
        codeset = "utf-8"
        lines = []
        if is_rw_with_EBCDIC(path):
            # The open() with encoding cp1047 (IBM-1047), working as cp037
            # (IBM-037), doesn't identify the newline correctly.  Use \x85 to
            # identify the newline.
            codeset = "cp1047"
            newline = '\x85'
            for err_line in open(path, "r", encoding=codeset, errors="ignore").readlines():
                right_line_list = err_line.split(newline)
                length = len(right_line_list)
                i = 0
                for right_line in right_line_list:
                    i = i + 1
                    # not empty element or not the last emtpy element
                    if right_line or length != i:
                        lines.append(right_line + newline)
        else:
            lines = open(path, "r",
                         encoding=codeset,
                         errors="ignore").readlines()

        remove_line = False
        lines_after_remove = []
        for line in lines:
            if line.strip() == start.strip():
                remove_line = True
            if not remove_line:
                lines_after_remove.append(line)

            if line.strip() == end.strip():
                remove_line = False

        io.open(path, "w", encoding=codeset).writelines(lines_after_remove)


def remove_lines_setup(rc_file_path):
    remove_between(rc_file_path, "# CLAI setup\n", "# End CLAI setup\n")


def remove_setup_register():
    rc_files = get_rc_files()

    for file in rc_files:
        remove_lines_setup(file)


def is_user_install(bin_path):
    plugins_config = ConfigStorage(
        alternate_path=f"{bin_path}/configPlugins.json"
    ).read_config(None).user_install

    return plugins_config


def execute(args):
    bin_path = os.getenv('CLAI_PATH', None)

    if "-h" in args or "--help" in args:
        print(
            "usage: uninstall.py [-h] [--help] [--user]\n \
            \nUninstall CLAI.\n \
            \noptional arguments: \
            \n-h, --help            show this help message and exit \
            \n--user                Uninstalls a local installation of clai for the current user"
        )
        sys.exit(0)

    path = clai_installed(get_setup_file())
    print(f"path= {path}")
    print(f"bin path= {bin_path}")

    if not path and bin_path is None:
        print_error("CLAI is not installed.")
        sys.exit(1)

    if not bin_path:
        bin_path = path

    stat_uninstall(path)
    users = unregister_the_user(path)

    if "--user" in args or is_user_install(bin_path):
        remove_system_folder(bin_path)
    elif not users:
        remove_system_folder()

    remove_setup_file(get_setup_file())
    remove_setup_register()

    print_complete("CLAI has been uninstalled correctly, you will need to restart your shell.")

    return 0


if __name__ == "__main__":
    sys.exit(execute(sys.argv))