Add full chat extraction
This commit is contained in:
parent
2734c18b11
commit
3eb43d4d19
@ -47,7 +47,7 @@ private suspend fun SpaceClient.writeMessages(
|
||||
writer.appendLine(
|
||||
"""
|
||||
|* **(${message.created}) ${message.author.name}:**
|
||||
|${message.text.trimIndent()}
|
||||
|${message.text}
|
||||
""".replaceIndentByMargin(prefix)
|
||||
)
|
||||
|
||||
@ -106,6 +106,8 @@ suspend fun SpaceClient.extractMessages(
|
||||
|
||||
val file = parentDirectory.resolve("$name.md")
|
||||
|
||||
logger.info("Extracting messages from channel $chatId to $file")
|
||||
|
||||
file.parent.createDirectories()
|
||||
|
||||
file.outputStream(
|
||||
|
@ -10,6 +10,7 @@ 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.chats
|
||||
import space.jetbrains.api.runtime.resources.projects
|
||||
import space.jetbrains.api.runtime.types.FolderIdentifier
|
||||
import space.jetbrains.api.runtime.types.ProjectIdentifier
|
||||
@ -170,7 +171,7 @@ private class ExtractRepositoriesCommand : ExtractCommand("repos", "Extract repo
|
||||
|
||||
}
|
||||
|
||||
private class ExtractChatsCommand: ExtractCommand("chat","Extract all messages from a chat"){
|
||||
private class ExtractMessagesCommand : ExtractCommand("messages", "Extract all messages from a chat") {
|
||||
|
||||
val path: String by option(
|
||||
ArgType.String,
|
||||
@ -182,7 +183,7 @@ private class ExtractChatsCommand: ExtractCommand("chat","Extract all messages f
|
||||
|
||||
val spaceUrl = urlMatch.groups["spaceUrl"]?.value ?: error("Space Url token not recognized")
|
||||
|
||||
val chatId = urlMatch.groups["chatId"]?.value ?: error("Chat id token not recognized")
|
||||
val chatId = urlMatch.groups["chatId"]?.value
|
||||
|
||||
val appInstance = SpaceAppInstance(
|
||||
clientId ?: System.getProperty("space.clientId"),
|
||||
@ -197,13 +198,19 @@ private class ExtractChatsCommand: ExtractCommand("chat","Extract all messages f
|
||||
)
|
||||
|
||||
runBlocking {
|
||||
if (chatId == null) {
|
||||
spaceClient.chats.channels.listAllChannels(query = "").data.forEach {
|
||||
spaceClient.extractMessages(it.channelId, Path(path))
|
||||
}
|
||||
} else {
|
||||
spaceClient.extractMessages(chatId, Path(path))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
companion object {
|
||||
private val urlRegex =
|
||||
"""(?<spaceUrl>https?:\/\/[^\/]*)\/im\/group\/(?<chatId>.*)""".toRegex()
|
||||
"""(?<spaceUrl>https?:\/\/[^\/]*)(\/im\/group\/(?<chatId>.*))?""".toRegex()
|
||||
}
|
||||
}
|
||||
|
||||
@ -272,7 +279,12 @@ private class ExtractProjectCommand : ExtractCommand("project", "Extract all dat
|
||||
fun main(args: Array<String>) {
|
||||
val parser = ArgParser("space-export")
|
||||
|
||||
parser.subcommands(ExtractDocumentsCommand(), ExtractRepositoriesCommand(), ExtractProjectCommand(), ExtractChatsCommand())
|
||||
parser.subcommands(
|
||||
ExtractDocumentsCommand(),
|
||||
ExtractRepositoriesCommand(),
|
||||
ExtractProjectCommand(),
|
||||
ExtractMessagesCommand()
|
||||
)
|
||||
|
||||
parser.parse(args)
|
||||
}
|
Loading…
Reference in New Issue
Block a user