/* Copyright © 2017 Andrey Klimov. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

Homepage: http://lazyhome.ru
GIT:      https://github.com/anklimov/lighthub
e-mail    anklimov@gmail.com

*/

#include "item.h"
#include "aJSON.h"
#include <DmxSimple.h>
#include "FastLED.h"
#include <ModbusMaster.h>
#include <PubSubClient2.h>

short modbusBusy=0;
extern aJsonObject * modbusitem; 


int modbusSet(int addr, uint16_t _reg, int _mask, uint16_t value);
extern PubSubClient client;
//extern const char* outprefix;
const char outprefix[] PROGMEM  = "/myhome/s_out/";

static  unsigned long lastctrl = 0;
static  aJsonObject * lastobj = NULL; 

int txt2cmd (char * payload)
{
  int cmd=0;
  
  // Check for command
   if (strcmp(payload,"ON")==0) cmd=CMD_ON;
      else if  (strcmp(payload,"OFF")==0) cmd=CMD_OFF;
           else if (strcmp(payload,"REST")==0) cmd=CMD_RESTORE;
                  else if  (strcmp(payload,"TOGGLE")==0) cmd=CMD_TOGGLE;
                       else if  (strcmp(payload,"HALT")==0) cmd=CMD_HALT;
  return cmd;
}

const short defval[4] = {0,0,0,0}; //Type,Arg,Val,Cmd

Item::Item(aJsonObject * obj)//Constructor
{ 
       itemArr= obj;
       Parse();
}
       
void Item::Parse()    
{
  if (isValid())
  {
  // Todo - avoid static enlarge for every types 
  for (int i=aJson.getArraySize(itemArr);i<4;i++) aJson.addItemToArray(itemArr,aJson.createItem(int(defval[i]))); //Enlarge item to 4 elements. VAL=int if no other definition in conf
              
              
  itemType = aJson.getArrayItem(itemArr,I_TYPE)->valueint;
  itemArg  = aJson.getArrayItem(itemArr,I_ARG);      
  itemVal  = aJson.getArrayItem(itemArr,I_VAL);  

  
        Serial.print(F(" Item:"));
        Serial.print(itemArr->name);Serial.print(F(" T:"));
        Serial.print(itemType);Serial.print(" =");Serial.println(getArg());
  }      
}

Item::Item(char * name) //Constructor
{
  if (name) 
       itemArr= aJson.getObjectItem(items, name);
  else itemArr=NULL;
    
Parse();    
}

uint8_t Item::getCmd()
{
  aJsonObject *t = aJson.getArrayItem(itemArr,I_CMD);
  if (t) 
      return t->valueint;
  else return -1;    
}


void Item::setCmd(uint8_t cmd)
{
  aJsonObject *t = aJson.getArrayItem(itemArr,I_CMD);
  if (t) 
  t->valueint=cmd;
}

int Item::getArg(short n) //Return arg int or first array element if Arg is array
{ if (!itemArg) return -1;
  if (itemArg->type==aJson_Int) return itemArg->valueint;
      else if  (itemArg->type==aJson_Array) return aJson.getArrayItem(itemArg,n)->valueint;
           else return -2;
}
/*
int Item::getVal(short n) //Return Val from Value array
{ if (!itemVal) return -1;
      else if  (itemVal->type==aJson_Array) 
            {
            aJsonObject *t = aJson.getArrayItem(itemVal,n);  
            if (t)    return t->valueint;
               else   return -3;
            }           
           else return -2;
}
*/

long int Item::getVal() //Return Val if val is int or first elem of Value array
{ if (!itemVal) return -1;
  if (itemVal->type==aJson_Int) return itemVal->valueint;
      else if  (itemVal->type==aJson_Array) 
            {
            aJsonObject *t = aJson.getArrayItem(itemVal,0);  
            if (t)    return t->valueint;
               else   return -3;
            }           
           else return -2;
}
/*
void Item::setVal(short n, int par)  // Only store  if VAL is array defined in config to avoid waste of RAM
{
  if (!itemVal || itemVal->type!=aJson_Array) return;
  Serial.print(F(" Store p="));Serial.print(n);Serial.print(F(" Val="));Serial.println(par);
  for (int i=aJson.getArraySize(itemVal);i<=n;i++) aJson.addItemToArray(itemVal,aJson.createItem(int(0))); //Enlarge array of Values
  
  aJsonObject *t = aJson.getArrayItem(itemVal,n);
  if (t) t->valueint=par;  
}
*/

