// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/updater/app/app_unzip_worker.h"

#include <memory>
#include <utility>

#include "base/command_line.h"
#include "base/functional/bind.h"
#include "base/memory/scoped_refptr.h"
#include "chrome/updater/app/app.h"
#include "chrome/updater/constants.h"
#include "components/services/unzip/public/mojom/unzipper.mojom.h"
#include "components/services/unzip/unzipper_impl.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/self_owned_receiver.h"
#include "mojo/public/cpp/platform/platform_channel.h"
#include "mojo/public/cpp/system/invitation.h"
#include "mojo/public/cpp/system/message_pipe.h"

namespace updater {

namespace {

class AppUnzipWorker : public App {
 public:
  AppUnzipWorker();

 private:
  ~AppUnzipWorker() override;
  int Initialize() override;
  void FirstTaskRun() override;
};

class UnzipperImplWithShutdown : public unzip::UnzipperImpl {
 public:
  explicit UnzipperImplWithShutdown(base::OnceClosure shutdown_callback)
      : shutdown_callback_(std::move(shutdown_callback)) {}

  ~UnzipperImplWithShutdown() override { std::move(shutdown_callback_).Run(); }

 private:
  base::OnceClosure shutdown_callback_;
};

}  // namespace

AppUnzipWorker::AppUnzipWorker() = default;

AppUnzipWorker::~AppUnzipWorker() = default;

int AppUnzipWorker::Initialize() {
  return kErrorOk;
}

void AppUnzipWorker::FirstTaskRun() {
  // This process must be started with the command line switch
  // `--mojo-platform-channel-handle=N`. In other words, the command line
  // must be prepared by
  // `mojo::PlatformChannel::PrepareToPassRemoteEndpoint()`.
  mojo::PlatformChannelEndpoint endpoint =
      mojo::PlatformChannel::RecoverPassedEndpointFromCommandLine(
          *base::CommandLine::ForCurrentProcess());
  if (!endpoint.is_valid()) {
    Shutdown(kErrorMojoConnectionFailure);
    return;
  }

  mojo::ScopedMessagePipeHandle pipe =
      mojo::IncomingInvitation::AcceptIsolated(std::move(endpoint));
  if (!pipe->is_valid()) {
    Shutdown(kErrorMojoConnectionFailure);
    return;
  }

  mojo::MakeSelfOwnedReceiver(
      std::make_unique<UnzipperImplWithShutdown>(
          base::BindOnce(&AppUnzipWorker::Shutdown, this, kErrorOk)),
      mojo::PendingReceiver<unzip::mojom::Unzipper>(std::move(pipe)));
}

scoped_refptr<App> MakeAppUnzipWorker() {
  return base::MakeRefCounted<AppUnzipWorker>();
}

}  // namespace updater