advent 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d20f95e7e3d0a52a78b93287b93c1864643d94abb70155af037434e254ef18b4
-  data.tar.gz: 1c4452e0c6b36b2cda5687398c18f819038b06e7cb71056293b8f427246542d9
+  metadata.gz: 360b1871cc09fd664dd8562fd1a29afd9b2e8bbc7a83d9d711c7933a6c62b5ec
+  data.tar.gz: 2f9cde07fa43650354a3f11c7cb869efa7c969d4254083ca807bd7a855fcf9d0
 SHA512:
-  metadata.gz: 9e39e7bc77607f5cb2d056377b45a763f6564f3205314e9000a128a53faa6cd758e5c427f9e6255d57a495cf971793703ba45a5eb2dfbb85bf6e15d5e21b1143
-  data.tar.gz: 626cb6bb873ac8c5927a302b155c1f2239207e054b8eb186764a0e91929a8afe009af3cf5fb1abd5dd282de36043b9878d7d0d9266dcb4bb6be29b05fa0d8fee
+  metadata.gz: e27b9323e1ad7a07484fe9302dfd49b430bd4fc5433ef0e412e1af14f734bdd1c0a8c719803e108e4b7b3aec144ec942119bc6a07bc93381f4aae9ef634324b1
+  data.tar.gz: b8e99258c57406a1cf945e868a1642d8ed8636c983fd960fd3f889bc6760c170bb346e3cdb0cc82e547f880dbd8c6f71898e941defdfd41b97a73974bc4cdec0

data/README.md

@@ -16,6 +16,26 @@ If bundler is not being used to manage dependencies, install the gem by executin
 
 ## Usage
 
+Advent expects you to have a working directory resembling something like:
+
+    $ tree
+    .
+    ├── 2015
+    └── 2016
+
+Some commands can be run from within a directory for a specific year, but it's
+better to run from the parent directory where possible.
+
+The typical flow for tackling a daily challenge would be:
+
+    $ advent generate 2015 1 # generate files to work in
+    $ advent download 2015 1 # download the input file
+
+    $ vim 2015/day1.rb 2015/day1_test.rb # do your work
+
+    $ ruby 2015/day1_test.rb # run any tests you may have
+    $ advent solve 2015/day1.rb # get your answers to submit
+
 A list of commands and help is available using `advent`:
 
     $ advent help

data/lib/advent/cli.rb

@@ -10,6 +10,7 @@ module Advent
     include Thor::Actions
 
     class_option :root_path, default: Dir.pwd, hide: true, check_default_type: false
+    class_option :http_module, default: Net::HTTP, check_default_type: false
 
     def initialize(*args)
       super
@@ -30,13 +31,37 @@ module Advent
         dir = root_path.basename.to_s
         dir =~ /^20[0-9]{2}/
       end
+    end
+
+    desc "download YEAR DAY", "Download the input for YEAR and DAY"
+    def download(year_or_day, day = nil)
+      year, day = determine_year_and_day(year_or_day, day)
+
+      if (error_message = validate(year, day))
+        say_error error_message, :red
+        return
+      end
+
+      subpath = if in_year_directory?
+        ""
+      else
+        "#{year}/"
+      end
+
+      unless Advent.session.exist?
+        session = ask "What is your Advent of Code session cookie value?", echo: false
+        Advent.session.value = session
+
+        say "\n\nThanks. Psst, we're going to save this for next time. It's in .advent_session if you need to update or delete it.\n\n"
+      end
 
-      def root_path
-        @_root_path ||= if options.root_path.is_a?(Pathname)
-          options.root_path
-        else
-          Pathname.new(options.root_path)
-        end
+      input = Advent::Input.new(root_path.join(subpath), day: day.to_i)
+
+      if input.download(Advent.session.value, options.http_module)
+        say "Input downloaded to #{input.file_path}.", :green
+        say "\nUsing #load_input in your daily solution will load the input file for you."
+      else
+        say_error "Something went wrong, maybe an old session cookie?", :red
       end
     end
 
