57 lines
989 B
Go
57 lines
989 B
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
func GenInputFile(day int) string {
|
|
var d string
|
|
if day < 10 {
|
|
d = fmt.Sprintf("0%d", day)
|
|
} else {
|
|
d = fmt.Sprintf("%d", day)
|
|
}
|
|
|
|
pwd, _ := os.Getwd()
|
|
path := fmt.Sprintf("%s\\day%s\\input.txt", pwd, d)
|
|
fi, _ := os.Stat(path)
|
|
if fi != nil {
|
|
return path
|
|
}
|
|
|
|
fmt.Println("Creating new input file...")
|
|
f, _ := os.OpenFile(path, os.O_CREATE, 0700)
|
|
f.WriteString(readHttp(2022, day))
|
|
return path
|
|
}
|
|
|
|
func readHttp(year, day int) string {
|
|
fmt.Println("Fetching data into file...")
|
|
|
|
url := fmt.Sprintf("https://adventofcode.com/%d/day/%d/input", year, day)
|
|
session := os.Getenv("sessionAoC")
|
|
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
req.AddCookie(&http.Cookie{Name: "session", Value: session})
|
|
client := &http.Client{}
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return string(body)
|
|
}
|
|
|