#include <chrono>
#include <iostream>
#include <ostream>
#include <thread>

#include "pwm/PWMControl.h"
#include <boost/json/object.hpp>
#include <fan/HwmonFan.h>

#define TIMEOUT 10

using namespace std;

HwmonFan::HwmonFan(std::shared_ptr<PWMControl> pwmControl,
                   std::shared_ptr<Sensor> rpmSensor)
    : mPWMControl(pwmControl), mRpmSensor(rpmSensor) {
  mPWMControl->EnableManualControl();
}

void HwmonFan::PWM(int percent) {
  if (percent < mMinPWM) {
    mPWMControl->Power(mMinPWM);
  } else {
    mPWMControl->Power(percent);
  }
}

int HwmonFan::RPM() { return mRpmSensor->value(); }

void HwmonFan::Label(std::string label) { mLabel = label; }

void HwmonFan::MinPWM(int value) { mMinPWM = value; }

int HwmonFan::MinPWM() { return mMinPWM; }

int HwmonFan::StartPWM() { return mStartPWM; }

void HwmonFan::StartPWM(int value) { mStartPWM = value; }

void HwmonFan::FindPWMLimits() {
  cout << "Looking for minimal PWM" << endl;
  int minPWM = 0;
  mMinPWM = 0;
  mStartPWM = 0;

  for (int curPWM = 100; curPWM > 0; curPWM -= 5) {
    PWM(curPWM);
    this_thread::sleep_for(chrono::seconds(TIMEOUT));

    int curRPM = RPM();

    if (curRPM <= 0) {
      minPWM = curPWM + 5;
      break;
    }
  }

  cout << "Setting minimal PWM: " << minPWM << endl;
  mMinPWM = minPWM;

  if (minPWM == 0) {
    cout << "Fan never stopped. ";
  } else {
    int startPWM = 0;

    cout << "Looking for start PWM!" << endl;
    for (int curPWM = minPWM - 5; curPWM < 100; curPWM += 5) {
      PWM(curPWM);
      this_thread::sleep_for(chrono::seconds(TIMEOUT));

      int curRPM = RPM();

      if (curRPM > 0) {
        cout << "Setting start PWM: " << startPWM << endl;
        startPWM = curPWM;
        break;
      }
    }

    mStartPWM = startPWM;
  }
}

json HwmonFan::toJson() const {
  json obj;
  obj = {mPWMControl->toJson(),
         mRpmSensor->toJson(),
         {"Label", mLabel},
         {"MinPWM", mMinPWM},
         {"StartPWM", mStartPWM}};
  return obj;
}

const string HwmonFan::toString() const {
  if (!mLabel.empty()) {
    return mLabel;
  } else {
    return "fan:" + mPWMControl->toString();
  }
}