class ShiftNote

cookie() click to toggle source

This is basically an API token. @return [String] the cookie of the authenticated user

# File lib/shiftnote.rb, line 67
def cookie
  shiftnote = RestClient.get('https://ww1.shiftnote.com/BulletinBoard/', Cookie: @cookie)

  doc = Nokogiri::HTML.parse(shiftnote.body)

  begin
    doc.search('div#MyScheduleDiv').at('script').text
  rescue NoMethodError
    generate_cookie(@credentials[:username], @credentials[:password])
  end
  @cookie
end

date_of_next(day) click to toggle source

Method to find the next instance of the given date. @!visibility private @param day [String] what day to find

# File lib/shiftnote.rb, line 59
def date_of_next(day)
  date  = Date.parse(day)
  delta = date > Date.today ? 0 : 7
  date + delta
end

employee(id) click to toggle source

Get an employee (by id) @param id [Integer] the id of the employee

# File lib/shiftnote.rb, line 101
def employee(id)
  Employee.new(id: id)
end

generate_cookie(username, password) click to toggle source

Generate a cookie, based on login credentials. @param username [String] the username of the user. @param password [String] the password of the user. @return [String] the cookie!

# File lib/shiftnote.rb, line 23
def generate_cookie(username, password)
  uri = URI.parse('https://ww1.shiftnote.com/login')
  request = Net::HTTP::Post.new(uri)
  request.content_type = 'application/x-www-form-urlencoded'
  request.set_form_data(
    'FullSite' => 'False',
    'ReturnUrl' => '',
    'username' => username,
    'Password' => password,
    'Submit' => 'Submit',
    'remember' => 'true'
  )

  req_options = {
    use_ssl: uri.scheme == 'https'
  }

  response = Net::HTTP.start(uri.hostname, uri.port, req_options) do |http|
    http.request(request)
  end
  # response.code
  raise Errors::InvalidCredentials, 'The provided credentials are invalid, please try again.' if response.body.include?("There was no match for the username and password provided, please try again!")

  @cookie = response.header['Set-Cookie']
end

get_next_week_shifts(day = "Monday") click to toggle source

@param day [String] what day the week starts on. @return [ScheduleThisWeek] the schedule for next week.

# File lib/shiftnote.rb, line 107
def get_next_week_shifts(day = "Monday")
  nextmonday = date_of_next(day)

  week2 = RestClient.get("https://ww1.shiftnote.com/Schedules/ScheduleMine/?startDate=#{nextmonday.month}%2F#{nextmonday.day}%2F#{nextmonday.year}&noContainer=true&_=#{Time.now.to_i * 1000}", Cookie: cookie)

  week2doc = Nokogiri::HTML.parse(week2.body)
  dataweek2 = JSON.parse(scrapejunk(week2doc.at('script').text))
  User.new(dataweek2).schedule_this_week
end

scrapejunk(string) click to toggle source

Useful method (for the gem that is) that gets rid of stuff we don't need. @!visibility private @param string [String] a string to scrape the junk off

# File lib/shiftnote.rb, line 52
def scrapejunk(string)
  string.gsub('<script>', '').delete(';').gsub('</script>', '').gsub('window.scheduleMinebindings = ShiftNote.Bind(window.scheduleMinemodel)', '').gsub('window.scheduleMinemodel = ', '')
end

send_message(recipients, subject, body, urgent) click to toggle source

Send a message to a user (or group of them) @param recipients [Array<Integer>] who will receive this message

# File lib/shiftnote.rb, line 119
def send_message(recipients, subject, body, urgent)
  data = {
    "ToUserIDs" => recipients,
    "Subject" => subject,
    "Body" => body,
    "Priority" => urgent
  }
  RestClient.post('https://ww1.shiftnote.com/Message/SendMessages', URI.encode_www_form(data), Cookie: @cookie)
end

user() click to toggle source

Get the information of the currently logged in user. @return [User] the user

# File lib/shiftnote.rb, line 82
def user
  shiftnote = RestClient.get('https://ww1.shiftnote.com/BulletinBoard/', Cookie: @cookie)

  doc = Nokogiri::HTML.parse(shiftnote.body)

  begin
    data = doc.search('div#MyScheduleDiv').at('script').text
  rescue NoMethodError
    generate_cookie(@credentials[:username], @credentials[:password])
    retry
  end

  data = data.gsub('<script>', '').delete(';').gsub('</script>', '').gsub('window.scheduleMinebindings = ShiftNote.Bind(window.scheduleMinemodel)', '').gsub('window.scheduleMinemodel = ', '')

  User.new(JSON.parse(data))
end

Public Class Methods

Public Instance Methods