[ruby] Fix Psych Usage

endoflife-date · Oct 8, 2022 · b01fead · b01fead
1 parent 4bb80fd
commit b01fead
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/_plugins/create-json-files.rb b/_plugins/create-json-files.rb
@@ -10,15 +10,23 @@
 require 'fileutils'
 require 'json'
 require 'yaml'
-import 'date'
+require 'date'
 
 API_DIR = 'api'.freeze
 
+def load_yaml(file)
+  if YAML.respond_to?(:unsafe_load)
+    YAML.unsafe_load_file(file)
+  else
+    YAML.load_file(self[:encoded_value])
+  end
+end
+
 class Product
   attr_reader :hash
 
   def initialize(markdown_file)
-    @hash = YAML.load_file(markdown_file, permitted_classes: [Date])
+    @hash = load_yaml(markdown_file)
   end
 
   def permalink

diff --git a/_plugins/validate-releases.rb b/_plugins/validate-releases.rb
@@ -9,10 +9,18 @@
 # read as strings instead.
 STRING_KEYS = ['latest', 'releaseCycle']
 
+def load_yaml(file)
+  if YAML.respond_to?(:unsafe_load)
+    YAML.unsafe_load_file(file)
+  else
+    YAML.load_file(self[:encoded_value])
+  end
+end
+
 def process_files
   success = true
   Dir['products/*.md'].each do |markdown_file|
-    hash = YAML.load_file(markdown_file, permitted_classes: [Date])
+    hash = @hash = load_yaml(markdown_file)
     hash['releases'].each do |r|
       STRING_KEYS.each do |k|
         if r.key? k