###########################################################################################
# Helper classes for the different CI commands
###########################################################################################
import sys, os, re
import shutil
from System.Net import WebClient

class SearchBuild:
    def __init__(self):
        self.artifacts = list()
        self.error = False

    # search for files with a given file-ending, and call the method
    # doit() for every occurance
    def search(self, ending, folder):
        for root, dirs, files in os.walk(folder):
            for file in files:
                if file.endswith(ending):
                    self.artifacts += self.doit(os.path.join(root, file))
        print("%u artifact(s) created" % len(self.artifacts))

    def save(self, basepath, destination):
        for filename in self.artifacts:
            dirname = os.path.dirname(os.path.relpath(filename, basepath))
            if dirname.startswith(".."):
                dirname = os.path.dirname(filename).lstrip("/")
            
            destdir = os.path.join(destination, dirname)
            if not os.path.exists(destdir):
                os.mkdir(destdir)
            shutil.copy2(filename, destdir)
            print("%s => %s" % (filename, destdir))


    # dummy rule
    def doit(self, filename):
        artifacts = list()
        return artifacts

#
# Install a library from a URL (starting with http(s))
#
def install_library(url, librarymanager):
    if not url.startswith("http"):
        return
    
    downloaddir = os.path.join(os.environ["USERPROFILE"], "downloads")
    if not os.path.exists(downloaddir):
        os.makedirs(downloaddir)
    basename, extension = os.path.splitext(url)
    localname = os.path.join(downloaddir, "dl" + extension)
    print("*** download %s to %s\n" % (url, localname))

    web_client = WebClient()
    web_client.DownloadFile(url, localname)
    repo = librarymanager.repositories[0]
    print("*** installing %s\n" % url)
    librarymanager.install_library (localname, repo, True)


#
# Check if there is a *.requirements file, and install
# all requirements from there. The parameter is a library
# URL, while we are checking if a *.requirements file exists
# beside this library.
#
def install_requirements(library_filename, librarymanager):
    reqfilename = os.path.splitext(library_filename)[0] + ".requirements"
    
    if os.path.isfile(reqfilename):
        reqfile = open(reqfilename, "r")
        
        for req in reqfile:
            req = req.strip()
            print("installing requirement '%s'." % req.strip())
            install_library(req, librarymanager)

#
# search for missing libraries in current primary project
#
def install_missing_libraries(proj, librarymanager):
    print("*** install missing libraries")
    
    # compile a regular expression to match lib placeholders
    p = re.compile('[^,]+, [^(]+\([^)]+\)')
    
    # search for libman
    objects = proj.get_children(recursive=True)
    for object in objects:
        if object.is_libman:
            print("*** found library manager")
            for libref in iter(object):
                libInfo = str(libref)
                if libref.is_placeholder:
                    if libref.effective_resolution is None and libref.default_resolution is not None:
                        libInfo = str(libref.default_resolution)
                        m = re.findall('([^,]+), ([^(]+) \(([^)]+)\)', libInfo)
                        if len(m) > 0:
                            libname, libversion, libvendor = m[0]
                            indexurl = "https://store.codesys.com/CODESYSLibs/%s/%s/%s/index" % (libvendor, libname, libversion)
                            web_client = WebClient()
                            try:
                                filename = web_client.DownloadString(indexurl).rstrip()
                            except:
                                filename = None

                            if filename != None:
                                liburl = "https://store.codesys.com/CODESYSLibs/%s/%s/%s/%s" % (libvendor, libname, libversion, filename)
                                install_library(liburl, librarymanager)


#
# update all devices in project
#
def update_device(proj, device_repository):
    print("*** update device")
    
    # search for devices to update
    objects = proj.get_children(recursive=True)
    for object in objects:
        if object.is_device:
            print("*** found device")
            DeviceId = object.get_device_identification()
            devices = device_repository.get_all_devices()
            for device in devices:
                if device.device_id.type == DeviceId.type and device.device_id.id == DeviceId.id:
                    deviceToUpdate = device
            if deviceToUpdate != None:
                print ("*** found device to update %s" % deviceToUpdate.device_id)
                object.update(device=deviceToUpdate.device_id)