@@ -45,11 +70,7 @@ module Advent
     # is used, otherwise both the year and day will be required to generate the
     # output.
     def generate(year_or_day, day = nil)
-      year, day = if in_year_directory?
-        [root_path.basename.to_s, parse_number(year_or_day)]
-      else
-        [year_or_day, parse_number(day)]
-      end
+      year, day = determine_year_and_day(year_or_day, day)
 
       if (error_message = validate(year, day))
         say_error error_message, :red
@@ -81,6 +102,22 @@ module Advent
 
     private
 
+    def determine_year_and_day(year_or_day, day)
+      if in_year_directory?
+        [root_path.basename.to_s, parse_number(year_or_day)]
+      else
+        [year_or_day, parse_number(day)]
+      end
+    end
+
+    def root_path
+      @_root_path ||= if options.root_path.is_a?(Pathname)
+        options.root_path
+      else
+        Pathname.new(options.root_path)
+      end
+    end
+
     def parse_number(str)
       if (m = str.match(/[0-9]+/))
         m[0]

data/lib/advent/input.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "cgi/cookie"
+require "fileutils"
+require "net/http"
+require "uri"
+
+module Advent
+  class Input
+    def initialize(dir, day:)
+      @dir = dir
+      @day = day
+    end
+
+    def file_path
+      @dir.join(".day#{@day}.input.txt")
+    end
+
+    def exist?
+      File.exist? file_path
+    end
+
+    def download(session, http = Net::HTTP)
+      session_cookie = CGI::Cookie.new("session", session)
+      response = http.get_response(input_url, {"Cookie" => session_cookie.to_s})
+
+      if success?(response)
+        FileUtils.mkdir_p file_path.dirname
+        File.write file_path, response.body
+        true
+      else
+        false
+      end
+    end
+
+    private
+
+    def input_url
+      URI("https://adventofcode.com/#{year}/day/#{@day}/input")
+    end
+
+    def year
+      @_year ||= @dir.basename
+    end
+
+    def success?(response)
+      response.code >= "200" && response.code < "300"
+    end
+  end
+end

data/lib/advent/session.rb

@@ -0,0 +1,28 @@
+module Advent
+  # Simple class for handling the session cookie file
+  class Session
+    FILE_NAME = ".advent_session"
+
+    attr_reader :file_name
+
+    def initialize(file_name = FILE_NAME)
+      @file_name = file_name
+    end
+
+    def clear
+      File.delete file_name if exist?
+    end
+
+    def exist?
+      File.exist? file_name
+    end
+
+    def value=(val)
+      File.write file_name, val
+    end
+
+    def value
+      File.read file_name if exist?
+    end
+  end
+end

data/lib/advent/solution.rb

@@ -16,11 +16,12 @@ module Advent
 
       @year = year
       @day = day
+
+      @input = Input.new(source_location.dirname, day: day)
     end
 
     def load_input
-      dir = source_location.dirname
-      File.read(dir.join(".day#{@day}.input.txt"))
+      File.read(@input.file_path)
     end
 
     private

data/lib/advent/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Advent
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

data/lib/advent.rb

@@ -1,9 +1,17 @@
 # frozen_string_literal: true
 
+require_relative "advent/input"
+require_relative "advent/session"
 require_relative "advent/solution"
 require_relative "advent/test_case"
 require_relative "advent/version"
 
 module Advent
   class Error < StandardError; end
+
+  class << self
+    def session
+      @_session = Session.new
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: advent
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Daniel Grieve
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-11-22 00:00:00.000000000 Z
+date: 2022-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -39,6 +39,8 @@ files:
 - lib/advent.rb
 - lib/advent/cli.rb
 - lib/advent/cli/solver.rb
+- lib/advent/input.rb
+- lib/advent/session.rb
 - lib/advent/solution.rb
 - lib/advent/templates/solution.rb.tt
 - lib/advent/templates/solution_test.rb.tt