com.waicool20.skrypton.setup.Installer.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of skrypton-native-windows64 Show documentation
A browser automation library that uses SikuliX and doesn't grab grab your mouse
The newest version!
/*
 * The MIT License (MIT)
 *
 * Copyright (c) skrypton-api by waicool20
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package com.waicool20.skrypton.setup

import org.sikuli.script.ImagePath
import org.slf4j.LoggerFactory
import java.io.InputStream
import java.net.URI
import java.nio.file.*
import java.nio.file.attribute.PosixFilePermission
import kotlin.streams.toList

object Installer {
    private val logger = LoggerFactory.getLogger(javaClass)
    private val skryptonAppDir = SKryptonSetup.skryptonAppDir
    private val codeSource = javaClass.protectionDomain.codeSource.location.toURI().path

    fun silentInstall() {
        createDirectories()
        copySKryptonLibs()
        installSikuliXLibs()
        cleanup()
    }

    fun createDirectories(callback: (dir: String) -> Unit = {}) {
        if (Files.notExists(skryptonAppDir)) {
            logger.debug("[CREATE] $skryptonAppDir")
            Files.createDirectories(skryptonAppDir)
            callback(skryptonAppDir.toString())
        }
    }

    fun copySKryptonLibs(callback: (file: String, index: Int, total: Int) -> Unit = { _, _, _ -> }) {
        if (codeSource.endsWith(".jar")) {
            val jarURI = URI.create("jar:file:$codeSource")
            val env = mapOf(
                    "create" to "false",
                    "encoding" to "UTF-8"
            )
            (FileSystems.newFileSystem(jarURI, env)).use { fs ->
                copyResources(fs.getPath("/skrypton"), 1, callback)
            }
        } else {
            // For use in IDE
            val releaseResources = Paths.get("native/build/resources/release/skrypton")
            val debugResources = Paths.get("native/build/resources/debug/skrypton")
            when {
                Files.exists(releaseResources) -> copyResources(releaseResources, 5, callback)
                Files.exists(debugResources) -> copyResources(debugResources, 5, callback)
                else -> tornadofx.error("No resource files were found")
            }
        }
    }

    private fun copyResources(rootPath: Path,
                      dropElements: Int,
                      callback: (file: String, index: Int, total: Int) -> Unit) {
        val jobs = Files.walk(rootPath).filter { it.nameCount > dropElements && !Files.isDirectory(it) }
                .map { it to skryptonAppDir.resolve("${it.subpath(dropElements, it.nameCount)}") }.toList()

        jobs.forEachIndexed { index, (source, dest) ->
            copy(source, dest)
            if (OS.isUnix()) {
                val perms = Files.getPosixFilePermissions(dest).toMutableSet()
                perms.addAll(listOf(PosixFilePermission.OWNER_EXECUTE))
                Files.setPosixFilePermissions(dest, perms)
            }
            callback(source.fileName.toString(), index, jobs.size)
        }
    }

    fun  copy(source: T, target: Path) {
        Files.createDirectories(target.parent)
        when (source) {
            is Path -> {
                logger.debug("[COPY] $source to $target")
                Files.copy(source, target, StandardCopyOption.REPLACE_EXISTING)
            }
            is InputStream -> {
                logger.debug("[COPY] $target")
                Files.copy(source, target, StandardCopyOption.REPLACE_EXISTING)
            }
            else -> error("Can only copy from Path or InputStream object")
        }
    }

    fun installSikuliXLibs() {
        logger.debug("Installing SikuliX libraries")
        ImagePath.getBundleFolder()
    }

    fun cleanup() {
        logger.debug("All done!")
    }

}