#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]["hrn"] = "WAV progressive";
    capa["methods"][0u]["priority"] = 1;
    config->addStandardPushCapabilities(capa);
    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;
    if (M.getCodec(thisIdx) == "PCM"){
      if (M.getSize(thisIdx) > 8 && swappy.allocate(len)){
        if (M.getSize(thisIdx) == 16){
          for (uint32_t i = 0; i < len; i += 2){
            swappy[i] = dataPointer[i + 1];
            swappy[i + 1] = dataPointer[i];
          }
        }
        if (M.getSize(thisIdx) == 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);
    }
    size_t mainTrack = 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 (!M.getLive()){
      DTSC::Keys keys(M.keys(mainTrack));
      total_data = 0;
      size_t keyCount = keys.getEndValid();
      for (size_t i = 0; i < keyCount; ++i){total_data += keys.getSize(i);}
    }
    Bit::htobl_le(riffHeader + 4, 80 + total_data);
    myConn.SendNow(riffHeader, 12);
    // Send format details
    uint16_t fmt = 0;
    std::string codec = M.getCodec(mainTrack);
    if (codec == "ALAW"){fmt = 6;}
    if (codec == "ULAW"){fmt = 7;}
    if (codec == "PCM"){fmt = 1;}
    if (codec == "FLOAT"){fmt = 3;}
    if (codec == "MP3"){fmt = 85;}
    myConn.SendNow(RIFF::fmt::generate(
        fmt, M.getChannels(mainTrack), M.getRate(mainTrack), M.getBps(mainTrack),
        M.getChannels(mainTrack) * (M.getSize(mainTrack) << 3), M.getSize(mainTrack)));
    // Send sample count per channel
    if (fmt != 1){// Not required for PCM
      if (!M.getLive()){
        myConn.SendNow(RIFF::fact::generate(
            ((M.getLastms(mainTrack) - M.getFirstms(mainTrack)) * M.getRate(mainTrack)) / 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;
  }
}// namespace Mist