pshq_engine 0.0.1

data/app/models/organization.rb

@@ -0,0 +1,189 @@
+class Organization < ActiveRecord::Base
+  resourcify
+  
+  #  acts_as_paranoid
+  self.per_page = 24
+  require 'will_paginate/array'
+  has_many :employees
+  has_many :locations
+  has_many :assets
+  has_many :vendors
+  has_one :subdomain
+  has_one :plan
+  has_many :importfiles
+  has_many :asset_scans
+  has_many :asset_assignments
+  has_many :notifications, :as => :source
+  has_one :setting
+  has_many :user_notification_settings
+  has_many :depreciation_settings
+  has_many :cancellations, :through => :users
+  has_many :feedbacks, :through => :users
+  has_many :jobs, :dependent => :destroy
+  has_many :certifications, :dependent => :destroy
+  has_many :alerts, :dependent => :destroy
+  has_many :messages, :dependent => :destroy
+  has_many :shifts, :through => :users
+  has_many :shift_trades
+  has_many :shift_offs
+  has_many :swaps, :through => :users
+  has_many :tasks, :dependent => :destroy
+  has_many :timeoffs, :through => :users
+  has_many :schedules, :dependent => :destroy
+  has_many :notes
+  has_many :vacations, :dependent => :destroy
+  has_many :candidates
+  has_many :templates
+
+  has_many :member
+  accepts_nested_attributes_for :member
+  has_many :users, :through => :member
+  
+  validates  :name , :city , :state, :address ,:postal_code, :presence => true 
+  #  validates  :name, :uniqueness  => true
+
+  
+  has_many :project_payments
+  has_many :projects, :through =>:project_payments
+  
+  has_attached_file :logo, :styles => { :medium => "300x300!", :thumb => "100x100!" }, :default_url => "assets/users/org.png"
+  validates_attachment_content_type :logo, :content_type => /\Aimage\/.*\Z/
+  
+  # Functions for TMA
+  def managers(exclude = nil)
+    all_managers = Array.new
+    
+    
+    self.users.each do |user|
+      user.member.each do|mem|
+        if mem.has_role? :manager, self or 
+            mem.has_role? :owner, self then
+          if exclude then
+            all_managers.push user if user.id != exclude.id
+          else
+            all_managers.push user
+          end
+        end
+      end
+    end
+   
+    return all_managers
+  end
+  
+  
+  def push_notification_user(type , source , summary, notifier, org_users , organization )
+    Notification.create_and_send_notification(
+      :type => type,
+      :source => source,
+      :summary => summary,
+      :notifier => notifier,
+      :managers => org_users,
+      :organizationID => organization.id
+    )
+  end
+  
+  # Functions for MT2W
+  def get_idles(date, start_time, end_time)
+    idles = []
+    users.each do |user|
+      idles << user if user.idle?(date, start_time, end_time)
+    end
+
+    idles
+  end
+  # @return [manageable users list]
+  # @param [Object] manager
+  def manageable_users(manager)
+    result = []
+    users = []
+    unless self.show_disabled_users
+      self.users.all.each do |user|
+        if user.member.find_by_organization_id(self.id).disabled.nil? or user.member.find_by_organization_id(self.id).disabled === false
+          users << user
+        end
+      end
+    else
+      users = self.users
+    end
+    
+    unless manager.get_role_name(self.id) == 'employee'
+      if self.show_owner 
+        users.each{|user| result << user}
+      else
+        users.each{|user| result << user if user.get_role_name(self.id) != 'owner'}
+      end
+    else
+      users.each{|user| result << user if user.get_role_name(self.id) == 'employee'}
+    end
+    result
+  end
+
+  def self.mt2w_access
+    where(:myt2w_access => true)
+  end
+  
+  def schedule_users(manager, org_id)
+    result = []
+    users = self.users.enable
+    if !self.show_owner
+      users.each{|user| result << user.id if user.get_role_name(org_id) != 'owner'}  
+    else
+      users.each{|user| result << user.id}
+    end
+    User.where(id: result)
+  end
+
+  def complete_address
+    "#{address}, #{city}, #{state}, #{country}"
+  end
+  def get_managers
+    self.users.with_role(:manager)
+  end
+   
+  def get_employees
+    employees = []
+    employee_roles = [:employee]
+    employee_roles.each do |role|
+      self.users.with_role(role).each{ |user| employees << user }
+    end
+    employees
+  end
+  
+  #Date change 
+  def self.date_format(date)
+    change_date = date.strftime('%m/%d/%Y')
+    return change_date
+  end
+  
+  def get_vendor_user
+    vendors = []
+    self.member.each do |x|
+      if (x.roles[0].name == 'vendor')
+        vendors << x.user
+      end
+    end
+    vendors.sort!
+    
+  end
+  
+  def vendor_search
+    array = []
+    user = @organization.users
+    user.each do |u|
+      if u.get_role_name(@organization.id) == 'vendor'
+        array << u
+      end
+    end
+    @arrayobj = array.collect { |u| u.id }
+    return @arrayobj
+  end
+  
+  def get_role(org_id)
+    member =  self.member.where(:organization_id => org_id).first
+    if member.present?
+      role = member.roles.first
+      return member.roles.first
+    end
+  end    
+  
+end

data/app/models/plan.rb

@@ -0,0 +1,4 @@
+class Plan < ActiveRecord::Base
+  has_many :project_payments
+  default_scope { order(:display_order) }
+end

data/app/models/project.rb

@@ -0,0 +1,5 @@
+class Project < ActiveRecord::Base
+	has_many :project_payments
+  has_many :member_devices
+	has_many :organizations, :through =>:project_payments
+end

data/app/models/project_payment.rb

@@ -0,0 +1,4 @@
+class ProjectPayment < ActiveRecord::Base
+	belongs_to :organizations
+	belongs_to :plan
+end

data/app/models/recurring_time_off.rb

@@ -0,0 +1,4 @@
+class RecurringTimeOff < ActiveRecord::Base
+  belongs_to :timeoff
+  belongs_to :user
+end

data/app/models/response.rb

@@ -0,0 +1,23 @@
+class Response
+  
+  def self.failed_user_authentication_response
+    {
+      :json => {
+        :success=>false, 
+        :message => "Please Enter Correct Email/Password"
+      }, 
+      :status => 400
+    }
+  end
+  
+  def self.user_authentication_disabled_organization_response
+    {
+      :json =>{
+        :success=>false, 
+        :message => "Please renew your subscription or Contact your supplier"
+      }, 
+      :status => 401
+    }
+  end
+  
+end

data/app/models/role.rb

@@ -0,0 +1,22 @@
+class Role < ActiveRecord::Base
+  has_and_belongs_to_many :users, :join_table => :users_roles
+  has_and_belongs_to_many :members
+  belongs_to :resource, :polymorphic => true
+
+  validates :resource_type,
+            :inclusion => { :in => Rolify.resource_types },
+            :allow_nil => true
+
+  scopify
+  
+  #function for MT2W
+  def self.get_invitable_roles
+    invite_roles = YAML.load(ENV['ROLES'])
+    invite_roles.delete('super_admin')
+    invite_roles.delete('owner')
+    invite_roles.delete('general_manager')
+
+    self.where(:name => invite_roles)
+  end
+  
+end

data/app/models/schedule.rb

@@ -0,0 +1,228 @@
+# == Schema Information
+#
+# Table name: schedules
+#
+#  id         :integer          not null, primary key
+#  publish    :boolean
+#  user_id    :integer
+#  job_id     :integer
+#  created_at :datetime
+#  updated_at :datetime
+#  start_date :date
+#  end_date   :date
+#
+
+class Schedule < ActiveRecord::Base
+
+  
+  belongs_to :user
+  belongs_to :organization
+  
+  has_many :shifts, :dependent => :destroy
+  has_many :cancellations, :through => :shifts
+  has_many :swaps, :through => :shifts
+  has_many :templates
+  
+  validates_presence_of :user
+  validates_presence_of :organization
+  validates_uniqueness_of :user_id, scope: [:start_date, :end_date, :template_id, :organization_id]
+  validates_date :start_date
+  validates_date :end_date, :after => :start_date
+  validate :double_schedule
+
+  def published?
+    self.publish == true
+  end
+  
+  def self.publish
+    where(:publish => true)
+  end
+
+  def self.publish! schedule_ids, org_id
+    schedules = where(:id => schedule_ids)
+    schedules.each do |schedule|
+      schedule.notify_employee(org_id)
+    end
+    return schedules.update_all(:publish => true)
+  end
+
+  def notify_employee org_id
+    Notifier.delay.schedule_published(self, org_id)
+  end
+
+  def self.schedule_resend(schedules)
+    where(:id => schedules).each do |schedule|
+      schedule.notify_employee
+    end
+  end
+
+  def get_sum
+    total_hours.try(:round, 1)
+  end
+
+  def calculate_total_hours
+    sum = 0.0
+    prop = self.user.member.find_by_organization_id(self.organization_id).in_train ? 'training_hours' : 'hours'
+    shifts.each do |shift|
+        shift[prop] = "2000-01-01 00:00:00 UTC" if shift[prop].nil?
+        sum = sum + shift[prop].hour
+        sum = sum + ( shift[prop].min + 0.00 ) / 60
+    end
+    self.update_attributes(:total_hours => sum)
+  end
+
+  def check_timeoff_conflict start_time, end_time, due_date
+    self.shifts.where(:due_date => due_date).each do |shift|
+      if (start_time.strftime("%H:%M") >= shift.start_time.strftime("%H:%M") and start_time.strftime("%H:%M") < shift.end_time.strftime("%H:%M")) or (end_time.strftime("%H:%M") >= shift.start_time.strftime("%H:%M") and end_time.strftime("%H:%M") < shift.end_time.strftime("%H:%M"))
+        return true
+      end 
+    end
+    return false
+  end
+
+  private
+  def double_schedule
+    # schedule = self.class.where(:start_date => self.start_date, :end_date => self.end_date,:template_id => nil).first
+    # if self.id
+    #   # existing row
+    #   if schedule && schedule.user.id == self.user_id && schedule.id != self.id
+    #     errors.add(:start_date, "already scheduled from #{self.start_date} to #{self.end_date}")
+    #   end
+    # else
+    #   # new row
+    #   if schedule && schedule.user.id == self.user_id
+    #     errors.add(:start_date, "already scheduled from #{self.start_date} to #{self.end_date}")
+    #   end
+    # end
+  end
+  
+  def self.create_with_json(data, date=nil,  m_to_s, org_id)
+    if date
+      now_date = DateTime.parse(date)
+    else
+      now_date = Time.zone.now
+    end
+
+    wday = now_date.wday
+    if m_to_s
+      start_day = now_date - wday.day + 1.day
+      end_day = now_date - wday.day + 7.day
+    else
+      start_day = now_date - wday.day + 0.day
+      end_day = now_date - wday.day + 6.day
+    end
+    data.each do |schedule|
+      if schedule["id"].nil?
+        # create new one
+        new_schedule = self.new(:user_id => schedule["employee_id"],
+          :publish => schedule["publish"],
+          :start_date => start_day,
+          :end_date => end_day,
+          :total_hours =>   schedule["total_hours"],
+          :organization_id => org_id
+        )
+        unless new_schedule.save
+          return new_schedule.errors
+        else
+          if new_schedule.publish?
+            new_schedule.notify_employee
+          end
+        end
+
+        result = Shift.create_with_json(new_schedule, schedule["shifts"],false)
+        unless result == true
+          return result
+        end
+      else
+        # modify existing one
+        ext_schedule = self.find(schedule["id"])
+        if ext_schedule
+#          ext_schedule.job_id = schedule["job_id"]
+          ext_schedule.user_id = schedule["employee_id"]
+          ext_schedule.publish = schedule["publish"]
+          ext_schedule.start_date = start_day
+          ext_schedule.end_date = end_day
+          ext_schedule.total_hours = schedule["total_hours"]
+          changed = ext_schedule.changed?
+          result = ext_schedule.save
+          unless result
+            return ext_schedule.errors
+          end
+          if ext_schedule.publish?
+            ext_schedule.notify_employee
+          end
+        end
+        result = Shift.create_with_json(ext_schedule, schedule["shifts"],changed) if ext_schedule
+        unless result == true
+          return result
+        end
+      end
+    end
+
+    true
+  end
+
+  def self.current_week
+    cur_day = Time.zone.now.strftime('%Y-%m-%d')
+    where("start_date <= '#{cur_day}' and end_date >= '#{cur_day}'")
+  end
+
+  def self.week_schedule_by_date date
+    where("start_date <= '#{date}' and end_date >= '#{date}'")
+  end
+  
+  def self.not_template
+    where(:template_id => nil)
+  end
+
+  def self.published_only
+    where(:publish => true)
+  end
+  
+  def self.schedule_by_date(start_date,end_date )
+    where("start_date >= '#{start_date}' and end_date <= '#{end_date}'")
+  end
+
+  def self.published
+    where(:publish => true)
+  end
+
+  # def notify_employee
+  #   Notifier.delay.schedule_published(self)
+  # end
+
+  def self.editable users, org, date, role
+    weeks_schedules = []
+    users = users.includes(:member)
+    if role === 'employee'
+      schedules = where(:organization_id => org.id, :user_id => users, :publish => true).not_template.week_schedule_by_date(date).group_by(&:user_id)
+    else
+      schedules = where(:organization_id => org.id, :user_id => users).not_template.week_schedule_by_date(date).group_by(&:user_id)
+    end
+    users.each_with_index do |user, i|
+      sch = {}
+      member    = user.member.find_by_organization_id(org.id)
+      schedule  = schedules[user.id].try(:first)
+      sch["employee_id"]      = user.id
+      sch["member_id"]        = member.id
+      sch["employee_name"]    = user.name
+      sch["role_name"]        = member.roles.first.name
+      sch["first_name"]       = user.first_name
+      sch["last_name"]        = user.last_name
+      sch["priority"]         = member.priority
+      sch["trainee"]          = member.in_train
+      sch["request_hours"]    = member.request_weekly_hours || "0"
+      sch["id"]               = schedule.try(:id)
+      sch["start_date"]       = schedule.try(:start_date) || Shift.get_day(org.m_to_s, 0, date)
+      sch["end_date"]         = schedule.try(:end_date) || Shift.get_day(org.m_to_s, 6, date)
+      sch["publish"]          = schedule.try(:publish)
+      sch["schdeule_sum"]     = schedule.try(:get_sum) || "0"
+      sch["organization_id"]  = org.id
+      sch["shifts"]           = Shift.week_shifts schedule, user, org, date
+      weeks_schedules << sch
+    end
+    weeks_schedules = weeks_schedules.sort_by {|sch| sch["priority"]}
+    return weeks_schedules
+  end
+
+end

data/app/models/setting.rb

@@ -0,0 +1,25 @@
+class Setting < ActiveRecord::Base
+  belongs_to :organization
+  has_many :notifications, :as => :source
+  
+  # check user update settings
+  def self.check_user(user)
+    already_exist = UserNotificationSetting.where("user_id=?",user)
+    if !already_exist.blank?
+      user_exitence = "exist"
+    else
+      user_exitence = "not_exist"
+    end
+    return user_exitence
+  end
+  
+  #Notification setting alert
+  def self.notification_alert(user, organization)
+    @notification = Notification.create(
+      :summary => "Notification setting has been changed",
+      :notification_type => "NSC",
+      :notified_by => user,
+    )
+    @notification.add_notifiers([user])
+  end
+end