eikon 0.1.1

data/lib/eikon/video_processor.rb

@@ -0,0 +1,72 @@
+# typed: strict
+require "sorbet-runtime"
+require "terrapin"
+
+module Eikon
+  class VideoProcessor
+    extend T::Sig
+
+    sig { returns(String) }
+    attr_reader :file_name
+
+    sig { params(file_name: String).void }
+    def initialize(file_name)
+      @file_name = T.let(file_name, String)
+    end
+
+    sig { returns(String) }
+    def split_video_into_images
+      file_name = get_file_name(@file_name)
+
+      setup_tmp_folder
+
+      output_folder_path = "tmp/videos/#{file_name}_#{Time.now.utc.strftime("%Y%m%d%H%M%S")}"
+      begin
+        make_dir_command = Terrapin::CommandLine.new("mkdir", ":folder_name")
+        make_dir_command.run(folder_name: output_folder_path)
+      rescue Terrapin::ExitStatusError
+        raise "Folder already exists for this export"
+      end
+
+      # ffmpeg -i pexels-ron-lach-7121125.mp4 -vf fps=10/60 out%d.png
+      line = Terrapin::CommandLine.new("ffmpeg", "-i :file_name -vf fps=10/60 :folder_name")
+      line.run(file_name: @file_name, folder_name: "#{output_folder_path}/#{file_name}_%d.png")
+      output_folder_path
+    end
+
+    sig { params(folder_path: String).returns(T::Array[T::Hash[String, T.any(Integer, String)]]) }
+    def get_frames_dhash(folder_path)
+      # All the file lists start with `.` and `..` so we need to remove those
+      file_names = Dir.entries(folder_path).drop(2)
+
+      dhashes = file_names.map do |file_name|
+        dhash = Eikon.dhash_for_image("#{folder_path}/#{file_name}")
+        { file_name: file_name, dhash: dhash }
+      end
+      dhashes
+    end
+
+  private
+
+    sig { params(file_name: String).returns(String) }
+    def get_file_name(file_name)
+      file_name_parts = file_name.split(".")
+      file_name_parts.select! { |e| !e.empty? }
+      file_name = file_name_parts.first
+      T.must(T.must(file_name).split("/").last)
+    end
+
+    sig { void }
+    def setup_tmp_folder
+      begin
+        make_dir_command = Terrapin::CommandLine.new("mkdir", "tmp/")
+        make_dir_command.run
+      rescue Terrapin::ExitStatusError; end
+
+      begin
+        make_dir_command = Terrapin::CommandLine.new("mkdir", "tmp/videos")
+        make_dir_command.run
+      rescue Terrapin::ExitStatusError; end
+    end
+  end
+end

data/lib/eikon.rb

@@ -0,0 +1,35 @@
+# typed: strict
+require "eikon/version"
+require "eikon/comparator"
+require "eikon/video_processor"
+require "eikon/image_processor"
+require "vips"
+require "ruby_jard"
+require "sorbet-runtime"
+
+module Eikon
+  extend T::Sig
+
+  class Error < StandardError; end
+
+  sig { params(filename: T.untyped).returns(String) }
+  def self.dhash_for_image(filename)
+    image = Eikon::ImageProcessor.new(filename)
+    image.preprocess_image
+    image.generate_byte_array
+    image.byte_array
+  end
+
+  sig { params(filename: String).returns(T::Array[T::Hash[String, T.any(Integer, String)]]) }
+  def self.dhash_for_video(filename)
+    video_processor = Eikon::VideoProcessor.new(filename)
+    storage_path = video_processor.split_video_into_images
+    dhashes = video_processor.get_frames_dhash(storage_path)
+    dhashes
+  end
+
+  sig { params(filename_1: String, filename_2: String).returns(Integer) }
+  def self.distance_between_files(filename_1, filename_2)
+    Eikon::Comparator.compare(dhash_for_image(filename_1), dhash_for_image(filename_2))
+  end
+end

data/sorbet/config

@@ -0,0 +1,2 @@
+--dir
+.