Minor bug fixes to validate user input (#5)

Author: csessh

lua/aoc/api.lua

@@ -1,6 +1,5 @@
 local cache = require "aoc.cache"
 local curl = require "plenary.curl"
-local inspect = require "vim.inspect"
 local cfg = require "aoc.config"
 
 ---@class APIWrapper
@@ -22,12 +21,12 @@ local validate_args = function(day, year)
    local d = tonumber(day)
    local y = tonumber(year)
 
-   if d < 1 or d > 31 then
-      vim.api.nvim_err_writeln("Invalid day: " .. d)
+   if not d or d < 1 or d > 31 then
+      vim.api.nvim_err_writeln("Invalid day: " .. day)
       return false
    end
 
-   if y < 2015 or y > tonumber(os.date "%Y") then
+   if not y and y < 2015 or y > tonumber(os.date "%Y") then
       vim.api.nvim_err_writeln("Invalid year: " .. year)
       return false
    end

lua/aoc/cache.lua

@@ -1,5 +1,3 @@
-local inspect = require "vim.inspect"
-
 ---@class PuzzleCache
 local M = {}
 

lua/aoc/init.lua

@@ -5,6 +5,14 @@ local cache = require "aoc.cache"
 ---@class AOC
 local M = {}
 
+---Strip input of any leading/trailing spaces
+---@param s string
+---@return string
+local trim = function(s)
+   s, _ = string.gsub(s, "%s+", "")
+   return s
+end
+
 ---@param args any
 M.setup = function(args)
    cfg.init(args)
@@ -14,6 +22,8 @@ M.setup = function(args)
       local year = vim.fn.input "Year: "
       vim.api.nvim_out_write "\n"
 
+      day = trim(day)
+      year = trim(year)
       api.save_puzzle_input(day, year)
    end, {})