jekyll-categorize-pages 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 17aee4a8a5c4b14c937afe207f6a96fb6a84a002
-  data.tar.gz: 3fc1a8c032dad7609fba7d3d5296280988047c59
+  metadata.gz: 76f301d8bfd1a555042e66e8be6d3296b4031eb7
+  data.tar.gz: 624a2fe951c103566c9eb9c1b28306c822156706
 SHA512:
-  metadata.gz: af5d585f40f51321e000e593dbd98f7af6a221fe184b15c02ba8423748289a65fb56f4b6dffb032269cc1cb64b106699ea329285a84712b16338c1c4b5e18a90
-  data.tar.gz: ccd2933eae2fe8c84bf452205f3fa5d841e83f6796783f480d7f04c0fcab2d11a0d759be8f4f27816f284801b228d7d08645c2e6fa7caf6a06d1d699a9817b03
+  metadata.gz: a5836409dc0ea7275477bbb8b91486d1b9360ea14cf16ef377d6d9ac800cf2494785ee1b89ad397285b8a052d11356071c8d89d48fcf6b8d25c7436311c152d8
+  data.tar.gz: ca023fa10dc6def534d9b51691142fe0f327dba83a0336579358811ee8211ee0bf3522ec439c8c3a4fb72c323a6f2c605908896e480458e47dbff86e7171e37f

data/lib/generator.rb

@@ -2,60 +2,107 @@
 require "jekyll"
 
 module Jekyll
-  
+  # You can create a generator when you need Jekyll
+  # to create additional content based on your own rules.
+
+  # A generator is a subclass of Jekyll::Generator that defines a generate method,
+  # which receives an instance of Jekyll::Site. The return value of generate is ignored.
+
+
   # Injects front matter defaults to set default category/sub-category values
-  class CategorizePagesGenerator < Jekyll::Generator
+  class InjectFm < Jekyll::Generator
     # priority :high
     safe true
 
-    def log(*args)
-      puts args
-    end
-
     def get_collection(name)
       @site.collections[name] ? @site.collections[name].docs : []
     end
 
-    def add_to_frontmatter(item)
-      item.data["category"] = @category # i.data is the front matter header
-      item.data["sub_category"] = @sub_category
+    def add_to_frontmatter(item, values)
+      # puts "adding: #{item.relative_path}"
+      # puts "values: #{values}"
+
+      item.data["category"] = values[:category] # i.data is the front matter header
+      item.data["sub_category"] = values[:sub_category]
+    end
+
+    def add_category_only(item, values)
+      # puts "adding: #{item.relative_path}"
+      # puts "values: #{values}"
+      item.data["category"] = values[:category]
+      # item.data["sub_category"] = values[:sub_category].slice(0, values[:sub_category].length - 3)
     end
 
     
     def isValid?(item)
-        folders_to_categorize = @site.data["pagerrr_folders_to_categorize"] || []
-        puts folders_to_categorize
-        # all_categories.include?(page[:n])
+        all_categories = @site.data["all_categories"] || [] # sub folders inside collection
         path = item.relative_path.split('/')
-        # category = path[0].sub(/^_/, '') # Remove first letter if underscore
-        @category = path[1] if path[1] != "index.md"
-        @sub_category = path[2] if path[2] != "index.md"
-        file = path[3] if path[3] != "index.md"
-        # puts '^^^^^^^^^^^^^'
-        # puts @category
-        # puts @sub_category
-
-      return folders_to_categorize.include?(@category)
+
+        category = path[1] if path[1] != "index.md"
+        
+        # ignore index.md pages
+        if path.include?("index.md")
+          return false
+        end
+
+        # if category folder has no sub categories
+        if path.length == 3
+          # puts "path: #{path}"
+          # puts "length: #{path.length}"
+          header_values = get_header_values_to_add(item)
+          add_category_only(item, header_values)
+          return false
+        end
+
+        return all_categories.include?(category)
+
+        # if path.include?("index.md")
+        #   return false
+        # else
+        #   return all_categories.include?(category)
+        # end
     end
 
 
+    def get_header_values_to_add(item)
+      all_categories = @site.data["all_categories"] || [] # sub folders inside collection
+      # puts "-----------------------"
+      # puts "all_categories: #{all_categories}"
+      # puts "item.relative_path: #{item.relative_path}"
+
+      path = item.relative_path.split('/')
+      category = path[1] if path[1] != "index.md"
+      # puts "@category: #{@category}"
+      sub_category = path[2] if path[2] != "index.md"
+      # puts "@sub_category: #{@sub_category}"
+      file = path[3] if path[3] != "index.md"
+      # puts "file: #{file}"
+
+      return {
+        :category => category,
+        :sub_category => sub_category,
+        :file => file
+      }
+    end
 
     def generate(site)
-      puts "RAN!"
-      # @site = site
-      # # The collections to add category/sub category front matter values
-      # collections = ["pages"] # TODO: Move this variable to _config/_data
-
-      # for c in collections
-      #   for item in get_collection(c)
-      #     valid_category = isValid?(item)
-      #     if valid_category
-      #       add_to_frontmatter(item)
-      #     end
-      #   end
-      # end
+      @site = site
+      # The collections to add category/sub category front matter values
+      collections = ["pages"] # TODO: Move this variable to _config/_data
+
+      for c in collections
+        for item in get_collection(c)
+          valid_category = isValid?(item)
+          if valid_category
+            header_values = get_header_values_to_add(item)
+            # puts "header_values: #{header_values}"
+            add_to_frontmatter(item, header_values)
+          end
+        end
+      end
 
     end
 
   end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jekyll-categorize-pages
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - ''