Class: Reaper::Main

Inherits:

Thor

• Object
• Thor
• Reaper::Main

Defined in:

lib/reaper.rb

Instance Method Summary

• #account ⇒ Object
• #delete(date_str) ⇒ Object
• #donate ⇒ Object
• #login ⇒ Object
• #me ⇒ Object
• #projects ⇒ Object
• #show(date_str) ⇒ Object
• #submit(date_str) ⇒ Object
• #test ⇒ Object
• #version ⇒ Object

Instance Method Details

#account ⇒ Object

# File 'lib/reaper.rb', line 366

def account
  abort 'Harvest login info not found' unless File.exist? LOGIN_FILE_PATH
  
  login_data = YAML.load File.read(LOGIN_FILE_PATH)
  if login_data[:harvest_login]
    login = login_data[:harvest_login]
    token = login[:token]
    account_id = login[:account_id]
    user_id = login[:user_id]
  
    if token && !token.empty? && 
      account_id && account_id.is_a?(Integer) && 
      user_id && user_id.is_a?(Integer)
      puts """Your current Harvest login info:
- Harvest token: #{token}
- Harvest account ID: #{account_id}
- Harvest user ID: #{user_id}"""
    end
  end
end

#delete(date_str) ⇒ Object

# File 'lib/reaper.rb', line 462

def delete(date_str)
  mon, fri = get_week_range_from_date_str date_str

  entries, _ = show date_str
  
  if !entries.empty?
    puts "Delete #{entries.size} entrires from #{mon} to #{fri}? (Y/n)"
    confirm = $stdin.gets.chomp
    abort 'Deletion cancelled' unless confirm == 'Y'
  end
  
  progressbar = ProgressBar.create(
    :title => 'Deleting', 
    :total => entries.size,
    :format => '%t %c/%C %B'
  )
  
  entries.each do |e|
    rsp = delete_entry e['id']
    if rsp 
      progressbar.increment
    else
      abort "Deleting request failed. Your deleting actions may not be completed. Please run `reaper show #{date_str}` to check."
    end
  end

  puts 'Deletion completed!'
end

#donate ⇒ Object

# File 'lib/reaper.rb', line 865

def donate
  puts 'If you think Reaper saves your time, please consider buying the author a milktea ❤️'
  pay_img = File.join Reaper.root, 'assets/alipay.jpg'
  `qlmanage -p #{pay_img} 2>/dev/null`
end

#login ⇒ Object

# File 'lib/reaper.rb', line 347

def login
  Reaper.openWebpage "https://id.getharvest.com/oauth2/authorize?client_id=#{HARVEST_CLIENT_ID}&response_type=token"
  
  start_login_server
  
  if $token
    me = Reaper.request 'users/me'
    $user_id = me['id']
    puts "Harvest user ID: #{$user_id}"
  
    login_data = { :harvest_login => 
      { :token => $token, :account_id => $acc_id, :user_id => $user_id } 
    }
    
    File.write(LOGIN_FILE_PATH, login_data.to_yaml)
  end
end

#me ⇒ Object

# File 'lib/reaper.rb', line 388

def me
  Reaper.check_auth

  puts "Fetching your Harvest profile..."
  rsp = Reaper.request 'users/me'

  if rsp
    rsp.each { |k, v| rsp[k] = v.scan(/.{1,30}/).join("\n") if (v && v.to_s.size > 30) }

    title = 'Harvest Profile'
    rows = rsp.to_a
    table = Terminal::Table.new :title => title, :rows => rows
    puts table
  end
end

#projects ⇒ Object

# File 'lib/reaper.rb', line 408

def projects
  Reaper.list_projects
end

#show(date_str) ⇒ Object

# File 'lib/reaper.rb', line 413

def show(date_str)
  mon, fri = get_week_range_from_date_str date_str
  
  Reaper.check_auth

  # both ruby and Harvest use ISO 8601 by default
  from = mon.to_s
  to = fri.to_s
  
  puts "Fetching your time entries from #{from} to #{to}"
  raw_entries = Reaper.request("time_entries?user_id=#{$user_id}&from=#{from}&to=#{to}")['time_entries']
  
  if raw_entries.empty?
    puts "Cannot find any time entries in the week of #{from}" 
    return []
  end
  
  entries = {}
  raw_entries.each do |e|
    date = Date.parse(e['spent_date'])
    tasks = entries[date] || []
    entries[date] = tasks if tasks.empty?
    tasks << {
      :id => e['id'],
      :created_at => DateTime.parse(e['created_at']),
      :project => e['project']['name'],
      :project_code => e['project']['code'],
      :task => e['task']['name'],
      :client => e['client']['name'],
      :hours => e['hours']
    }
  end
  
  # sort the entries hash to make sure:
  # - keys (spent date) are sorted by date
  # - values (daily entries) are sorted by entry created date
  
  entries = entries.sort.to_h
  
  entries.each do |k, v|
    v.sort_by! { |t| t[:created_at] }
  end
  
  print_time_entries mon, fri, entries

  return raw_entries, entries
end

#submit(date_str) ⇒ Object

# File 'lib/reaper.rb', line 493

def submit(date_str)
  submit_time_entries date_str, options[:excluded]
end

#test ⇒ Object

# File 'lib/reaper.rb', line 335

def test
  a = {"no"=>"0", "po"=>"0", "p0"=>"20769370", "t0"=>"11683694", "pct0"=>"56", "p1"=>"20500923", "t1"=>"11728114", "pct1"=>"22", "p2"=>"20500918", "t2"=>"11728114", "pct2"=>"22"}
  puts a
  
  max_index = a.keys.select { |e| e =~ /p\d+/ }
    .map { |e| e[1..-1] }
    .max
  
  puts max_index
end

#version ⇒ Object

# File 'lib/reaper.rb', line 872

def version
sickle = "MMMMMMMMMMMMMMMMMMMMMMMMMMMNdyo//oymMMMMMMMMMMMMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMmo` `/yNMMMMMMMMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMs`   :hMMMMMMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMN- `: .yMMMMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM: :d: -dMMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMN. sMh` oMMM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMy .MMN- /MM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM  dMMN- oM
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM` hMMMm` d
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM  dMMMM+ :
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMy `MMMMMd `
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMN. sMMMMMm  
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMN- /MMMMMMy .
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMh. oMMMMMMM. s
MMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMNy- -dMMMMMMM+ -M
MMMMMMMMMMMMd/ -/oydmNMMMMMMNmho: `/dMMMMMMMM+ .NM
MMMMMMMMMMd:  syo/-.   ``.`  `./odMMMMMMMMMm: :NMM
MMMMMMMMd: .+` /dMMMMNmmddmmNMMMMMMMMMMMMNo``sMMMM
MMMMMMd: -yMMNy` .odMMMMMMMMMMMMMMMMMMMmo` +NMMMMM
MMMMd:   -----. -o- ./ymMMMMMMMMMMMNdo- .oNMMMMMMM
MMd: -yhhhh+  -hMMMNy+-  .:/+oo+/:. `:odMMMMMMMMMM
m:  `:::::- -hMMMMMMMMMMmhsoo++ooshmMMMMMMMMMMMMMM
` +ssss:  -hMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM
. +mNy- -hMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM
N+.  `/hMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMM"

  slogan = 'Keep your PM away (TM)'
  slogan_len = slogan.length + 4
  puts "#{sickle}\n\n#{'*' * slogan_len}\n* #{slogan} *\n#{'*' * slogan_len}\n\nReaper: A smart Harvest filling helper. \nVersion #{Reaper::VERSION}"
end