pion 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d488fdce446400ab2ebc04a1a49467b2e4581217
-  data.tar.gz: 202de5134d57fb642ed01c2a1decf751d8c26572
+  metadata.gz: 84a7550fcbc0110049a63b42b3fc1c715c11660d
+  data.tar.gz: e726c4dc3f929192d4046749ce3855fa708d6427
 SHA512:
-  metadata.gz: 20b0985951ee2799029e3bcb78e386cdfd01805c0981799ede57eaa62896cb965875c6570f1059677eb6b3f212d2c10a7ffde69ee2ac76d00c7f5ddeff05f5f7
-  data.tar.gz: 2d75ffb4cb653ac648fbf66d02618e28efba5b4624eec6376c1ed2f7d7462a3f350eb24d03fbdec427f6763dbd3a6ab79396feb789f5ae9faf571f2a0759411c
+  metadata.gz: d622f4258b7f3cde7451d896863e2aba88de7e5c19306b92698310649ba790b02bbf9d2583d52e32d1e91fad248ef308edb228de8c798ae7992ad119d2bf7294
+  data.tar.gz: c434e48b6c03c92704895aba3d5b5fc46a9a40c85612c5b1edc81f2bfee318acb0e116e309bf58bf8105b3a31da797c5fe1dee40b4a689ab12f06bff2a835c41

data/lib/pion.rb

@@ -1,23 +1,28 @@
+# Pion!
 class Pion
+  # Attributes
+  attr_accessor :options
+
   # Version number for Pion
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 
   # Initialize a new Pion object
-  def initialize(directory, extension = "html", separator = "/")
-    raise ArgumentError, "Extension name must be a string." unless extension.is_a? String
-    raise ArgumentError, "Directory separator must be a string." unless separator.is_a? String
+  def initialize(options)
+    @options = method(:symbolize).call options
+
+    fail ArgumentError, "No directory given!" unless options.key? :directory
 
-    @directory = method(:get_directory).call directory
-    @extension = extension.gsub /\A\./, ""
-    @separator = separator
+    method(:set_options).call
   end
 
   # Generate urls from files
   def generate
     urls = []
 
-    Dir.glob "#{@directory}/**/*.#{@extension}" do |file|
-      urls << {file: file, url: method(:get_url).call(file)} unless file == "." or file == ".."
+    Dir.glob "#{options[:directory]}/**/*.#{options[:extension]}" do |file|
+      hash = { file: file, url:  method(:get_url).call(file) }
+
+      urls << hash unless file == "." || file == ".."
     end
 
     urls
@@ -25,20 +30,52 @@ class Pion
 
   private
 
-  # Get directory or raise errors if it's faulty
-  def get_directory(directory)
-    return directory if directory.is_a? String and File.directory? directory
-
-    raise ArgumentError, "Directory name must be a string." unless directory.is_a? String
-    raise ArgumentError, "Directory must be a valid directory." unless File.directory? directory
-  end
-
   # Convert filename to clean url
   def get_url(file)
-    url = file.gsub /\A#{@directory}/, ""
-    url = url.gsub /(\/index)?\.#{@extension}\z/, ""
-    url = url.gsub /#{Regexp.escape(@separator)}/, "/"
+    url = file.gsub(/\A#{options[:directory]}/, "")
+    url = url.gsub(/(\/index)?\.#{options[:extension]}\z/, "")
+    url = url.gsub(/#{Regexp.escape(options[:separator])}/, "/")
 
     url == "" ? "/" : url
   end
-end
+
+  # Set directory or fail errors if it's faulty
+  def set_directory
+    dir = options[:directory]
+
+    return if dir.is_a?(String) && File.directory?(dir)
+
+    if !dir.is_a? String
+      fail ArgumentError, "Directory must be a string."
+    elsif !File.directory? dir
+      fail ArgumentError, "Directory must be a valid directory."
+    end
+  end
+
+  def set_default(type, default)
+    options[type] = default unless options.key? type
+
+    name = type.to_s.capitalize
+
+    unless options[type].is_a? String
+      fail ArgumentError, "#{name} must be a string"
+    end
+  end
+
+  def set_options
+    method(:set_directory).call
+
+    method(:set_default).call :extension, "html"
+    method(:set_default).call :separator, "/"
+  end
+
+  def symbolize(hash)
+    fail ArgumentError, "No Hash given!" unless hash.is_a? Hash
+
+    hash.reduce({}) do |memo, (key, value)|
+      memo[key.to_sym] = value
+
+      memo
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pion
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Oscar Palmér