#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <string>

#include <mist/json.h>
#include <mist/dtsc.h>
#include <mist/procs.h>
#include <mist/timing.h>

namespace Info {
  int getInfo(int argc, char* argv[]) {
    if (argc < 2){
      fprintf( stderr, "Usage: %s <filename>\n", argv[0] );
      return 1;
    }
    DTSC::File F(argv[1]);
    JSON::Value fileSpecs = F.getMeta();
    if( !fileSpecs ) {
      char ** cmd = (char**)malloc(3*sizeof(char*));
      cmd[0] = (char*)"ffprobe";
      cmd[1] = argv[1];
      cmd[2] = NULL;
      int outFD = -1;
      Util::Procs::StartPiped("FFProbe", cmd, 0, 0, &outFD);
      while( Util::Procs::isActive("FFProbe")){ Util::sleep(100); }
      FILE * outFile = fdopen( outFD, "r" );
      char * fileBuf = 0;
      size_t fileBufLen = 0;
      while ( !(feof(outFile) || ferror(outFile)) && (getline(&fileBuf, &fileBufLen, outFile) != -1)){
        std::string line = fileBuf;
        if (line.find("Input") != std::string::npos){
          std::string tmp = line.substr(line.find(", ") + 2);
          fileSpecs["format"] = tmp.substr(0, tmp.find(","));
        }
        if (line.find("Duration") != std::string::npos){
          std::string tmp = line.substr(line.find(": ", line.find("Duration")) + 2);
          tmp = tmp.substr(0, tmp.find(","));
          int length = (((atoi(tmp.substr(0,2).c_str()) * 60) + atoi(tmp.substr(3,2).c_str())) * 60) + atoi(tmp.substr(6,2).c_str());
          fileSpecs["length"] = length;
          length *= 100;
          length += atoi(tmp.substr(9,2).c_str());
          fileSpecs["lastms"] = length * 10;
        }
        if (line.find("bitrate") != std::string::npos ){
          std::string tmp = line.substr(line.find(": ", line.find("bitrate")) + 2);
          fileSpecs["bps"] = atoi(tmp.substr(0, tmp.find(" ")).c_str()) * 128;
        }
        if (line.find("Stream") != std::string::npos ){
          std::string tmp = line.substr(line.find(" ", line.find("Stream")) + 1);
          int strmIdx = fileSpecs["streams"].size();
          int curPos = 0;
          curPos = tmp.find(": ", curPos) + 2;
          fileSpecs["streams"][strmIdx]["type"] = tmp.substr(curPos, tmp.find(":", curPos) - curPos);
          curPos = tmp.find(":", curPos) + 2;
          fileSpecs["streams"][strmIdx]["codec"] = tmp.substr(curPos, tmp.find_first_of(", ", curPos) - curPos);
          curPos = tmp.find(",", curPos) + 2;
          if (fileSpecs["streams"][strmIdx]["type"] == "Video"){
            fileSpecs["streams"][strmIdx]["encoding"] = tmp.substr(curPos, tmp.find(",", curPos) - curPos);
            curPos = tmp.find(",", curPos) + 2;
            fileSpecs["streams"][strmIdx]["width"] = atoi(tmp.substr(curPos, tmp.find("x", curPos) - curPos).c_str());
            curPos = tmp.find("x", curPos) + 1;
            fileSpecs["streams"][strmIdx]["height"] = atoi(tmp.substr(curPos, tmp.find(",", curPos) - curPos).c_str());
            curPos = tmp.find(",", curPos) + 2;
            fileSpecs["streams"][strmIdx]["bps"] = atoi(tmp.substr(curPos, tmp.find(" ", curPos) - curPos).c_str()) * 128;
            curPos = tmp.find(",", curPos) + 2;
            fileSpecs["streams"][strmIdx]["fpks"] = (int)(atof(tmp.substr(curPos, tmp.find(" ", curPos) - curPos).c_str()) * 1000);
            fileSpecs["streams"][strmIdx].removeMember( "type" );
            fileSpecs["video"] = fileSpecs["streams"][strmIdx];
          }else if (fileSpecs["streams"][strmIdx]["type"] == "Audio"){
            fileSpecs["streams"][strmIdx]["samplerate"] = atoi(tmp.substr(curPos, tmp.find(" ", curPos) - curPos).c_str());
            curPos = tmp.find(",", curPos) + 2;
            if (tmp.substr(curPos, tmp.find(",", curPos) - curPos) == "stereo"){
              fileSpecs["streams"][strmIdx]["channels"] = 2;
            }else if (tmp.substr(curPos, tmp.find(",", curPos) - curPos) == "mono"){
              fileSpecs["streams"][strmIdx]["channels"] = 1;
            }else{
              fileSpecs["streams"][strmIdx]["channels"] = tmp.substr(curPos, tmp.find(",", curPos) - curPos);
            }
            curPos = tmp.find(",", curPos) + 2;
            fileSpecs["streams"][strmIdx]["samplewidth"] = tmp.substr(curPos, tmp.find(",", curPos) - curPos);
            curPos = tmp.find(",", curPos) + 2;
            fileSpecs["streams"][strmIdx]["bps"] = atoi(tmp.substr(curPos, tmp.find(" ", curPos) - curPos).c_str()) * 128;
            fileSpecs["streams"][strmIdx].removeMember( "type" );
            fileSpecs["audio"] = fileSpecs["streams"][strmIdx];
          }
        }
      }
      fclose( outFile );
      fileSpecs.removeMember( "streams" );
    } else {
      fileSpecs["format"] = "dtsc";
      JSON::Value tracks = fileSpecs["tracks"];
      for(JSON::ObjIter trackIt = tracks.ObjBegin(); trackIt != tracks.ObjEnd(); trackIt++){
        if (fileSpecs["tracks"][trackIt->first].isMember("init")){
          fileSpecs["tracks"][trackIt->first].removeMember("init");
        }
        if (fileSpecs["tracks"][trackIt->first].isMember("frags")){
          fileSpecs["tracks"][trackIt->first].removeMember("frags");
        }
        if (fileSpecs["tracks"][trackIt->first].isMember("keys")){
          fileSpecs["tracks"][trackIt->first].removeMember("keys");
        }
      }
    }
    if (fileSpecs.isMember("video")){
      fileSpecs["video"].removeMember("init");
    }
    if (fileSpecs.isMember("audio")){
      fileSpecs["audio"].removeMember("init");
    }
    printf( "%s", fileSpecs.toString().c_str() );
    return 0;
  }
}

int main(int argc, char* argv[]) {
  return Info::getInfo(argc, argv);
}