#include "output_wav.h"
#include <mist/riff.h>
#include <mist/util.h>

namespace Mist{
  OutWAV::OutWAV(Socket::Connection &conn) : HTTPOutput(conn){}

  void OutWAV::init(Util::Config *cfg){
    HTTPOutput::init(cfg);
    capa["name"] = "WAV";
    capa["friendly"] = "WAV over HTTP";
    capa["desc"] = "Pseudostreaming in WAV format over HTTP";
    capa["url_rel"] = "/$.wav";
    capa["url_match"] = "/$.wav";
    capa["codecs"][0u][0u].append("ALAW");
    capa["codecs"][0u][0u].append("ULAW");
    capa["codecs"][0u][0u].append("MP3");
    capa["codecs"][0u][0u].append("PCM");
    capa["codecs"][0u][0u].append("FLOAT");
    capa["methods"][0u]["handler"] = "http";
    capa["methods"][0u]["type"] = "html5/audio/wav";
    capa["methods"][0u]["priority"] = 1;
    capa["push_urls"].append("/*.wav");

    JSON::Value opt;
    opt["arg"] = "string";
    opt["default"] = "";
    opt["arg_num"] = 1;
    opt["help"] = "Target filename to store WAV file as, or - for stdout.";
    cfg->addOption("target", opt);
  }

  bool OutWAV::isRecording(){return config->getString("target").size();}

  void OutWAV::sendNext(){
    char *dataPointer = 0;
    size_t len = 0;
    thisPacket.getString("data", dataPointer, len);

    //PCM must be converted to little-endian if > 8 bits per sample
    static Util::ResizeablePointer swappy;
    DTSC::Track & trk = myMeta.tracks[thisPacket.getTrackId()];
    if (trk.codec == "PCM"){
      if (trk.size > 8 && swappy.allocate(len)){
        if (trk.size == 16){
          for (uint32_t i = 0; i < len; i+=2){
            swappy[i] = dataPointer[i+1];
            swappy[i+1] = dataPointer[i];
          }
        }
        if (trk.size == 24){
          for (uint32_t i = 0; i < len; i+=3){
            swappy[i] = dataPointer[i+2];
            swappy[i+1] = dataPointer[i+1];
            swappy[i+2] = dataPointer[i];
          }
        }
        dataPointer = swappy;
      }
    }

    myConn.SendNow(dataPointer, len);
  }

  void OutWAV::sendHeader(){
    if (!isRecording()){
      H.Clean();
      H.SetHeader("Content-Type", "audio/wav");
      H.protocol = "HTTP/1.0";
      H.setCORSHeaders();
      H.SendResponse("200", "OK", myConn);
    }
    DTSC::Track &Trk = myMeta.tracks[getMainSelectedTrack()];
    // Send WAV header
    char riffHeader[] = "RIFF\377\377\377\377WAVE";
    // For live we send max allowed size
    // VoD size of the whole thing is RIFF(4)+fmt(26)+fact(12)+LIST(30)+data(8)+data itself
    uint32_t total_data = 0xFFFFFFFFul - 80;
    if (!myMeta.live){
      total_data = 0;
      for (std::deque<unsigned long>::iterator it = Trk.keySizes.begin(); it != Trk.keySizes.end();
           ++it){
        total_data += *it;
      }
    }
    Bit::htobl_le(riffHeader + 4, 80 + total_data);
    myConn.SendNow(riffHeader, 12);
    // Send format details
    uint16_t fmt = 0;
    if (Trk.codec == "ALAW"){fmt = 6;}
    if (Trk.codec == "ULAW"){fmt = 7;}
    if (Trk.codec == "PCM"){fmt = 1;}
    if (Trk.codec == "FLOAT"){fmt = 3;}
    if (Trk.codec == "MP3"){fmt = 85;}
    myConn.SendNow(RIFF::fmt::generate(fmt, Trk.channels, Trk.rate, Trk.bps,
                                       Trk.channels * (Trk.size << 3), Trk.size));
    // Send sample count per channel
    if (fmt != 1){//Not required for PCM
      if (!myMeta.live){
        myConn.SendNow(RIFF::fact::generate(((Trk.lastms - Trk.firstms) * Trk.rate) / 1000));
      }else{
        myConn.SendNow(RIFF::fact::generate(0xFFFFFFFFul));
      }
    }
    // Send MistServer identifier
    myConn.SendNow("LIST\026\000\000\000infoISFT\012\000\000\000MistServer", 30);
    // Start data chunk
    char dataChunk[] = "data\377\377\377\377";
    Bit::htobl_le(dataChunk + 4, total_data);
    myConn.SendNow(dataChunk, 8);
    sentHeader = true;
  }

  void OutWAV::onHTTP(){
    std::string method = H.method;

    H.Clean();
    H.setCORSHeaders();
    if (method == "OPTIONS" || method == "HEAD"){
      H.SetHeader("Content-Type", "audio/wav");
      H.protocol = "HTTP/1.0";
      H.SendResponse("200", "OK", myConn);
      H.Clean();
      return;
    }

    parseData = true;
    wantRequest = false;
  }
}