void Item::setVal(long int par)  // Only store if VAL is int (autogenerated or config-defined)
{
  if (!itemVal || itemVal->type!=aJson_Int) return;
  Serial.print(F(" Store "));Serial.print(F(" Val="));Serial.println(par);
  itemVal->valueint=par;   
}


boolean Item::isValid ()
{
 return  (itemArr && (itemArr->type==aJson_Array)); 
}



/*
void Item::copyPar (aJsonObject *itemV)
{ int n=aJson.getArraySize(itemV);
  //for (int i=aJson.getArraySize(itemVal);i<n;i++) aJson.addItemToArray(itemVal,aJson.createItem(int(0))); //Enlarge array of Values
   for (int i=0;i<n;i++) setPar(i,aJson.getArrayItem(itemV,i)->valueint); 
}
*/

boolean Item::getEnableCMD(int delta)
{
  return ((millis()-lastctrl>(unsigned long) delta) );//|| (itemArr!=lastobj));
}

int Item::Ctrl(short cmd, short n, int * Par, boolean send)
{

   Serial.print(F("Cmd="));Serial.println(cmd);
  //va_list vl;
  
  int _Par[3]={0,0,0};
  if (Par==NULL) Par=_Par;  
  
  int iaddr=getArg();
  HSVstore st; 
  //Store Parameter(s) into json VAL



  switch (cmd) 
  {
  int t;
   case CMD_TOGGLE: 
    switch (t=getCmd())
    {
    case CMD_ON:
    case CMD_SET:
      cmd=CMD_OFF;
      break;
    case CMD_OFF:
    case CMD_HALT:
    case  0:
    case -1:       //No stored command yet
      cmd=CMD_ON;
      break;       
    }//switch old cmd
   //Serial.print("Tog/oldcmd:");Serial.print(t);Serial.print(" new ");Serial.println(cmd);
   break;
 
   case CMD_RESTORE:
     if (itemType!=CH_GROUP) //individual threating of channels. Ignore restore command for groups
     switch (t=getCmd())
     { 
      case CMD_HALT: //previous command was HALT ?
      Serial.print("Restored from:");Serial.println(t);
      cmd=CMD_ON;    //turning on
      break; 
      default:
      return -3;
    }//switch old cmd 
  } //switch cmd
      
  switch (cmd) 
  {
     case 0: //no command
     
     setCmd(CMD_SET);
     
     switch (itemType)
     {
     
      case CH_RGBW: //only if configured VAL array
       if (!Par[1]) itemType=CH_WHITE;
      case CH_GROUP: //Save for groups as well 
      case CH_RGB:
          st.h=Par[0];
          st.s=Par[1];
          st.v=Par[2];
          setVal(st.aslong);         
          break;
          
      case CH_DIMMER:  //Everywhere, in flat VAL
      case CH_MODBUS:
      case CH_THERMO:
      case CH_VC:
      case CH_VCTEMP:
          setVal(Par[0]);
          
     }//itemtype
     
     lastctrl=millis(); //last controlled object ond timestamp update
     lastobj =itemArr;
     
     break;
     
     case CMD_ON: 
     if (getCmd()!=CMD_ON)
     
        {    
             short params=0;
             //retrive stored values
             st.aslong=getVal(); 
             if (st.aslong>0)  //Stored smthng

                switch (itemType)
               {
                  //case CH_GROUP: 
                  case CH_RGBW: 
                  case CH_RGB:
                  Par[0]=st.h;
                  Par[1]=st.s;
                  Par[2]=st.v;
                  params=3;
                  SendCmd(0,params,Par); // Send restored triplet
                  break;
          
                  case CH_DIMMER:         //Everywhere, in flat VAL
                  case CH_MODBUS:           
                  case CH_VC:
                  case CH_VCTEMP:
                  Par[0]=st.aslong;
                  params=1;
                  SendCmd(0,params,Par);  // Send restored parameter
                  break;
                  case CH_THERMO:
                  Par[0]=st.aslong;
                  params=0;
                  SendCmd(CMD_ON);  // Just ON (switch)
                  break;
                  default:
                  SendCmd(cmd);          // Just send ON
               }//itemtype
                   else 
                      {// Default settings
                        Serial.print(st.aslong); 
                        Serial.println(F(": No stored values - default"));
                        Par[0]=100;
                        Par[1]=0;
                        Par[2]=100;
                      }
                
                               
             for (short i=0;i<params ;i++) 
             {
              Serial.print(F("Restored: "));Serial.print(i);Serial.print("=");Serial.println(Par[i]);
              }
             
              
          setCmd(cmd);
        }
       else
        {  //Double ON - apply special preset - clean white full power
          switch (itemType)
          {
            case CH_RGBW:
              Serial.println(F("Force White"));
              itemType=CH_WHITE;
              Par[1]=0;   //Zero saturation
              Par[2]=100; //Full power
              
              // Store
              st.h=Par[0];
              st.s=Par[1];
              st.v=Par[2];
              setVal(st.aslong);  
              
              //Send to OH
              SendCmd(0,3,Par);          
              break;
          } //itemtype
        } //else

        //Serial.print("Sa:");Serial.println(Par[1]);
        if ((itemType==CH_RGBW) && (Par[1]==0)) itemType=CH_WHITE;
       
       
     break; //CMD_ON
     
     case CMD_OFF:
     
         Par[0]=0;Par[1]=0;Par[2]=0;
         setCmd(cmd);
         SendCmd(cmd);
     break;

     case CMD_HALT:
         
         if (isActive()>0) 
         {
         Par[0]=0;Par[1]=0;Par[2]=0;
         setCmd(cmd);
         SendCmd(CMD_OFF);
         Serial.println(" Halted");
         }
       
     
  }//switch cmd

  

  
       switch (itemType)
       {
       case CH_DIMMER: //Dimmed light       

       DmxSimple.write(iaddr, map(Par[0],0,100,0,255)); 
       
       break;
       
       case CH_RGBW: //Colour RGBW
       {
       int k;
       DmxSimple.write(iaddr+3, k=map((100-Par[1])*Par[2],0,10000,0,255));
       Serial.print(F("W:"));Serial.println(k); 
       }
       case CH_RGB: // RGB
       
       {

      
       CRGB rgb= CHSV(map(Par[0],0,365,0,255),map(Par[1],0,100,0,255),map(Par[2],0,100,0,255));
   
       DmxSimple.write(iaddr,   rgb.r);
       DmxSimple.write(iaddr+1, rgb.g);
       DmxSimple.write(iaddr+2, rgb.b);
          
        
        break; }  
        
        case CH_WHITE:
        DmxSimple.write(iaddr,   0);
        DmxSimple.write(iaddr+1, 0);
        DmxSimple.write(iaddr+2, 0);
        DmxSimple.write(iaddr+3, map(Par[2],0,100,0,255)); 
        break;

        case CH_MODBUS:
        {
          
           if ((itemArg->type == aJson_Array) && (aJson.getArraySize(itemArg)==3))
           {
           int _addr= aJson.getArrayItem(itemArg,0)->valueint;
           int _reg = aJson.getArrayItem(itemArg,1)->valueint;
           int _mask= aJson.getArrayItem(itemArg,2)->valueint;
               
          modbusSet(_addr,_reg,_mask,map(Par[0],0,100,0,0x3f));   
           }
        break;}
          
        case CH_GROUP://Group
        {
        //aJsonObject *groupArr= aJson.getArrayItem(itemArr, 1);      
        if (itemArg->type==aJson_Array)
        { 
          aJsonObject *i =itemArg->child;
          while (i)
            {
            Item it (i->valuestring);
//            it.copyPar(itemVal);   
            it.Ctrl(cmd,n,Par,send); //// was true
            i=i->next;
            } //while
        } //if
        } //case
        break;
        case CH_RELAY:
        {int k;
        pinMode(iaddr,OUTPUT);
        digitalWrite(iaddr,k=((cmd==CMD_ON)?HIGH:LOW));
        Serial.print(F("Pin:"));Serial.print(iaddr);Serial.print(F("="));Serial.println(k);
        break;
        case CH_THERMO:
        ///thermoSet(name,cmd,Par1); all cativities done - update temp & cmd
        break;
        case CH_PWM:
        pinMode(iaddr,OUTPUT);
        //timer 0 for pin 13 and 4
        //timer 1 for pin 12 and 11
        //timer 2 for pin 10 and 9
        //timer 3 for pin 5 and 3 and 2
        //timer 4 for pin 8 and 7 and 6
        //prescaler = 1 ---> PWM frequency is 31000 Hz
        //prescaler = 2 ---> PWM frequency is 4000 Hz
        //prescaler = 3 ---> PWM frequency is 490 Hz (default value)
        //prescaler = 4 ---> PWM frequency is 120 Hz
        //prescaler = 5 ---> PWM frequency is 30 Hz
        //prescaler = 6 ---> PWM frequency is <20 Hz
        int tval = 7;             // this is 111 in binary and is used as an eraser
        TCCR4B &= ~tval;   // this operation (AND plus NOT),  set the three bits in TCCR2B to 0
        TCCR3B &= ~tval; 
        tval=2;
        TCCR4B|=tval;
        TCCR3B|=tval;
         
        analogWrite(iaddr,k=map(Par[0],0,100,0,255));
        Serial.print(F("Pin:"));Serial.print(iaddr);Serial.print(F("="));Serial.println(k);
        break;
        }

        case CH_VC:
      
        VacomSetFan(getArg(),Par[0]);
        break;
       
        
        case CH_VCTEMP:  
          {
          Item it (itemArg->valuestring);
          if (it.isValid() && it.itemType==CH_VC)
                        VacomSetHeat(it.getArg(),Par[0],cmd);            
          break;  
          }
        
        
       } // switch itemtype
     //  break;
 
   
   
  }
