pshq_engine 0.0.1

data/app/models/token.rb

@@ -0,0 +1,22 @@
+class Token
+  # JWT_SECRET = Rails.application.secrets.jwt_secret
+  # JWT_ALGORITHM = Rails.application.secrets.jwt_algorithm
+
+  attr_reader :user_id, :payload
+
+  def initialize token
+  #   @payload = JWT.decode(token, JWT_SECRET, JWT_ALGORITHM).first.with_indifferent_access
+  #   @user_id = @payload[:user_id]
+  # rescue JWT::DecodeError
+  #   nil
+  end
+
+  # def valid?
+  #   user_id.presence && Time.now < Time.at(@payload[:exp].to_i)
+  # end
+
+  # def self.encode user_id
+  #   JWT.encode({ user_id: user_id, exp: (DateTime.now + 30).to_i }, JWT_SECRET, JWT_ALGORITHM)
+  # end
+
+end

data/app/models/transaction.rb

@@ -0,0 +1,2 @@
+class Transaction < ActiveRecord::Base
+end

data/app/models/unavailable.rb

@@ -0,0 +1,37 @@
+class Unavailable < ActiveRecord::Base
+  belongs_to :shift
+
+  validates_presence_of :reason
+  validates_presence_of :shift
+
+  after_save :notify_to_managers
+
+  def shift_detail
+    "AS #{shift.job_name} from #{shift.start_time.strftime('%H:%M')} to #{shift.end_time.strftime('%H:%M')} on #{shift.formatted_due_date}"
+  end
+  def approve(manager, cancellation)
+    self.approved_by = manager.id
+    self.approved_date = Time.zone.now
+    # self.notes = cancellation[:notes] if !cancellation[:notes].blank?
+    self.status = 'approved'
+    save
+  end
+  def deny(cancellation)
+    self.status = 'denied'
+    save
+  end
+  private
+  def notify_to_managers
+    if notify
+      managers = shift.schedule.organization.get_managers
+      managers.each do |manager|
+        manager.notify_message(self)
+      end
+    end
+  end
+
+
+
+
+
+end

data/app/models/user.rb

