add features for marine's program

Author: dorianmariecom

lib/code/object/list.rb

@@ -227,8 +227,8 @@ def call(**args)
           sig(args)
           code_sum
         when "uniq"
-          sig(args)
-          code_uniq
+          sig(args) { (Function | Class).maybe }
+          code_uniq(code_value, **globals)
         when "many?"
           sig(args)
           code_many?
@@ -1001,8 +1001,39 @@ def code_size
         Integer.new(raw.size)
       end
 
-      def code_uniq
-        List.new(raw.uniq)
+      def code_uniq(argument = nil, **globals)
+        code_argument = argument.to_code
+
+        unless code_argument.is_a?(Function) || code_argument.is_a?(Class)
+          return List.new(raw.uniq)
+        end
+
+        if code_argument.is_a?(Class)
+          return List.new(
+            raw
+              .select { |code_element| code_element.is_a?(code_argument.raw) }
+              .uniq
+          )
+        end
+
+        index = 0
+
+        List.new(
+          raw.uniq do |code_element|
+            if code_argument.is_a?(Function)
+              code_argument
+                .call(
+                  arguments: List.new([code_element, Integer.new(index), self]),
+                  **globals
+                )
+                .tap { index += 1 }
+            else
+              code_element.tap { index += 1 }
+            end
+          rescue Error::Next => e
+            e.code_value.tap { index += 1 }
+          end
+        )
       end
 
       def code_sum

lib/code/object/time.rb

@@ -74,6 +74,17 @@ def initialize(*args, **_kargs, &_block)
           self.raw = args.first.to_time.in_time_zone(::Time.zone)
         elsif args.first.is_a?(::ActiveSupport::TimeWithZone)
           self.raw = args.first.dup
+        elsif args.first.is_a?(Integer) || args.first.is_a?(Decimal) ||
+            args.first.is_a?(::Integer) || args.first.is_a?(::Float) ||
+            args.first.is_a?(::BigDecimal)
+          code_value = args.first.to_code
+          timestamp =
+            if code_value.is_a?(Decimal)
+              code_value.raw.to_r
+            else
+              code_value.raw
+            end
+          self.raw = ::Time.zone.at(timestamp)
         else
           self.raw = ::Time.zone.now
         end

spec/code/object/time_spec.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+
+RSpec.describe Code::Object::Time do
+  [
+    ['Time.new(1).format("%s")', ":1"],
+    ['Time.new(1100042342).format("%s")', ":1100042342"],
+    ['Time.new(1.2).format("%s.%L")', '"1.200"'],
+    ['Time.new(11212.1212).format("%s.%L")', '"11212.121"']
+  ].each do |input, expected|
+    it "#{input} == #{expected}" do
+      expect(Code.evaluate(input)).to eq(Code.evaluate(expected))
+    end
+  end
+end

spec/code_spec.rb

@@ -284,6 +284,9 @@
     ["[1, 2, 3].map { |i| next(0) if i.even? i ** 2}", "[1, 0, 9]"],
     ["[1, 2, 3].select { |n| n.even? }", "[2]"],
     ["[1, 2, 3].select { |n| n.even? }.select { |n| n.odd? }", "[]"],
+    ["[1, 2, 3, 4].uniq { |n| n % 2 }", "[1, 2]"],
+    ['[1, "1", 2, "2"].uniq(String)', '["1", "2"]'],
+    ["[1, 2, 3, 4].uniq(String)", "[]"],
     ["[1, 2].map(&:to_string)", "[:1, :2]"],
     ["[1, 2].map(&:to_string)", '["1", "2"]'],
     ["[1, 2].map(&:to_string)", '["1", "2"]'],