prehook_lib.py

import os
import re


class ImportFixer:
    def __init__(self, target_path, stencil_dir):
        self.target_dir = os.path.dirname(target_path)
        self.target_path = target_path
        self.stencil_dir = stencil_dir
        self.rel_stencil_dir = os.path.relpath(self.stencil_dir,
                                               start=self.target_dir)
        with open(target_path, 'r', encoding="utf-8") as f:
            self.content = f.read()

    def fix_import(self, name, location, filename=None):
        rel_loc = os.path.relpath(location, start=self.target_dir)
        self.content = re.sub(
            rf'my-gdrive\(["\'](.*-{name}\.arr)["\']\)',
            rf'file("{rel_loc}/{filename}")'
            if filename else rf'file("{rel_loc}/\1")', self.content)
        OLD_DOCDIFF_IMPORT="gdrive-js(\"docdiff_qtm-validation.js\", \"11H5gJQtW9TJaiFkWw51fR4_oIibmLr7X\")"
        NEW_DOCDIFF_IMPORT="shared-gdrive(\"docdiff_qtm-validation.arr\", \"12O---8ZF_VkukhEDGK63Z7hSvDf_PhEW\")"
        self.content = self.content.replace(OLD_DOCDIFF_IMPORT, NEW_DOCDIFF_IMPORT)

    def finalize(self):
        self.content = re.sub(r'shared-gdrive\(["\'](.*?)["\'].*?\n?.*?\)',
                              rf'file("{self.rel_stencil_dir}/\1")',
                              self.content, re.M)
        with open(self.target_path, 'w', encoding="utf-8") as f:
            f.write(self.content)

class CPOProvideFixer:
    def __init__(self, target_path, stencil_dir):
        self.target_dir = os.path.dirname(target_path)
        self.target_path = target_path
        self.stencil_dir = stencil_dir
        self.rel_stencil_dir = os.path.relpath(self.stencil_dir,
                                               start=self.target_dir)
        with open(target_path, 'r', encoding="utf-8") as f:
            self.content = f.read()
        
    def fix_provide(self, names, location):
        rel_loc = os.path.relpath(location, start=self.target_dir)
        self.content = re.sub(
            rf'use context essentials2021',
            rf'use context essentials2021\nprovide {names} end',
            self.content)
        
    def finalize(self):
        self.content = re.sub(r'shared-gdrive\(["\'](.*?)["\'].*?\n?.*?\)',
                              rf'file("{self.rel_stencil_dir}/\1")',
                              self.content, re.M)
        with open(self.target_path, 'w', encoding="utf-8") as f:
            f.write(self.content)