ladybird/Userland/Utilities/wallpaper.cpp
Lucas CHOLLET 1a97382305 LibGUI: Make Application's construction fallible
The pattern to construct `Application` was to use the `try_create`
method from the `C_OBJECT` macro. While being safe from an OOM
perspective, this method doesn't propagate errors from the constructor.
This patch make `Application` use the `C_OBJECT_ABSTRACT` and manually
define a `create` method that can bubble up errors from the
construction stage.

This commit also removes the ability to use `argc` and `argv` to
create an `Application`, only `Main`'s `Arguments` can be used.

From a user point of view, the patch renames `try_create` => `create`,
hence the huge number of modified files.
2023-05-05 16:41:21 +01:00

83 lines
3.3 KiB
C++

/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
* Copyright (c) 2022, James Puleo <james@jame.xyz>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/DeprecatedString.h>
#include <AK/Random.h>
#include <AK/StringBuilder.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/DirIterator.h>
#include <LibCore/System.h>
#include <LibGUI/Application.h>
#include <LibGUI/Desktop.h>
#include <LibMain/Main.h>
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
TRY(Core::System::pledge("stdio rpath unix sendfd recvfd"));
bool show_all = false;
bool show_current = false;
bool set_random = false;
DeprecatedString path;
Core::ArgsParser args_parser;
args_parser.add_option(show_all, "Show all wallpapers", "show-all", 'a');
args_parser.add_option(show_current, "Show current wallpaper", "show-current", 'c');
args_parser.add_option(set_random, "Set random wallpaper", "set-random", 'r');
args_parser.add_positional_argument(path, "Wallpaper to set", "path", Core::ArgsParser::Required::No);
args_parser.parse(arguments);
auto app = TRY(GUI::Application::create(arguments));
TRY(Core::System::pledge("stdio rpath unix sendfd"));
if (show_all) {
Core::DirIterator wallpapers_directory_iterator("/res/wallpapers", Core::DirIterator::SkipDots);
if (wallpapers_directory_iterator.has_error())
return Error::from_string_literal("Unable to iterate /res/wallpapers directory");
while (wallpapers_directory_iterator.has_next()) {
auto name = wallpapers_directory_iterator.next_path();
outln("{}", name);
}
} else if (show_current) {
auto current_wallpaper_path = GUI::Desktop::the().wallpaper_path();
outln("{}", current_wallpaper_path);
} else if (set_random) {
Core::DirIterator wallpapers_directory_iterator("/res/wallpapers", Core::DirIterator::SkipDots);
if (wallpapers_directory_iterator.has_error())
return Error::from_string_literal("Unable to iterate /res/wallpapers directory");
Vector<DeprecatedString> wallpaper_paths;
auto current_wallpaper_path = GUI::Desktop::the().wallpaper_path();
while (wallpapers_directory_iterator.has_next()) {
auto next_full_path = wallpapers_directory_iterator.next_full_path();
if (next_full_path != current_wallpaper_path)
wallpaper_paths.append(move(next_full_path));
}
if (wallpaper_paths.is_empty())
return Error::from_string_literal("No wallpapers found");
auto& chosen_wallpaper_path = wallpaper_paths.at(get_random_uniform(wallpaper_paths.size()));
auto chosen_wallpaper_bitmap = TRY(Gfx::Bitmap::load_from_file(chosen_wallpaper_path));
if (!GUI::Desktop::the().set_wallpaper(chosen_wallpaper_bitmap, chosen_wallpaper_path))
return Error::from_string_literal("Failed to set wallpaper");
outln("Set wallpaper to {}", chosen_wallpaper_path);
} else {
if (path.is_null())
return Error::from_string_literal("Must provide a path to a wallpaper");
auto wallpaper_bitmap = TRY(Gfx::Bitmap::load_from_file(path));
if (!GUI::Desktop::the().set_wallpaper(wallpaper_bitmap, path))
return Error::from_string_literal("Failed to set wallpaper");
}
return 0;
}