/*

short thermoSet(char * name, short cmd, short t)
{
  
if (items)
  {
  aJsonObject *item= aJson.getObjectItem(items, name);
  if (item && (item->type==aJson_Array) && (aJson.getArrayItem(item, I_TYPE)->valueint==CH_THERMO))
      { 

        for (int i=aJson.getArraySize(item);i<4;i++) aJson.addItemToArray(item,aJson.createItem(int(0))); //Enlarge item to 4 elements
             if (!cmd) aJson.getArrayItem(item, I_VAL)->valueint=t;
             aJson.getArrayItem(item, I_CMD)->valueint=cmd;
            
      }     
     
  } 
}


void PooledItem::Idle()
{
if (PoolingInterval)
  {
    Pool();
    next=millis()+PoolingInterval;
  }        
   
};





addr 10d
Снять аварию 42001 (2001=7d1) =>4

[22:20:33] Write task has completed successfully
[22:20:33] <= Response: 0A 06 07 D0 00 04 89 FF 
[22:20:32] => Poll: 0A 06 07 D0 00 04 89 FF 

100%
2003-> 10000
[22:24:05] Write task has completed successfully
[22:24:05] <= Response: 0A 06 07 D2 27 10 33 C0 
[22:24:05] => Poll: 0A 06 07 D2 27 10 33 C0 

ON
2001->1
[22:24:50] Write task has completed successfully
[22:24:50] <= Response: 0A 06 07 D0 00 01 49 FC 
[22:24:50] => Poll: 0A 06 07 D0 00 01 49 FC 

OFF 
2001->0
[22:25:35] Write task has completed successfully
[22:25:35] <= Response: 0A 06 07 D0 00 00 88 3C 
[22:25:34] => Poll: 0A 06 07 D0 00 00 88 3C 


POOL  2101x10
[22:27:29] <= Response: 0A 03 14 00 23 00 00 27 10 13 88 0B 9C 00 32 00 F8 00 F2 06 FA 01 3F AD D0 
[22:27:29] => Poll: 0A 03 08 34 00 0A 87 18 

*/


