shell/platform/windows/uwptool_main.cc

// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <Windows.h>
#include <winrt/base.h>

#include <algorithm>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>

#include "flutter/fml/command_line.h"
#include "flutter/shell/platform/windows/string_conversion.h"
#include "flutter/shell/platform/windows/uwptool_utils.h"

namespace {

// Prints a list of installed UWP apps to stdout.
void PrintInstalledApps() {
  flutter::ApplicationStore app_store;
  for (const flutter::Application& app : app_store.GetInstalledApplications()) {
    std::wcout << app.GetPackageId() << std::endl;
  }
}

// Launches the app installed on the system whose Application User Model ID is
// prefixed with app_id, with the specified arguments list.
//
// Returns -1 if no matching app, or multiple matching apps are found, or if
// the app fails to launch. Otherwise, the process ID of the launched app is
// returned.
int LaunchApp(const std::wstring_view app_id, const std::wstring_view args) {
  flutter::Application app(app_id);
  DWORD process_id = app.Launch(args);
  if (process_id == -1) {
    std::wcerr << L"Failed to launch app " << app.GetPackageId() << std::endl;
    return 1;
  }
  return 0;
}

// Prints the command usage to stderr.
void PrintUsage() {
  std::cerr << "usage: uwptool COMMAND [APP_ID]" << std::endl;
  std::cerr << "commands:" << std::endl;
  std::cerr << "  listapps        list installed applications" << std::endl;
  std::cerr << "  launch          launch an application" << std::endl;
}

}  // namespace

int main(int argc, char** argv) {
  winrt::init_apartment();

  auto command_line = fml::CommandLineFromArgcArgv(argc, argv);
  if (command_line.positional_args().size() < 1) {
    PrintUsage();
    return 1;
  }

  const std::vector<std::string>& args = command_line.positional_args();
  std::string command = args[0];
  if (command == "listapps") {
    PrintInstalledApps();
    return 0;
  } else if (command == "launch") {
    if (command_line.positional_args().size() < 1) {
      PrintUsage();
      return 1;
    }

    // Get the package ID.
    std::string package_id = args[1];

    // Concatenate the remaining args, space-separated.
    std::ostringstream app_args;
    for (int i = 2; i < args.size(); ++i) {
      app_args << args[i];
      if (i < args.size() - 1) {
        app_args << " ";
      }
    }
    return LaunchApp(flutter::Utf16FromUtf8(package_id),
                     flutter::Utf16FromUtf8(app_args.str()));
  }

  std::cerr << "Unknown command: " << command << std::endl;
  PrintUsage();
  return 1;
}