@@ -0,0 +1,392 @@
+class User < ActiveRecord::Base
+
+  
+  has_one :user_notification_setting
+  has_many :ratings, :class_name => "VendorRating"
+  has_many :repair_requests, :class_name => "AssetRepairRequest"
+  has_many :user_notifications, :class_name => "NotificationUser"
+  has_one :user_setting , :class_name => "UserNotificationSetting"
+  has_and_belongs_to_many :roles, :join_table => :users_roles
+  has_many :member, :dependent => :destroy
+  accepts_nested_attributes_for :member
+  has_many :organizations, :through => :member
+  accepts_nested_attributes_for :organizations
+  has_many :asset
+  belongs_to :vacation
+  has_one :user_vacation_info
+  has_many :recurring_time_offs
+  has_many :cancellations, :through => :shifts
+  has_many :alerts
+  has_and_belongs_to_many :certifications
+  has_many :swaps, :through => :shifts
+  has_many :tasks
+  has_many :timeoffs, :dependent => :destroy
+  has_many :recurring_time_offs, :dependent => :destroy
+  has_and_belongs_to_many :jobs
+  has_many :feedbacks
+  has_many :notes, :dependent => :destroy
+  accepts_nested_attributes_for :notes, :reject_if => lambda { |note| note[:content].blank? }
+  accepts_nested_attributes_for :certifications
+
+  has_many :schedules, :dependent => :destroy do
+    def current_week
+      cur_day = Time.zone.now.strftime('%Y-%m-%d')
+      where("start_date <= '#{cur_day}' and end_date >= '#{cur_day}'")
+    end
+
+    def week_schedule_by_date(date)
+      where("start_date <= '#{date}' and end_date >= '#{date}'")
+    end
+    
+    def not_template
+      where(:template_id => nil)
+    end
+    
+    def schedule_by_date(start_date,end_date )
+      where("start_date >= '#{start_date}' and end_date <= '#{end_date}'")
+    end
+
+    def published
+      where(:publish => true)
+    end
+  end
+
+  has_many :shifts, :through => :schedules
+  has_many :shift_trades, :through => :swaps do
+    def affirmatives
+      where(:status => 'affirmative')
+    end
+  end
+
+  has_many :candidates, :dependent => :destroy do
+    def initials
+      where(:status => 'initial')
+    end
+  end
+
+  has_attached_file :avatar, :styles => { :medium => "300x300!", :thumb => "100x100!" }, :default_url => "assets/missing.png"
+  validates_attachment_content_type :avatar, :content_type => /\Aimage\/.*\Z/
+  
+  #  validates_uniqueness_of :phone_number
+  #  validates_length_of :phone_number, :within => 6..13, if: Proc.new {|user| user.phone_number != nil}
+  validates_uniqueness_of :email
+
+  #  validates :first_name, :last_name  ,:email , :presence => true 
+  #  validates :email, :uniqueness  => true
+
+  include PgSearch
+  multisearchable   :against => [:first_name, :last_name  , :phone_number ,:email ]
+  pg_search_scope :custom_search,
+    :against => [:first_name, :last_name],
+    :using => {
+    :tsearch => {:prefix => true}
+  }
+  
+  rolify
+  
+  acts_as_messageable
+  
+  devise :database_authenticatable, :registerable, 
+    :recoverable, :rememberable,:timeoutable, :trackable, 
+    :validatable,:lockable, :invitable, :omniauthable, :omniauth_providers => [:facebook,:google_oauth2,:twitter], :timeout_in => 8.hours
+
+  # functions for Sripe of TrackMyAssets
+  def stripe_create(token, plan_id , coupon)
+    Stripe.api_key = "sk_test_eIdlWzka50kLJLNIqAIklamd"
+    already_coupon = Stripe::Coupon.all
+    already_coupon.data.each do |ex_coupon|
+      if ex_coupon and !ex_coupon.id.blank? and ex_coupon.id == coupon
+        @coupon_code = coupon
+      else
+        @coupon_code 
+      end
+    end
+    customer = Stripe::Customer.create(
+      :description => "http://trackmyassets.biz",
+      :plan => plan_id,
+      :source => token ,
+      :coupon =>  @coupon_code
+    )
+    return customer['id']
+  end
+ 
+  # Stripe Coupon Checking
+  def self.check_coupon(coupon_id)
+    Stripe.api_key = "sk_test_eIdlWzka50kLJLNIqAIklamd"
+    already_coupon = Stripe::Coupon.all
+    already_coupon.data.each do |ex_coupon|
+      if ex_coupon and !ex_coupon.id.blank? and ex_coupon.id == coupon_id
+        @coupon = coupon_id
+      else
+        
+      end
+    end
+    return @coupon
+  end
+  
+  def rating_for_vendors(vendor)
+    user_ratings_for_vendor = self.ratings.find_by_vendor_id(vendor.id)
+    if !user_ratings_for_vendor
+      user_ratings_for_vendor = vendor.ratings.new()
+      user_ratings_for_vendor.user_id = self.id
+      user_ratings_for_vendor.vendor_id = vendor.id
+    end
+    user_ratings_for_vendor
+  end
+  
+  def notifications
+    Notification.joins(:subscribers).where(notification_users: {user_id: self.id}).order(created_at: :desc)
+  end
+  
+  def unread_notifications_count
+    Notification.joins(:subscribers).where(notification_users: {user_id: self.id, seen: false}).order(created_at: :desc).count()
+  end
+    
+  def to_json
+    { 
+      :id => self.id,
+      :email => self.email,
+      :first_name => self.first_name,
+      :last_name => self.last_name,
+      :phone => self.phone_number,
+      #      :address => self.address,
+      :roles => self.roles,
+      :is_invited => !self.invitation_created_at.nil?,
+      :is_accepted => !self.invitation_accepted_at.nil?,
+      :is_locked => !self.locked_at.nil?
+    }
+  end
+
+  def reset_user_password_token
+    self.reset_password_token   = SecureRandom.hex(3)
+    self.reset_password_sent_at = Time.now.utc
+    self.save!(validate: false)
+    self.reset_password_token
+  end
+
+  def self.serialize_into_session(record)
+    [record.id.to_s, record.authenticatable_salt]
+  end
+  
+  def assign_api_token
+    update_attribute(:api_token, SecureRandom.uuid)
+  end
+  
+  def self.is_valid_user(email,password)
+    @user = User.find_by_email(email)
+    @user and @user.valid_password?(password) ? @user : nil
+  end
+
+  # Function for MT2W
+  def send_shift_reminder(min)
+    self.shifts.each do |shift|
+      d = shift.due_date
+      t = shift.start_time - min.to_i.minutes
+      dt = DateTime.new(d.year, d.month, d.day, t.hour, t.min, t.sec)
+      ShiftReminderWorker.perform_at(dt,self.email,"Your shift starts in #{min} minutes")
+    end
+  end
+
+  def self.enable
+    where(:disabled => false)
+  end
+  def self.find_for_facebook_oauth(auth)
+    where(auth.slice(:provider, :uid)).first_or_initialize.tap do |user|
+      already_user = User.where(:email => auth.info.email).first
+      if !user.persisted? and auth.info.email and already_user.blank?
+        user.provider = auth.provider
+        user.uid = auth.uid
+        user.email = auth.info.email
+        user.password = Devise.friendly_token[0,20]
+        user.first_name = auth.info.first_name
+        user.last_name = auth.info.last_name
+        #        user.username = auth.info.name   # assuming the user model has a name
+        user.avatar = process_uri(auth.info.image) # assuming the user model has an image
+        user.save(:validate=>false)
+        return user
+      elsif already_user
+        return already_user
+      else
+        return false
+      end
+    end
+  end
+
+  def self.process_uri(uri)
+    avatar_url = URI.parse(uri)
+    avatar_url.scheme = 'https'
+    avatar_url.to_s
+  end
+
+  def persisted?
+    !(new_record? || destroyed?)
+  end
+
+  def update_card(subscriber, card_info)
+    token = Stripe::Token.create(
+      card: {
+        number: card_info[:number],
+        exp_month: card_info[:exp_month],
+        exp_year: card_info[:exp_year],
+        cvc: card_info[:cvc]
+      }
+    )
+    customer = Stripe::Customer.retrieve(subscriber.customer_id)
+    card = customer.cards.create(card: token.id)
+    card.save
+    customer.default_card = card.id
+    customer.save
+  rescue Stripe::InvalidRequestError => e
+    logger.error "Stripe Error While Updating Card Info: #{e.message}"
+    errors.add :base, "#{e.message}"
+    false
+  end
+
+  def get_role_name(org_id)
+    member =  self.member.where(:organization_id => org_id).first
+    if member.present?
+      return member.roles.first.name
+    end
+  end
+  def get_role(org_id)
+    member =  self.member.where(:organization_id => org_id).first
+    if member.present?
+      return member.roles.first
+    end
+  end
+    
+  def can_do_this(org,setting)
+    role = get_role_name(org.id)
+    if (role == 'admin' || role == 'manager' || role == 'super_admin' || setting)
+      return true
+    else
+      return false 
+    end
+  end
+
+  def self.find_for_twitter_oauth(auth, signed_in_resource=nil)
+    user = User.where(:provider => auth.provider, :uid => auth.uid).first
+    if user
+      return user
+    else
+      user = User.new
+      name = auth.info.name
+      if name.include?(" ")
+        user.first_name = name.split(" ").first
+        user.last_name = name.split(" ").second
+      else
+        user.first_name = name
+        user.last_name = " "
+      end
+      user.provider = auth.provider
+      user.uid = auth.uid
+      user.password = Devise.friendly_token[0,20]
+      user.avatar = auth.info.image
+      user.save(:validate=>false)
+      return user
+    end
+  end
+
+  def self.find_for_google_oauth2(access_token, signed_in_resource=nil)
+    data = access_token.info
+    user = User.where(:provider => access_token.provider, :uid => access_token.uid ).first
+    if user
+      return user
+    else
+      registered_user = User.where(:email => data.email).first
+      if registered_user
+        return registered_user
+      else
+        user = User.new
+        user.provider = access_token.provider
+        user.uid = access_token.uid
+        user.email = data["email"]
+        user.password = Devise.friendly_token[0,20]
+        user.first_name = data.first_name
+        user.last_name = data.last_name
+        user.avatar = data.image # assuming the user model has an image
+        user.save(:validate=>false)
+        return user
+      end
+    end
+  end
+
+  def idle?(date, s_time, e_time)
+    shift = self.shifts.where( "due_date = :date AND ( ((start_time <= :s_time) AND (end_time >= :s_time)) OR ((:e_time >= start_time) AND (:e_time <= end_time)) OR ((start_time >= :s_time) AND (end_time <= :e_time)) ) ", {:date => date, :s_time => s_time, :e_time => e_time}).first
+    false if shift
+    true if shift.nil?
+  end
+
+  def notify_new_signup(org_id)
+    emails = self.organizations.find_by_id(org_id).users.with_role(:manager).collect { |user| user.email }
+    Notifier.delay.new_signup(emails,self.id)
+  end
+  def notify_alert(swap)
+    if self.email_notify
+      Notifier.delay.swap(self.email,swap.id)
+    end
+
+    if self.sms_notify
+      content = "Swap Date : #{swap.shift.due_date} Start Time : #{swap.shift.start_time} End Time : #{swap.shift.end_time}"
+      arguments = {
+        :to => self.phone_number,
+        :content => content
+      }
+      SmsWorker.perform_async(arguments)
+    end
+  end
+
+  def notify_message(lateness)
+    if self.email_notify
+      Notifier.delay.late(self.email,lateness.id)
+    end
+
+    if self.sms_notify
+      content = "Lateness Reason : #{lateness.reason} Start Time : #{lateness.shift.start_time} End Time : #{lateness.shift.end_time}"
+      arguments = {
+        :to => self.phone_number,
+        :content => content
+      }
+      SmsWorker.perform_async(arguments)
+    end
+  end
+
+  def send_instructions
+    template = 'send_instructions'
+    subject = "Reset password instructions"
+    Notifier.delay.send_instructions(self, subject, template)
+  end
+
+  def name
+    "#{first_name}"+ ' ' + "#{last_name}"
+  end
+
+  def mailboxer_email(object)
+    self.email
+  end
+
+  def self.encode user
+    JWT.encode({ user_id: user.id, exp: (DateTime.now + 30).to_i }, JWT_SECRET, JWT_ALGORITHM)
+  end
+
+  def self.for_oauth oauth
+    oauth.get_data
+    data = oauth.data
+    user = find_by_email(data[:email])
+    user = user.present? ? user : data
+  end
+  
+  def average_rating
+    avg_rating = VendorRating.where(vendor_id: self.id).average("rating")
+    return avg_rating.to_i
+  end
+  # Get Asset Vendor 
+  def get_asset_repair_vendor
+    Asset.where("vendor_id = ?", self.id)
+  end
+  
+  # Get Asset Purchase Vendor
+  def get_asset_purchase_vendor
+    Asset.where("purchase_vendor_id = ?", self.id)
+  end
+end

data/app/models/user_notification_setting.rb

@@ -0,0 +1,178 @@
+class UserNotificationSetting < ActiveRecord::Base
+  belongs_to :user
+  belongs_to :organization
+  has_many :notifications, :as => :source
+  
+  # Email function for setting timer
+  def self.email_setting(user)
+    get_data = notify_setting(user)
+    if get_data.blank?
+      email_delay = "immediately"
+      #      notification_update(user)
+    else
+      if get_data.email_date =="immediately"
+        email_delay = get_data.email_date
+        #        notification_update(user)
+      else
+        email_delay = get_data.email_date
+      end
+    end
+  end
+  
+  # Setting Date Time
+  def self.setting_date(user)
+    get_data = notify_setting(user)
+    if get_data.updated_at.blank?
+      st_date =  DateTime.now.to_date
+    else
+      st_date = get_data.updated_at.to_date
+    end
+  end
+  
+  # Repair cost for x setting
+  def self.repair_cost(user)
+    get_data = notify_setting(user)
+    if get_data.blank?
+      repaire_cost = 5
+    else
+      repaire_cost = get_data.cost
+    end
+  end
+  
+  # repair count for same asset
+  def self.repair_count(user)
+    get_data = notify_setting(user)
+    if get_data.blank?
+      repaire_count = 5
+    else
+      repaire_count = get_data.repair_count
+    end
+  end
+  
+  # Waranty Email Notification 
+  def self.waranty_date(user)
+    get_data = notify_setting(user)
+    if get_data.blank?
+      warant_date = "warnty_day"
+    else
+      warant_date = get_data.waranty_date
+    end
+  end
+  
+  
+  # User Setting for specific user
+  def self.user_notification(user_id)
+    self.find_by_user_id(user_id)
+  end
+  
+  # Convert Time
+  
+  def self.service_date(setting)
+    if setting.contract_date.blank?
+      time = 1.day
+    elsif  setting.contract_date =="day"
+      time = 1.day
+    elsif  setting.contract_date =="week"
+      time = 1.week
+    elsif  setting.contract_date =="month"
+      time = 1.month
+    elsif  setting.contract_date =="2_month"
+      time = 2.months
+    elsif  setting.contract_date =="3_month"
+      time = 3.months
+    end
+    return time 
+  end
+  
+  # Email for Assets
+  def self.create_asset_email(asset, organization , sender_email, user)
+    UserMailer.create_asset(asset, organization ,sender_email , user , user).deliver
+  end
+  
+  def self.update_asset_email(asset,organization, sender_email, user)
+    seen = true
+    type = "asset update"
+    UserMailer.update_asset(asset,organization ,sender_email,user,user).deliver
+    HistoryEmail.history_email(user.id ,type , seen , organization.id , asset.id)
+  end 
+  
+  def self.assigment(asset, organization, employee, sender_email, user , assignment)
+    seen = true
+    type = "asset assignment"
+    UserMailer.asset_assgiment(asset,organization, employee, sender_email, user , assignment , user).deliver
+    HistoryEmail.history_email(user.id ,type,seen,organization.id , assignment.id)
+  end
+  # End of Asset
+  # For Vendor Creation
+  def self.create_vendor(vendor, sender_email, organization, user )
+    seen = true
+    type = "vendor create"
+    UserMailer.ct_vendor(vendor,sender_email, organization, user , user ).deliver
+    HistoryEmail.history_email(user.id ,type,seen,organization.id,vendor.id)
+  end
+  
+  #  For ticket
+  def self.create_ticket(asset , organization , sender_email, user , ticket )
+    seen = true
+    type = "create ticket"
+    UserMailer.create_ticket(asset , organization , sender_email, user , ticket ,user).deliver
+    HistoryEmail.history_email(user.id ,type,seen,organization.id , ticket.id)
+  end
+  
+  # Ticket collect 
+  def self.colletion_tickect(asset , organization , sender_email, user , collection)
+    seen = true
+    type = "collect ticket"
+    UserMailer.collect_ticket(asset ,  organization , sender_email, user , collection , user).deliver
+    HistoryEmail.history_email(user.id ,type, seen,organization.id, collection.collected_by_id)
+  end
+  # end
+
+  def self.send_repair(asset , organization ,  sender_email, user , request)
+    seen = true
+    type="repair request"
+    UserMailer.repair_request(asset ,  organization ,  sender_email, user , request , user).deliver
+    HistoryEmail.history_email(user.id ,type , seen,organization.id , request.id)
+  end
+  
+  # Count Checking of repair asset
+  def self.repair_count_email(asset , vendor, organization, user , ticket)
+    total_repair = AssetRepairRequest.get_total_repair(asset.id)
+    member = Member.where("organization_id = ? " , organization.id).first
+    if member.has_role? 'owner'
+      onwer_user = User.where("id = ?",member.user_id).first
+      repair_count = UserNotificationSetting.repair_count(onwer_user.id)
+      if total_repair >= repair_count.to_i 
+        UserMailer.repair_count_exceding(asset , vendor,  organization , user , onwer_user, total_repair,repair_count , ticket).deliver
+      end
+    end
+  end
+  
+  # Checking the Main Notification for cost alert 
+  def self.collect_repair_alert(collection , organization , current_user)
+    purchasecost = collection.asset.purchased_cost
+    member = Member.where("organization_id = ? " , organization.id).first
+    if member.has_role? 'owner'
+      @onwer_user = User.where("id = ?",member.user_id).first
+      repair_cost = self.repair_cost(@onwer_user)
+      @estimate_cost = (purchasecost/100)*repair_cost.to_f
+      if @estimate_cost.to_f < collection.vending_cost.to_f || collection.estimate_by_user.to_f >= collection.vending_cost.to_f
+        if @onwer_user
+          UserMailer.repair_cost_alert(collection ,  organization , current_user , @onwer_user , @estimate_cost).deliver
+        end
+      end
+    end
+  end
+
+  private
+  # Def notification find
+  def self.notify_setting(user)
+    UserNotificationSetting.find_by_user_id(user)
+  end
+  
+  def self.notification_update(user)
+    user_notify = Notification.find_by_notified_by_id(user)
+    user_notify.update_attribute(:notifications_status, true)
+  end
+ 
+end

data/app/models/user_vacation_info.rb

@@ -0,0 +1,22 @@
+# == Schema Information
+#
+# Table name: user_vacation_infos
+#
+#  id          :integer          not null, primary key
+#  user_id     :integer
+#  vacation_id :integer
+#  spent_hours :float
+#  created_at  :datetime
+#  updated_at  :datetime
+#
+
+class UserVacationInfo < ActiveRecord::Base
+
+  belongs_to :member
+  belongs_to :vacation
+
+  validates_presence_of :member
+  validates_presence_of :vacation
+
+  validates_uniqueness_of :member_id, :scope => :vacation_id
+end

data/app/models/vacation.rb

@@ -0,0 +1,24 @@
+# == Schema Information
+#
+# Table name: vacations
+#
+#  id              :integer          not null, primary key
+#  name            :string(255)
+#  total_hours     :integer
+#  organization_id :integer
+#  created_at      :datetime
+#  updated_at      :datetime
+#
+
+class Vacation < ActiveRecord::Base
+
+  belongs_to :organization
+
+  has_many :users
+  has_many :user_vacation_infos
+  has_many :members, :through => :user_vacation_infos
+
+  validates_presence_of :name
+  validates_presence_of :total_hours
+
+end