extern ModbusMaster node;

int Item::VacomSetFan (int addr, int8_t  val)
{
  Serial.print("VC#");Serial.print(addr);Serial.print("=");Serial.println(val);


  if (modbusBusy) return -1;
    modbusBusy=1;
    
  uint8_t j, result;
  uint16_t data[1];
   node.begin(9600,SERIAL_8N1,13);

  node.setSlave(addr);

  if (val) {
    node.writeSingleRegister(2001-1,4+1);//delay(500);
    //node.writeSingleRegister(2001-1,1);
           } 
    else node.writeSingleRegister(2001-1,0);
  delay (500);
  node.writeSingleRegister(2003-1,val*100);

  /*
  result = node.readHoldingRegisters(2003, 10);
  
  // do something with data if read is successful
  if (result == node.ku8MBSuccess)
  { Serial.print(F(" FM Val :"));
    for (j = 0; j < 10; j++)
    {
      data[j] = node.getResponseBuffer(j);
      Serial.print(data[j],HEX);Serial.print("-");
     
    }
    Serial.println();
  } else  {Serial.print(F("Modbus pooling error=")); Serial.println(result,HEX); }
  */
   modbusBusy=0;


  
  
}

int Item::VacomSetHeat(int addr, int8_t val, int8_t  cmd)
  {
  Serial.print("VC#");Serial.print(addr);Serial.print("=");Serial.print(val);Serial.print(" cmd=");Serial.println(cmd);
    
  }
