4
0
Fork 0
mirror of https://github.com/Anvilcraft/modpacktools synced 2024-06-02 10:39:59 +02:00
modpacktools/src/main/kotlin/ley/anvil/modpacktools/commands/DownloadMods.kt
2020-08-14 18:46:36 +02:00

85 lines
3.1 KiB
Kotlin

package ley.anvil.modpacktools.commands
import ley.anvil.addonscript.wrapper.FileOrLink
import ley.anvil.modpacktools.MPJH
import ley.anvil.modpacktools.command.AbstractCommand
import ley.anvil.modpacktools.command.CommandReturn
import ley.anvil.modpacktools.command.CommandReturn.Companion.success
import ley.anvil.modpacktools.command.LoadCommand
import ley.anvil.modpacktools.util.DownloadFileTask
import ley.anvil.modpacktools.util.downloadFiles
import ley.anvil.modpacktools.util.fPrintln
import net.sourceforge.argparse4j.impl.Arguments.storeTrue
import net.sourceforge.argparse4j.impl.type.FileArgumentType
import net.sourceforge.argparse4j.inf.ArgumentParser
import net.sourceforge.argparse4j.inf.Namespace
import java.io.File
import java.net.URL
import java.util.stream.Collectors.toMap
@LoadCommand
object DownloadMods : AbstractCommand("DownloadMods") {
override val helpMessage: String = "Downloads all mods."
override fun ArgumentParser.addArgs() {
addArgument("dir")
.type(FileArgumentType().verifyCanCreate())
.help("the directory to download the mods to")
addArgument("-f", "--force")
.action(storeTrue())
.help("if true, mods that are already in the download folder will be downloaded again")
addArgument("-a", "--all")
.action(storeTrue())
.help("Downloads not only mods but everything with a dir installer")
}
override fun execute(args: Namespace): CommandReturn {
val json = MPJH.asWrapper
val fileList = mutableListOf<FileOrLink>()
for(
rel in json!!.defaultVersion!!.getRelations(
arrayOf("client"),
if(args.getBoolean("all")) null else arrayOf("mod")
)!!
) //TODO only client? what if someone wants a server?
if(rel.hasFile())
fileList.add(rel.file.get())
downloadFiles(
fileList.stream()
.filter {it.isURL}
.filter {
val (installer, dir) = it.installer.split(':')
installer == "internal.dir" && (args.getBoolean("all") || dir == "mods")
}
.collect(
toMap<FileOrLink, URL, File>(
{URL(it.link)},
{
val dir = it.installer.split(':').last()
if(args.getBoolean("all"))
File(args.get<File>("dir"), dir)
else
args.get<File>("dir")
},
{_: File, f: File -> f}
)
),
{r: DownloadFileTask.Return ->
println("${r.responseCode} ${r.responseMessage} ${r.url} ${r.file}")
if(r.exception != null) {
fPrintln("ERROR DOWNLOADING ${r.url}")
r.exception.printStackTrace()
}
},
!args.getBoolean("force"),
true
)
return success()
}
}