Initial commit

2020-12-01 02:12:55 +01:00
commit de750de96a
20 changed files with 516 additions and 0 deletions
@@ -0,0 +1,56 @@
+import kong.unirest.Unirest
+import org.gradle.api.GradleException
+import org.gradle.api.Plugin
+import org.gradle.api.Project
+import org.gradle.api.tasks.SourceSetContainer
+import org.gradle.kotlin.dsl.create
+import org.gradle.kotlin.dsl.getByType
+import java.io.File
+import java.time.LocalDateTime
+
+open class AdventOfCodeExtension {
+    var session: String? = null
+    var year: Int = 2020
+}
+
+class AdventOfCodePlugin : Plugin<Project> {
+    override fun apply(project: Project) {
+        project.tasks.create("aoc") {
+            group = "Advent of Code"
+            description = "Download the input of the day"
+
+            val extension = project.extensions.create<AdventOfCodeExtension>("adventOfCode")
+
+            doLast {
+                if (extension.session == null) {
+                    throw GradleException("advent of code session not set")
+                }
+
+                val resourceDir: File = project
+                    .project(":days")
+                    .extensions
+                    .getByType<SourceSetContainer>()
+                    .getByName("main")
+                    .resources
+                    .srcDirs
+                    .first()
+
+                val currentDay = LocalDateTime.now().dayOfMonth.toString()
+                val outFile = File(resourceDir, "day" + currentDay.padStart(length = 2, padChar = '0') + ".txt")
+                val url = "https://adventofcode.com/${extension.year}/day/$currentDay/input"
+
+                Unirest.get(url)
+                    .cookie("session", extension.session)
+                    .header("Accept", "text/plain")
+                    .asString()
+                    .ifFailure {
+                        throw GradleException("Failed to get input, got status ${it.status}")
+                    }
+                    .ifSuccess {
+                        outFile.writeText(it.body)
+                        println("Saved input for day $currentDay")
+                    }
+            }
+        }
+    }
+}