int Item::SendCmd(short cmd,short n, int * Par)
 {

/// ToDo: relative patches, configuration
  
  char addrstr[32];
  //char addrbuf[17];
  char valstr[16]="";

  strcpy_P (addrstr,outprefix);
  strncat (addrstr,itemArr->name,sizeof(addrstr)); ////
 
  
  switch (cmd)
  {
    case CMD_ON:
    strcpy(valstr,"ON");
    break;
    case CMD_OFF:
    case CMD_HALT:
    strcpy(valstr,"OFF");
    break;
    // TODO send Par
    case 0:
    case CMD_SET:
    if (Par) 
              for (short i=0;i<n;i++) 
                      {
                      char num [4];  
                      snprintf(num,sizeof(num),"%d",Par[i]);
                      strncat(valstr,num,sizeof(valstr));
                      if (i!=n-1)    
                          {
                            strcpy(num,",");
                            strncat(valstr,num,sizeof(valstr));
                          }
                      }
     break;                 
     default:
     return -1;
  }
  
  Serial.print(addrstr);Serial.print("->");Serial.println(valstr);
  client.publish(addrstr, valstr);
  return 0; 
 }

 int Item::isActive()
{
  HSVstore st;
  int val=0; 

if (!isValid()) return -1;
//Serial.print(itemArr->name);
int cmd=getCmd();
switch (cmd)
{
  case CMD_ON:
  //Serial.println(" active");
  return 1;
  case CMD_OFF:
  case CMD_HALT:
  //Serial.println(" inactive");
  return 0;
}

 st.aslong=getVal(); 
 
switch (itemType)
{
  //case CH_GROUP: 
   case CH_RGBW: 
   case CH_RGB:
                 
   val=st.v;
   break;
   
   case CH_DIMMER:         //Everywhere, in flat VAL
   case CH_MODBUS:
   case CH_THERMO:
   case CH_VC:
   case CH_VCTEMP:
   val=st.aslong;
} //switch 
Serial.print(":=");Serial.println(val);
 if (val) return 1; else return 0;
}



  int modbusSet(int addr, uint16_t _reg, int _mask, uint16_t value)
  {
   
   if (modbusBusy) return -1;
   modbusBusy=1;
   node.begin(9600,SERIAL_8E1,13);
   node.setSlave(addr);  
  
if (_mask) 
      {value <<= 8; value |= (0xff);}
  else {value &= 0xff; value |= (0xff00);}
  
  Serial.print(addr);Serial.print("=>");Serial.print(_reg,HEX);Serial.print(":");Serial.println(value,HEX);
  
  node.writeSingleRegister(_reg,value);
  modbusBusy=0; 
  }


