added 260907 version of worker toolkit

worker-toolkit-flaredown/repo/backend/app/models/checkin.rb

@@ -0,0 +1,165 @@
+class Checkin
+  include Mongoid::Document
+
+  HBI_PERIODICITY = 7
+  PR_PERIODICITY = 7
+  PR_START_FROM = 7
+
+  attr_accessor :includes
+
+  FIELD_TYPE = %w[tag food].freeze
+
+  #
+  # Fields
+  #
+  field :date, type: DateTime
+  field :food_ids, type: Array, default: []
+  field :note, type: String
+  field :postal_code, type: String
+  field :tag_ids, type: Array
+  field :weather_id, type: Integer
+  field :encrypted_user_id, type: String, encrypted: {type: :integer}
+  field :position_id, type: Integer
+  field :promotion_skipped_at, type: Date
+
+  #
+  # Relations
+  #
+  has_one :harvey_bradshaw_index
+  has_one :promotion_rate, dependent: :destroy
+  has_many :treatments, class_name: "Checkin::Treatment"
+  has_many :conditions, class_name: "Checkin::Condition"
+  has_many :symptoms, class_name: "Checkin::Symptom"
+  accepts_nested_attributes_for :conditions, :symptoms, :treatments, allow_destroy: true
+
+  #
+  # Indexes
+  #
+  index(encrypted_user_id: 1)
+  index(date: 1, encrypted_user_id: 1)
+
+  #
+  # Validations
+  #
+  validates :encrypted_user_id, presence: true
+  validates :date, presence: true
+
+  #
+  # Scopes
+  #
+  scope :by_date, ->(startkey, endkey) { where(:date.gte => startkey, :date.lte => endkey) }
+
+  def user
+    @user ||= User.find(user_id)
+  end
+
+  def weather
+    @weather ||= Weather.find_by(id: weather_id)
+  end
+
+  def position
+    @position ||= Position.find_by(id: position_id)
+  end
+
+  def tags
+    if includes
+      @_tags_included ||= Tag.where(id: tag_ids - (includes[:tags] || []))
+    else
+      @_tags ||= Tag.where(id: tag_ids)
+    end
+  end
+
+  def foods
+    if includes
+      @_foods_included ||= Food.where(id: food_ids - (includes[:foods] || []))
+    else
+      @_foods ||= Food.where(id: food_ids)
+    end
+  end
+
+  def available_for_hbi?
+    return true if harvey_bradshaw_index
+    return true unless latest_hbi
+
+    HBI_PERIODICITY - ((latest_hbi.date)...date).count < 1
+  end
+
+  def available_for_promotion?
+    return true if promotion_rate
+    return false if user_has_already_rated?
+    return false unless date.today?
+    return ready_for_promotion? if latest_skipped_pr_at.blank?
+
+    PR_PERIODICITY - ((latest_skipped_pr_at)...date).count < 1
+  end
+
+  class Condition
+    include Mongoid::Document
+    include Checkin::Trackable
+    include Checkin::Fiveable
+
+    field :condition_id, type: Integer
+
+    validates :condition_id, uniqueness: {scope: :checkin_id}
+  end
+
+  class Symptom
+    include Mongoid::Document
+    include Checkin::Trackable
+    include Checkin::Fiveable
+
+    field :symptom_id, type: Integer
+
+    validates :symptom_id, uniqueness: {scope: :checkin_id}
+  end
+
+  class Treatment
+    include Mongoid::Document
+    include Checkin::Trackable
+
+    #
+    # Fields
+    #
+    field :treatment_id, type: Integer
+    field :value, type: String
+    field :is_taken, type: Boolean
+
+    #
+    # Indexes
+    #
+    index(treatment_id: 1)
+    index(treatment_id: 1, is_taken: 1, value: 1)
+
+    #
+    # Validations
+    #
+    validates :treatment_id, uniqueness: {scope: :checkin_id}
+  end
+
+  def self.ids_by_category_attrs(category_name, trackable_id)
+    where(id: {"$in" => "Checkin::#{category_name.camelize}".constantize
+      .where(:"#{category_name}_id" => trackable_id, :value.ne => nil).pluck(:checkin_id)})
+  end
+
+  private
+
+  def latest_hbi
+    @_latest_hbi ||= HarveyBradshawIndex.where(encrypted_user_id: encrypted_user_id).order(date: :desc).first
+  end
+
+  def latest_skipped_pr_at
+    @_lates_skipped_pr ||=
+      Checkin
+        .where(encrypted_user_id: encrypted_user_id)
+        .order_by(promotion_skipped_at: :desc)
+        .first&.promotion_skipped_at
+  end
+
+  def ready_for_promotion?
+    user.created_at <= PR_START_FROM.day.ago
+  end
+
+  def user_has_already_rated?
+    @_rated ||= PromotionRate.where(encrypted_user_id: encrypted_user_id).any?
+  end
+end