# == Schema Information
#
# Table name: easy_ml_model_files
#
#  id            :bigint           not null, primary key
#  filename      :string           not null
#  path          :string           not null
#  configuration :json
#  model_type    :string
#  model_id      :bigint
#  created_at    :datetime         not null
#  updated_at    :datetime         not null
#
module EasyML
  class ModelFile < ActiveRecord::Base
    self.table_name = "easy_ml_model_files"
    include Historiographer::Silent
    historiographer_mode :snapshot_only

    self.filter_attributes += [:configuration]

    validates :filename, presence: true
    belongs_to :model, class_name: "EasyML::Model"

    include EasyML::Concerns::Configurable
    add_configuration_attributes :s3_bucket, :s3_prefix, :s3_region, :s3_access_key_id, :s3_secret_access_key, :root_dir

    def synced_file
      EasyML::Support::SyncedFile.new(
        filename: filename,
        s3_bucket: s3_bucket,
        s3_prefix: s3_prefix,
        s3_region: s3_region,
        s3_access_key_id: s3_access_key_id,
        s3_secret_access_key: s3_secret_access_key,
        root_dir: root_dir,
      )
    end

    def exist?
      fit?
    end

    def fit?
      return false if root_dir.nil?
      return false if full_path.nil?

      File.exist?(full_path)
    end

    def read
      File.read(full_path)
    end

    def upload(path)
      synced_file.upload(path)
      set_path(path)
    end

    def set_path(path)
      path = get_full_path(path)
      basename = Pathname.new(path).basename.to_s
      unless path.start_with?(full_dir)
        new_path = File.join(full_dir, basename).to_s
        FileUtils.mkdir_p(Pathname.new(new_path).dirname.to_s)
        FileUtils.cp(path, new_path)
        path = new_path
      end
      self.filename = basename
      self.path = get_relative_path(path)
    end

    def get_full_path(path)
      path = path.to_s

      path = Rails.root.join(path) unless path.match?(Regexp.new(Rails.root.to_s))
      path
    end

    def get_relative_path(path)
      path = path.to_s
      path = path.to_s.split(Rails.root.to_s).last
      path.to_s.split("/")[0..-2].reject(&:empty?).join("/")
    end

    def download
      return unless full_path.present?

      synced_file.download(full_path) unless File.exist?(full_path)
      full_path
    end

    def sha
      Digest::SHA256.file(full_path).hexdigest
    end

    def full_path(filename = nil)
      filename = self.filename if filename.nil?
      return nil if filename.nil?
      return nil if relative_dir.nil?

      Rails.root.join(relative_dir, filename).to_s
    end

    def relative_dir
      root_dir.to_s.gsub(Regexp.new(Rails.root.to_s), "").gsub!(%r{^/}, "")
    end

    def full_dir
      Rails.root.join(relative_dir).to_s
    end

    def model_root
      File.expand_path("..", full_dir)
    end

    def cleanup!
      [model_root].each do |dir|
        EasyML::Support::FileRotate.new(dir, []).cleanup(extension_allowlist)
      end
    end

    def cleanup(files_to_keep)
      [model_root].each do |dir|
        EasyML::Support::FileRotate.new(dir, files_to_keep).cleanup(extension_allowlist)
      end
    end

    def extension_allowlist
      %w[bin model json]
    end
  end
end