hackatime/app/controllers/my/heartbeats_controller.rb
2025-08-07 15:36:36 -04:00

85 lines
2.9 KiB
Ruby

module My
class HeartbeatsController < ApplicationController
before_action :ensure_current_user
def export
all_data = params[:all_data] == "true"
if all_data
heartbeats = current_user.heartbeats.order(time: :asc)
if heartbeats.any?
start_date = Time.at(heartbeats.first.time).to_date
end_date = Time.at(heartbeats.last.time).to_date
else
start_date = Date.current
end_date = Date.current
end
else
start_date = params[:start_date].present? ? Date.parse(params[:start_date]) : 30.days.ago.to_date
end_date = params[:end_date].present? ? Date.parse(params[:end_date]) : Date.current
start_time = start_date.beginning_of_day.to_f
end_time = end_date.end_of_day.to_f
heartbeats = current_user.heartbeats
.where("time >= ? AND time <= ?", start_time, end_time)
.order(time: :asc)
end
export_data = {
export_info: {
exported_at: Time.current.iso8601,
date_range: {
start_date: start_date.iso8601,
end_date: end_date.iso8601
},
total_heartbeats: heartbeats.count,
total_duration_seconds: heartbeats.duration_seconds
},
heartbeats: heartbeats.map do |heartbeat|
{
id: heartbeat.id,
time: Time.at(heartbeat.time).iso8601,
entity: heartbeat.entity,
type: heartbeat.type,
category: heartbeat.category,
project: heartbeat.project,
language: heartbeat.language,
editor: heartbeat.editor,
operating_system: heartbeat.operating_system,
machine: heartbeat.machine,
branch: heartbeat.branch,
user_agent: heartbeat.user_agent,
is_write: heartbeat.is_write,
line_additions: heartbeat.line_additions,
line_deletions: heartbeat.line_deletions,
lineno: heartbeat.lineno,
lines: heartbeat.lines,
cursorpos: heartbeat.cursorpos,
dependencies: heartbeat.dependencies,
source_type: heartbeat.source_type,
created_at: heartbeat.created_at.iso8601,
updated_at: heartbeat.updated_at.iso8601
}
end
}
filename = "heartbeats_#{current_user.slack_uid}_#{start_date.strftime('%Y%m%d')}_#{end_date.strftime('%Y%m%d')}.json"
respond_to do |format|
format.json {
send_data export_data.to_json,
filename: filename,
type: "application/json",
disposition: "attachment"
}
end
end
private
def ensure_current_user
redirect_to root_path, alert: "You must be logged in to view this page!!" unless current_user
end
end
end