space-export/src/main/kotlin/main.kt

package ru.mipt.npm.space.documentextractor

import io.ktor.client.engine.cio.CIO
import kotlinx.cli.ArgParser
import kotlinx.cli.ArgType
import kotlinx.cli.default
import kotlinx.cli.required
import kotlinx.coroutines.coroutineScope
import space.jetbrains.api.runtime.SpaceAppInstance
import space.jetbrains.api.runtime.SpaceAuth
import space.jetbrains.api.runtime.SpaceClient
import space.jetbrains.api.runtime.ktorClientForSpace
import space.jetbrains.api.runtime.resources.projects
import space.jetbrains.api.runtime.types.FolderIdentifier
import space.jetbrains.api.runtime.types.ProjectIdentifier
import java.nio.file.Files
import java.nio.file.Path
import kotlin.io.path.Path

suspend fun main(args: Array<String>) {
    val parser = ArgParser("space-document-extractor")

    val spaceUrl by parser.option(
        ArgType.String,
        description = "Url of the space instance like 'https://mipt-npm.jetbrains.space'"
    ).required()

    val project by parser.option(
        ArgType.String,
        description = "The key of the exported project"
    ).required()

    val path: String? by parser.option(
        ArgType.String,
        description = "Target directory. Default is './output/project-key'."
    )

    val folderId: String? by parser.option(
        ArgType.String,
        description = "FolderId for the folder to export. By default uses project root."
    )

    val generateHtml by parser.option(
        ArgType.Boolean,
        description = "If defined, generate HTML directory and zip"
    ).default(false)

    val htmlOutputPath by parser.option(
        ArgType.String,
        description = "Path for html output directory sibling to 'output' directory"
    ).default("html")

    val clientId by parser.option(
        ArgType.String,
        description = "Space application client ID (if not defined, use environment value 'space.clientId')"
    )

    val clientSecret by parser.option(
        ArgType.String,
        description = "Space application client secret (if not defined, use environment value 'space.clientSecret')"
    )

    parser.parse(args)

    val target: Path = path?.let { Path(it) } ?: folderId?.let { Path("output") } ?: Path("output/$project")

    Files.createDirectories(target)

    val space: SpaceClient = SpaceClient(
        ktorClientForSpace(CIO),
        SpaceAppInstance(
            clientId ?: System.getProperty("space.clientId"),
            clientSecret ?: System.getProperty("space.clientSecret"),
            spaceUrl
        ),
        SpaceAuth.ClientCredentials()
    )

    coroutineScope {
        println("Processing project \"${space.projects.getProject(ProjectIdentifier.Key(project)).name}\"")
        space.downloadAndProcessDocumentsInProject(
            target,
            ProjectIdentifier.Key(project),
            folderId?.let { FolderIdentifier.Id(it) } ?: FolderIdentifier.Root
        )
        if (generateHtml) {
            generateHtml(target, target.resolveSibling(htmlOutputPath))
        }
    }
}