int Item::checkFM()
  {
      if (modbusBusy) return -1;
    modbusBusy=1;
    
  uint8_t j, result;
  uint16_t data[1];
   node.begin(9600,SERIAL_8N1,13);

  node.setSlave(getArg());
  
  result = node.readHoldingRegisters(2101-1, 10);
  
  // do something with data if read is successful
  if (result == node.ku8MBSuccess)
  { Serial.print(F(" FM Val :"));
    for (j = 0; j < 10; j++)
    {
      data[j] = node.getResponseBuffer(j);
      Serial.print(data[j],HEX);Serial.print("-");
     
    }
    Serial.println();
  } else  {Serial.print(F("Modbus pooling error=")); Serial.println(result,HEX); }

result = node.readHoldingRegisters(20-1, 4);
  
  // do something with data if read is successful
  if (result == node.ku8MBSuccess)
  { Serial.print(F(" PI Val :"));
    for (j = 0; j < 4; j++)
    {
      data[j] = node.getResponseBuffer(j);
      Serial.print(data[j]);Serial.print("-");
     
    }
    Serial.println();
  } else  {Serial.print(F("Modbus pooling error=")); Serial.println(result,HEX); }

  

   modbusBusy=0;
  }

  
  int Item::checkModbus()
  {
  if (modbusBusy) return -1;
  modbusBusy=1;
  
  node.begin(9600,SERIAL_8E1,13);
   
  uint8_t   result;
  
  uint16_t  addr =  getArg(0);
  uint16_t  reg  =  getArg(1);
  short     mask  = getArg(2);
 
  int       data;
  
  node.setSlave(addr);
  result = node.readHoldingRegisters(reg, 1);

  if (result == node.ku8MBSuccess)
  { 
       data=node.getResponseBuffer(0);
       Serial.print(F("Modbus Val: ")); Serial.println(data,HEX);
       checkModbus(data);  
  } 
     else  {Serial.print(F("Modbus pooling error=")); Serial.println(result,HEX); }

   modbusBusy=0;

// Looking 1 step ahead for modbus item, which uses same register
 Item nextItem(modbusitem->next);
 if (modbusitem &&  nextItem.isValid() && nextItem.itemType==CH_MODBUS && nextItem.getArg(0)==addr && nextItem.getArg(1)==reg)
 {
  nextItem.checkModbus(data); 
  modbusitem=modbusitem->next;
  if (!modbusitem) modbusitem=items->child; 
 }
   
}


int Item::checkModbus(int data)
  {
  short     mask  = getArg(2);
  int       d=data;
  if (mask) d>>=8; d&=0xff;
  d=map(d,0,0x3f,0,100);
  int cmd=getCmd();
   Serial.println(d);
       if (getVal()!=d || d && cmd==CMD_OFF || d && cmd==CMD_HALT) //volume changed or turned on manualy 
            { 
            if (d)  
                
                { // Actually turned on
                  if (cmd==CMD_OFF || cmd==CMD_HALT) SendCmd(CMD_ON); //update OH with ON if it was turned off before
                  SendCmd(0,1,&d); //update OH with value
                  setCmd(CMD_ON);  //store command
                  setVal(d);       //store value
                  } 
              else {
                  if (cmd!=CMD_HALT && cmd!=CMD_OFF) 
                 {
                  setCmd(CMD_OFF); // store command (not value)
                  SendCmd(CMD_OFF);// update OH
                  }
                    }
                } //if data changed
  }             
   
int Item::Pool()
{
  switch (itemType)
          {
            case CH_MODBUS:
                checkModbus();
                break;
            case CH_VC:
                checkFM();
          }      
  }