mirror of https://github.com/captn3m0/muse-dl.git
Add URL to options
This commit is contained in:
parent
2c07c5940d
commit
d7043e233a
|
@ -2,7 +2,7 @@ require "./spec_helper"
|
|||
|
||||
describe Muse::Dl::Parser do
|
||||
it "should parse options" do
|
||||
parser = Muse::Dl::Parser.new(["--no-bookmarks", "--tmp-dir", "/tmp", "--no-cleanup", "--output", "file.pdf"])
|
||||
parser = Muse::Dl::Parser.new(["--no-bookmarks", "--tmp-dir", "/tmp", "--no-cleanup", "--output", "file.pdf", "https://muse.jhu.edu/book/68534"])
|
||||
parser.bookmarks.should eq false
|
||||
parser.tmp.should eq "/tmp"
|
||||
parser.cleanup.should eq false
|
||||
|
@ -10,10 +10,19 @@ describe Muse::Dl::Parser do
|
|||
end
|
||||
|
||||
it "should have reasonable defaults" do
|
||||
parser = Muse::Dl::Parser.new
|
||||
parser = Muse::Dl::Parser.new(["https://muse.jhu.edu/book/68534"])
|
||||
parser.bookmarks.should eq true
|
||||
parser.cleanup.should eq true
|
||||
parser.tmp.should eq "/tmp"
|
||||
parser.output.should eq "tempfilename.pdf"
|
||||
parser.url.should eq "https://muse.jhu.edu/book/68534"
|
||||
end
|
||||
|
||||
it "should raise error on missing options" do
|
||||
begin
|
||||
parser = Muse::Dl::Parser.new([] of String)
|
||||
rescue e : Exception
|
||||
e.class.should eq Muse::Dl::Errors::MissingLink
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -0,0 +1,4 @@
|
|||
module Muse::Dl::Errors
|
||||
class MissingLink < Exception
|
||||
end
|
||||
end
|
|
@ -3,7 +3,7 @@ require "./errors/*"
|
|||
|
||||
module Muse::Dl
|
||||
class Fetch
|
||||
def get_info(url : String) : (Muse::Dl::Book | Muse::Dl::Journal)
|
||||
def self.get_info(url : String) : (Muse::Dl::Book | Muse::Dl::Journal)
|
||||
match = /https:\/\/muse.jhu.edu\/(book|journal)\/(\d+)/.match url
|
||||
if match
|
||||
begin
|
||||
|
|
|
@ -11,7 +11,11 @@ module Muse::Dl
|
|||
|
||||
# TODO: Put your code here
|
||||
class Main
|
||||
def self.run
|
||||
def self.run(args : Array(String))
|
||||
parser = Parser.new(args)
|
||||
Fetch.get_info(parser.url)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
Muse::Dl::Main.run(ARGV)
|
||||
|
|
|
@ -1,11 +1,14 @@
|
|||
require "./errors/missing_link.cr"
|
||||
|
||||
module Muse::Dl
|
||||
class Parser
|
||||
@bookmarks : Bool
|
||||
@tmp : String
|
||||
@cleanup : Bool
|
||||
@output : String
|
||||
@url : String | Nil
|
||||
|
||||
getter :bookmarks, :tmp, :cleanup, :output
|
||||
getter :bookmarks, :tmp, :cleanup, :output, :url
|
||||
|
||||
def find_next(arg : Array(String), flag : String, default)
|
||||
search = arg.index flag
|
||||
|
@ -21,6 +24,11 @@ module Muse::Dl
|
|||
@cleanup = !arg.index "--no-cleanup"
|
||||
@tmp = find_next(arg, "--tmp-dir", "/tmp")
|
||||
@output = find_next(arg, "--output", "tempfilename.pdf")
|
||||
begin
|
||||
@url = arg[-1]
|
||||
rescue e : Exception
|
||||
raise Errors::MissingLink.new
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in New Issue