xbmc-txupdate.cpp

/*
 *      Copyright (C) 2014 Team Kodi
 *      http://xbmc.org
 *
 *  This Program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2, or (at your option)
 *  any later version.
 *
 *  This Program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with Kodi; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 *  http://www.gnu.org/copyleft/gpl.html
 *
 */

#ifdef _MSC_VER
#define _CRT_SECURE_NO_WARNINGS
#endif

#if defined( WIN32 ) && defined( TUNE )
  #include <crtdbg.h>
  _CrtMemState startMemState;
  _CrtMemState endMemState;
#endif

#include <string>
#include <stdio.h>
#include "lib/ProjectHandler.h"
#include "lib/HTTPUtils.h"
#include "lib/xbmclangcodes.h"
#include "lib/Settings.h"

using namespace std;

void PrintUsage()
{
  printf
  (
  "Usage: xbmc-txpudate PROJECTDIR [working mode]\n\n"
  "PROJECTDIR: the directory which contains the xbmc-txupdate.xml settings file and the .passwords file.\n"
  "            This will be the directory where your merged and transifex update files get generated.\n\n"
  "Working modes:\n"
  "     -d   Only download to local cache, without performing a merge.\n"
  "     -dm  Download and create merged and tx update files, but no upload performed.\n"
  "     -dmu Download, merge and upload the new files to transifex.\n"
  "     -u   Only upload the previously prepared files. Note that this needs download and merge ran before.\n\n"
  "     No working mode arguments used, performs as -dm\n\n"
  );
  #ifdef _MSC_VER
  printf
  (
  "Note for Windows users: In case you have whitespace or any special character\n"
  "in the directory argument, please use apostrophe around them. For example:\n"
  "xbmc-txupdate.exe \"C:\\xbmc dir\\language\"\n"
  );
  #endif
  return;
};

