#!/usr/bin/env ruby -w
# encoding: UTF-8
#
# = FitFile_spec.rb -- Fit4Ruby - FIT file processing library for Ruby
#
# Copyright (c) 2014, 2015 by Chris Schlaeger <cs@taskjuggler.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of version 2 of the GNU General Public License as
# published by the Free Software Foundation.
#

$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))

require 'fit4ruby'

ENV['TZ'] = 'UTC'

describe Fit4Ruby do
  let(:fit_file) { 'test.fit' }
  let(:timestamp) { Time.now }
  let(:activity) do
    ts = timestamp
    a = Fit4Ruby::Activity.new
    a.total_timer_time = 30 * 60.0
    a.new_user_data({ :age => 33, :height => 1.78, :weight => 73.0,
                      :gender => 'male', :activity_class => 4.0,
                      :max_hr => 178 })

    a.new_event({ :event => 'timer', :event_type => 'start_time' })
    a.new_device_info({ :timestamp => ts,
                        :device_index => 0, :manufacturer => 'garmin',
                        :garmin_product => 'fenix3',
                        :serial_number => 123456789 })
    a.new_device_info({ :timestamp => ts,
                        :device_index => 1, :manufacturer => 'garmin',
                        :garmin_product => 'fenix3_gps'})
    a.new_device_info({ :timestamp => ts,
                        :device_index => 2, :manufacturer => 'garmin',
                        :garmin_product => 'hrm_run',
                        :battery_status => 'ok' })
    a.new_device_info({ :timestamp => ts,
                        :device_index => 3, :manufacturer => 'garmin',
                        :garmin_product => 'sdm4',
                        :battery_status => 'ok' })
    a.new_data_sources({ :timestamp => ts, :distance => 1,
                         :speed => 1, :cadence => 3, :elevation => 1,
                         :heart_rate => 2 })
    laps = 0
    0.upto(a.total_timer_time / 60) do |mins|
      ts += 60
      a.new_record({
        :timestamp => ts,
        :position_lat => 51.5512 - mins * 0.0008,
        :position_long => 11.647 + mins * 0.002,
        :distance => 200.0 * mins,
        :altitude => (100 + mins * 0.5).to_i,
        :speed => 3.1,
        :vertical_oscillation => 9 + mins * 0.02,
        :stance_time => 235.0 * mins * 0.01,
        :stance_time_percent => 32.0,
        :heart_rate => 140 + mins,
        :cadence => 75,
        :activity_type => 'running',
        :fractional_cadence => (mins % 2) / 2.0
      })

      if mins > 0 && mins % 5 == 0
        a.new_lap({ :timestamp => ts, :sport => 'running',
                    :message_index => laps, :total_cycles => 195 })
        laps += 1
      end
    end
    a.new_session({ :timestamp => ts, :sport => 'running' })
    a.new_event({ :timestamp => ts, :event => 'recovery_time',
                  :event_type => 'marker',
                  :recovery_time => 2160 })
    a.new_event({ :timestamp => ts, :event => 'vo2max',
                  :event_type => 'marker', :vo2max => 52 })
    a.new_event({ :timestamp => ts, :event => 'timer',
                  :event_type => 'stop_all' })
    ts += 1
    a.new_device_info({ :timestamp => ts,
                        :device_index => 0, :manufacturer => 'garmin',
                        :garmin_product => 'fenix3',
                        :serial_number => 123456789 })
    a.new_device_info({ :timestamp => ts,
                        :device_index => 1, :manufacturer => 'garmin',
                        :garmin_product => 'fenix3_gps'})
    a.new_device_info({ :timestamp => ts,
                        :device_index => 2, :manufacturer => 'garmin',
                        :garmin_product => 'hrm_run',
                        :battery_status => 'low' })
    a.new_device_info({ :timestamp => ts,
                        :device_index => 3, :manufacturer => 'garmin',
                        :garmin_product => 'sdm4',
                        :battery_status => 'ok' })
    ts += 120
    a.new_event({ :timestamp => ts, :event => 'recovery_hr',
                  :event_type => 'marker', :recovery_hr => 132 })

    a.aggregate
    a
  end

  before do
    File.delete(fit_file) if File.exist?(fit_file)
    expect(File.exist?(fit_file)).to be false
  end

  after { File.delete(fit_file) }

  it 'should write an Activity FIT file and read it back' do
    Fit4Ruby.write(fit_file, activity)
    expect(File.exist?(fit_file)).to be true

    b = Fit4Ruby.read(fit_file)
    expect(b.laps.count).to eq 6
    expect(b.lengths.count).to eq 0
    expect(b.inspect).to eq(activity.inspect)
  end

  context 'activity with Lengths' do
    let(:activity) do
      ts = timestamp
      laps = 0
      lengths = 0
      a = Fit4Ruby::Activity.new

      a.total_timer_time = 30 * 60.0
      a.new_device_info({ :timestamp => ts,
        :device_index => 0, :manufacturer => 'garmin',
        :serial_number => 123456789 })

      0.upto(a.total_timer_time / 60) do |mins|
        ts += 60
        if mins > 0 && mins % 5 == 0
          a.new_lap({ :timestamp => ts, :sport => 'swimming',
            :message_index => laps, :total_cycles => 195 })
          laps += 1

          a.new_length({ :timestamp => ts, :event => 'length',
            :message_index => lengths, :total_strokes => 45 })
          lengths += 1
        end
      end
      a
    end

    it 'should write an Activity FIT file and read it back' do
      Fit4Ruby.write(fit_file, activity)
      expect(File.exist?(fit_file)).to be true

      b = Fit4Ruby.read(fit_file)
      expect(b.laps.count).to eq 6
      expect(b.lengths.count).to eq 6
      expect(b.laps.select { |l| l.sport == 'swimming' }.count).to eq 6
      expect(b.lengths.select { |l| l.total_strokes == 45 }.count).to eq 6
      expect(b.inspect).to eq(activity.inspect)
    end

  end

end