eggplants/extension-manifest-converter

# Copyright 2021 Google Inc. All Rights Reserved.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#     http://www.apache.org/licenses/LICENSE-2.0
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from __future__ import annotations

import os
import shutil
from pathlib import Path

from ..utils.unzip import Unzip
from ..utils.zip import Zip
from .logger import Logger
from .type import Type, TypeEnum
from .worker import Worker


class Frontend:
    def __init__(self, args: dict[str, str]) -> None:
        source = args["source"]
        original_source = source
        dirname = os.path.dirname(source)

        # TODO: If not a directory, create one and move this item into it.
        type = Type().getFileType(source)

        if type is TypeEnum.UNKNOWN:
            # TODO: Error
            return
        elif type is TypeEnum.ZIP:
            foldername = str(Path(os.path.basename(source)).with_suffix(""))
            destination = dirname + os.sep + foldername
            if os.path.exists(destination):
                os.remove(destination)
            Unzip().extract(source, dirname, foldername)
            source = str(Path(source).with_suffix(""))
        elif type != TypeEnum.DIR:
            source_dir = source + "_dir"
            os.mkdir(source_dir)
            os.rename(source, source_dir + os.sep + os.path.basename(source))
            source = source_dir

        worker = Worker()
        worker.work(source)

        # Maybe remove working or source directory.
        # Destination directories are created alongside source with _delete
        # appended.
        if type is TypeEnum.ZIP:
            Logger().log("Replacing .zip file")

            # Create zip for working directory.
            prev_source = source
            source = source + "_delete"
            os.mkdir(dirname + "/tmp_delete")
            shutil.move(source, dirname + "/tmp_delete/extension/")
            Zip.zip(dirname + "/tmp_delete", source)

            # Delete working directory.
            shutil.rmtree(dirname + "/tmp_delete")
            shutil.rmtree(prev_source)
            os.remove(original_source)
            os.rename(source + ".zip", original_source)
        # A manifest.json file replaces the original.
        # TODO: Consistent replacement behavior with dir, zip, and manifest.json.
        elif type is TypeEnum.MANIFEST:
            Logger().log("Replacing manifest.json")
            os.remove(source + os.sep + os.path.basename(original_source))
            os.rmdir(source)
            source = source + "_delete/"
            os.rename(source + os.path.basename(original_source), original_source)
            os.rmdir(source)
        elif type is TypeEnum.DIR:
            # TODO: A better destination could be something other than _delete.
            Logger().log("Converted extension into: {}_delete".format(source))