/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim:set ts=2 sw=2 sts=2 et cindent: */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef nsUpdateDriver_h__

#define nsUpdateDriver_h__

#include "nscore.h"

#include "nsIUpdateService.h"

#include "nsIThread.h"

#include "nsCOMPtr.h"

#include "nsString.h"

#include "mozilla/Attributes.h"

class nsIFile;

#if defined(XP_WIN)

#include <windows.h>

  typedef HANDLE     ProcessType;

#elif defined(XP_UNIX)

  typedef pid_t      ProcessType;

#else

#include "prproces.h"

  typedef PRProcess* ProcessType;

#endif

/**

 * This function processes any available updates.  As part of that process, it

 * may exit the current process and relaunch it at a later time.

 *

 * Two directories are passed to this function: greDir (where the actual

 * binary resides) and appDir (which contains application.ini for XULRunner

 * apps). If this is not a XULRunner app then appDir is identical to greDir.

 *

 * The argc and argv passed to this function should be what is needed to

 * relaunch the current process.

 *

 * The appVersion param passed to this function is the current application's

 * version and is used to determine if an update's version is older than the

 * current application version.

 *

 * If you want the update to be processed without restarting, set the restart

 * parameter to false.

 *

 * This function does not modify appDir.

 */

nsresult ProcessUpdates(nsIFile *greDir, nsIFile *appDir, nsIFile *updRootDir,

                        int argc, char **argv, const char *appVersion,

                        bool restart = true, ProcessType *pid = nullptr);

// The implementation of the update processor handles the task of loading the

// updater application for staging an update.

// XXX ehsan this is living in this file in order to make use of the existing

// stuff here, we might want to move it elsewhere in the future.

class nsUpdateProcessor final : public nsIUpdateProcessor

{

public:

  nsUpdateProcessor();

  NS_DECL_THREADSAFE_ISUPPORTS

  NS_DECL_NSIUPDATEPROCESSOR

private:

  ~nsUpdateProcessor();

  struct StagedUpdateInfo {

    StagedUpdateInfo()

      : mArgc(0),

        mArgv(nullptr),

        mIsOSUpdate(false)

    {}

    ~StagedUpdateInfo() {

      for (int i = 0; i < mArgc; ++i) {

        delete[] mArgv[i];

      }

      delete[] mArgv;

    }

    nsCOMPtr<nsIFile> mGREDir;

    nsCOMPtr<nsIFile> mAppDir;

    nsCOMPtr<nsIFile> mUpdateRoot;

    nsCOMPtr<nsIFile> mOSApplyToDir;

    int mArgc;

    char **mArgv;

    nsCString mAppVersion;

    bool mIsOSUpdate;

  };

private:

  void StartStagedUpdate();

  void WaitForProcess();

  void UpdateDone();

  void ShutdownWatcherThread();

private:

  ProcessType mUpdaterPID;

  nsCOMPtr<nsIThread> mProcessWatcher;

  StagedUpdateInfo mInfo;

};

#endif  // nsUpdateDriver_h__