enhanced: import and export, and minor corrections

Author: Yuri-Rassokhin

lib/flex-cartesian.rb

@@ -5,7 +5,7 @@
 require 'yaml'
 
 module FlexOutput
-  def output(separator: " | ", colorize: false, align: false)
+  def output(separator: " | ", colorize: false, align: true)
     return puts "(empty struct)" unless respond_to?(:members) && respond_to?(:values)
 
     values_list = members.zip(values.map { |v| v.inspect })
@@ -35,6 +35,10 @@ def add_function(name, &block)
     @derived[name.to_sym] = block
   end
 
+  def remove_function(name)
+    @derived.delete(name.to_sym)
+  end
+
   def cartesian(dims = nil, lazy: false)
   dimensions = dims || @dimensions
   return nil unless dimensions.is_a?(Hash)
@@ -90,7 +94,7 @@ def progress_each(dims = nil, lazy: false, title: "Processing")
     end
   end
 
-  def output(separator: " | ", colorize: false, align: false, format: :plain, limit: nil)
+  def output(separator: " | ", colorize: false, align: true, format: :plain, limit: nil)
   rows = []
   cartesian do |v|
     rows << v
@@ -123,14 +127,42 @@ def output(separator: " | ", colorize: false, align: false, format: :plain, limi
   end
 end
 
-  def self.from_json(path)
+def import(path, format: :json)
+  data = case format
+  when :json
+    JSON.parse(File.read(path), symbolize_names: true)
+  when :yaml
+    YAML.safe_load(File.read(path), symbolize_names: true)
+  else
+    raise ArgumentError, "Unsupported format: #{format}. Only :json and :yaml are supported."
+  end
+
+  raise TypeError, "Expected parsed data to be a Hash" unless data.is_a?(Hash)
+
+  @dimensions = data
+  self
+end
+
+def export(path, format: :json)
+  case format
+  when :json
+    File.write(path, JSON.pretty_generate(@dimensions))
+  when :yaml
+    File.write(path, YAML.dump(@dimensions))
+  else
+    raise ArgumentError, "Unsupported format: #{format}. Only :json and :yaml are supported."
+  end
+end
+
+
+  def from_json(path)
     data = JSON.parse(File.read(path), symbolize_names: true)
-    new(data)
+    @dimensions = data
   end
 
-  def self.from_yaml(path)
+  def from_yaml(path)
     data = YAML.safe_load(File.read(path), symbolize_names: true)
-    new(data)
+    @dimensions = data
   end
 
 private