int main(int argc, char* argv[])
{
  setbuf(stdout, NULL);
  if (argc > 3 || argc < 2)
  {
    printf ("\nBad arguments given\n\n");
    PrintUsage();
    return 1;
  }

  std::string WorkingDir, strMode;
  bool bDownloadNeeded = false;
  bool bMergeNeeded = false;
  bool bUploadNeeded = false;
  bool bForceUpload;

  if (argv[1])
   WorkingDir = argv[1];
  if (WorkingDir.empty() || !g_File.DirExists(WorkingDir))
  {
    printf ("\nMissing or wrong project directory specified: %s, stopping.\n\n", WorkingDir.c_str());
    PrintUsage();
    return 1;
  }

  if (argc == 3)
  {
    if (argv[2])
      strMode = argv[2];
    if (strMode.empty() && strMode[0] != '-')
    {
      printf ("\nMissing or wrong working mode format used. Stopping.\n\n");
      PrintUsage();
      return 1;
    }
    if (strMode == "-d")
      bDownloadNeeded = true;
    else if (strMode == "-dm" || strMode == "-m")
      {bDownloadNeeded = true; bMergeNeeded = true;}
    else if (strMode == "-dmu")
      {bDownloadNeeded = true; bMergeNeeded = true; bUploadNeeded = true;}
    else if (strMode == "-fu")
    {
      bUploadNeeded = true;
      bForceUpload = true;
    }
    else if (strMode == "-u")
      bUploadNeeded = true;

    else
    {
      printf ("\nWrong working mode arguments used. Stopping.\n\n");
      PrintUsage();
      return 1;
    }
  }
  if (argc == 2)
    {bDownloadNeeded = true; bMergeNeeded = true;}

  printf("\nXBMC-TXUPDATE v%s by Team Kodi\n", VERSION.c_str());

  try
  {
    if (WorkingDir[WorkingDir.length()-1] != DirSepChar)
      WorkingDir.append(&DirSepChar);

    CLog::SetbWriteSyntaxLog(bDownloadNeeded);
    CLog::Init(WorkingDir + "xbmc-txupdate.log", WorkingDir + "txupdate-syntax.log");
    CLog::Log(logINFO, "Root Directory: %s", WorkingDir.c_str());

    g_HTTPHandler.LoadCredentials(WorkingDir + ".passwords.xml");
    g_HTTPHandler.SetCacheDir(WorkingDir + ".httpcache");

    CProjectHandler TXProject;
    TXProject.InitUpdateXMLHandler(WorkingDir);
    g_LCodeHandler.Init("https://raw.github.com/xbmc/translations/master/tool/TXLanguages/all_languages.json");
    g_LCodeHandler.ReadWhiteBlackLangList(WorkingDir + "langwhitelist.xml");
    g_LCodeHandler.ReadWhiteBlackLangList(WorkingDir + "langblacklist.xml");

    if (bDownloadNeeded)
    {
//      if (!g_File.FileExist(WorkingDir + ".httpcache" + DirSepChar + ".lastdloadtime") ||
//          g_File.GetFileAge(WorkingDir + ".httpcache" + DirSepChar + ".lastdloadtime") > g_Settings.GetHTTPCacheExpire() * 60)
//      {
//        g_File.DeleteDirectory(WorkingDir + ".httpcache"); // Clean the complete cache as all our files in there are outdated
//        g_HTTPHandler.SetCacheDir(WorkingDir + ".httpcache");
//      }

      g_File.WriteFileFromStr(WorkingDir + ".httpcache" + DirSepChar + ".dload_merge_status", "fail");
      g_File.WriteFileFromStr(WorkingDir + ".httpcache" + DirSepChar + ".lastdloadtime", "Last download time: " + g_File.GetCurrTime());

      printf("\n");
      printf("----------------------------------------\n");
      printf("DOWNLOADING RESOURCES FROM TRANSIFEX.NET\n");
      printf("----------------------------------------\n");

      CLog::Log(logLINEFEED, "");
      CLog::Log(logINFO, "****************************************");
      CLog::Log(logINFO, "DOWNLOADING RESOURCES FROM TRANSIFEX.NET");
      CLog::Log(logINFO, "****************************************");

      TXProject.FetchResourcesFromTransifex();

      printf("\n");
      printf("-----------------------------------\n");
      printf("DOWNLOADING RESOURCES FROM UPSTREAM\n");
      printf("-----------------------------------\n");

      CLog::Log(logLINEFEED, "");
      CLog::Log(logINFO, "***********************************");
      CLog::Log(logINFO, "DOWNLOADING RESOURCES FROM UPSTREAM");
      CLog::Log(logINFO, "***********************************");

      TXProject.FetchResourcesFromUpstream();

      if (bMergeNeeded)
      {
        printf("\n");
        printf("-----------------\n");
        printf("MERGING RESOURCES\n");
        printf("-----------------\n");

        CLog::Log(logLINEFEED, "");
        CLog::Log(logINFO, "*****************");
        CLog::Log(logINFO, "MERGING RESOURCES");
        CLog::Log(logINFO, "*****************");

        TXProject.CreateMergedResources();

        printf("\n");
        printf("--------------------------------------------\n");
        printf("WRITING MERGED AND TXUPDATE RESOURCES TO HDD\n");
        printf("--------------------------------------------\n");

        CLog::Log(logLINEFEED, "");
        CLog::Log(logINFO, "********************************************");
        CLog::Log(logINFO, "WRITING MERGED AND TXUPDATE RESOURCES TO HDD");
        CLog::Log(logINFO, "********************************************");

        TXProject.WriteResourcesToFile(WorkingDir);
        g_File.CopyFile(WorkingDir + "xbmc-txupdate.xml", WorkingDir + ".httpcache" + DirSepChar + ".last_xbmc-txupdate.xml");
      }
      g_File.WriteFileFromStr(WorkingDir + ".httpcache" + DirSepChar + ".dload_merge_status", "ok");
    }

    if (bUploadNeeded)
    {
      CLog::SetbWriteSyntaxLog(false);
      if (!bForceUpload && g_File.ReadFileToStrE(WorkingDir + ".httpcache" + DirSepChar + ".dload_merge_status") != "ok")
        CLog::Log(logERROR, "There was no successful download and merge run before. Please (re)run download and merge.");

      if (!bForceUpload && g_File.ReadFileToStrE(WorkingDir + ".httpcache" + DirSepChar + ".last_xbmc-txupdate.xml") !=
          g_File.ReadFileToStrE(WorkingDir + "xbmc-txupdate.xml"))
        CLog::Log(logERROR, "xbmc-txupdate.xml file changed since last download and merge. Please (re)run download and merge.");

      printf("\n");
      printf("-----------------------------------------\n");
      printf("UPLOADING LANGUAGE FILES TO TRANSIFEX.NET\n");
      printf("-----------------------------------------\n");

      TXProject.UploadTXUpdateFiles(WorkingDir);
    }

    CLog::SetbWriteSyntaxLog(bDownloadNeeded);

    if (CLog::GetWarnCount() ==0)
    {
      printf("\n");
      printf("--------------------------------------------\n");
      printf("PROCESS FINISHED SUCCESFULLY WITHOUT WARNINGS\n");
      printf("SYNTAX WARNINGS %i\n", CLog::GetSyntaxWarnCount());
      printf("--------------------------------------------\n\n");
    }
    else
    {
      printf("\n");
      printf("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
      printf("PROCESS FINISHED WITH %i WARNINGS\n", CLog::GetWarnCount());
      printf("SYNTAX WARNINGS %i\n", CLog::GetSyntaxWarnCount());
      printf("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n\n");
    }

    CLog::Close();
    g_HTTPHandler.Cleanup();
    return 0;
  }
  catch (const int calcError)
  {
    g_HTTPHandler.Cleanup();
    CLog::Close();
    return 100;
  }
}