module S3Relay
  class Upload

    include Mongoid::Document

    field :uuid
    field :user_id, type: BSON::ObjectId
    field :parent_type, type: String
    field :parent_id, type: BSON::ObjectId
    field :upload_type, type: String
    field :filename, type: String
    field :content_type, type: String
    field :state, type: String
    field :data, type: Hash, default: Hash.new
    field :pending_at, type: DateTime
    field :imported_at, type: DateTime

    belongs_to :parent, polymorphic: true, optional: true

    validates :uuid,         presence: true, uniqueness: true
    validates :upload_type,  presence: true
    validates :filename,     presence: true
    validates :content_type, presence: true
    validates :pending_at,   presence: true

    after_initialize :finalize
    after_create :notify_parent

    def self.pending
      where(state: "pending")
    end

    def self.imported
      where(state: "imported")
    end

    def pending?
      state == "pending"
    end

    def imported?
      state == "imported"
    end

    def mark_imported!
      update_attributes(state: "imported", imported_at: Time.now)
    end

    def notify_parent
      return unless parent.present?

      if parent.respond_to?(:import_upload)
        parent.import_upload(id)
      end
    end

    def public_url
      S3Relay::PrivateUrl.new(uuid, filename).public_url
    end

    def public_url_500x500
      file_full_s3_key = public_url.split('amazonaws.com/')[1]
      "http://epartrade.s3-website-us-west-1.amazonaws.com/500x500/#{file_full_s3_key}"
    end

    def public_url_with_size size
      file_full_s3_key = public_url.split('amazonaws.com/')[1]
      "http://epartrade.s3-website-us-west-1.amazonaws.com/#{size}/#{file_full_s3_key}"
    end

    def private_url
      S3Relay::PrivateUrl.new(uuid, filename).generate
    end

    def as_json(options = { })
      h = super(options)
      h[:public_url] = public_url
      h[:public_url_500x500] = public_url_500x500
      h
    end

    private

    def finalize
      self.state      ||= "pending"
      self.pending_at ||= Time.now
    end

  end
end