Skip to content

Commit

Permalink
background-view: Try to match using pid before checking app_id
Browse files Browse the repository at this point in the history 
This way, most apps should work without having to set an app_id manually.
  • Loading branch information
@soreau
soreau committed Oct 7, 2023
1 parent c235eed commit 540a020
Showing 1 changed file with 128 additions and 17 deletions.
145 changes: 128 additions & 17 deletions src/background-view.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@
#include <wayfire/view-helpers.hpp>
#include <wayfire/config.h>


class wayfire_bgview_set_pointer_interaction : public wf::pointer_interaction_t
{
public:
Expand Down Expand Up @@ -137,7 +138,12 @@ class wayfire_background_view_root_node : public wf::scene::translation_node_t
class unmappable_view_t : public virtual wf::view_interface_t
{
public:
virtual void bg_view_unmap() = 0;
virtual void bg_view_unmap()
{}
wlr_surface *get_keyboard_focus_surface()
{
return nullptr;
}

wf::wl_listener_wrapper on_unmap;
std::shared_ptr<wayfire_background_view_root_node> root_node;
Expand Down Expand Up @@ -172,6 +178,10 @@ class wayfire_background_view_xdg : public wf::xdg_toplevel_view_base_t, public
public:
wayfire_background_view_xdg(wlr_xdg_toplevel *toplevel) : xdg_toplevel_view_base_t(toplevel, true)
{}
wlr_surface *get_keyboard_focus_surface()
{
return nullptr;
}

void bg_view_unmap() override
{
Expand All @@ -191,6 +201,11 @@ class wayfire_background_view_xwl : public wf::xwayland_view_base_t, public unma
this->kb_focus_enabled = !inhibit_input;
}

wlr_surface *get_keyboard_focus_surface()
{
return nullptr;
}

void map()
{
do_map(xw->surface, true);
Expand All @@ -203,6 +218,12 @@ class wayfire_background_view_xwl : public wf::xwayland_view_base_t, public unma
};
#endif

struct background_view
{
std::shared_ptr<unmappable_view_t> view;
pid_t pid;
};

class wayfire_background_view : public wf::plugin_interface_t
{
/* The command option should be set to a client like mpv, projectM or
Expand All @@ -222,7 +243,7 @@ class wayfire_background_view : public wf::plugin_interface_t
wf::option_wrapper_t<std::string> app_id{"background-view/app_id"};

// List of all background views assigned to an output
std::map<wf::output_t*, std::shared_ptr<unmappable_view_t>> views;
std::map<wf::output_t*, background_view> views;

wf::wl_listener_wrapper on_new_inhibitor;
wf::wl_idle_call idle_cleanup_inhibitors;
Expand All @@ -239,13 +260,53 @@ class wayfire_background_view : public wf::plugin_interface_t
on_new_inhibitor.connect(&wf::get_core().protocols.idle_inhibit->events.new_inhibitor);
}

/* Borrowed from sway */
pid_t get_parent_pid(pid_t child)
{
pid_t parent = -1;
char file_name[100];
char *buffer = NULL;
const char *sep = " ";
FILE *stat = NULL;
size_t buf_size = 0;

snprintf(file_name, sizeof(file_name), "/proc/%d/stat", child);

if ((stat = fopen(file_name, "r")))
{
if (getline(&buffer, &buf_size, stat) != -1)
{
strtok(buffer, sep); // pid
strtok(NULL, sep); // executable name
strtok(NULL, sep); // state
char *token = strtok(NULL, sep); // parent pid
parent = strtol(token, NULL, 10);
}

free(buffer);
fclose(stat);
}

if (parent)
{
return (parent == child) ? -1 : parent;
}

return -1;
}

void close_all_views()
{
for (auto& [output, view] : views)
for (auto& [output, bg_view] : views)
{
view->close();
view->on_unmap.disconnect();
view->bg_view_unmap();
if (!bg_view.view)
{
continue;
}

bg_view.view->close();
bg_view.view->on_unmap.disconnect();
bg_view.view->bg_view_unmap();
}

views.clear();
Expand All @@ -259,9 +320,9 @@ class wayfire_background_view : public wf::plugin_interface_t
return;
}

for (size_t i = 0; i < wf::get_core().output_layout->get_outputs().size(); i++)
for (auto & o : wf::get_core().output_layout->get_outputs())
{
wf::get_core().run(std::string(command) + add_arg_if_not_empty(file));
views[o].pid = wf::get_core().run(std::string(command) + add_arg_if_not_empty(file));
}
};

Expand Down Expand Up @@ -294,36 +355,85 @@ class wayfire_background_view : public wf::plugin_interface_t

new_view->on_unmap.set_callback([this, o] (auto)
{
views[o]->bg_view_unmap();
views[o].view->bg_view_unmap();
views.erase(o);
});
views[o] = new_view;
views[o].view = new_view;

// Remove any idle inhibitors which were already set
remove_idle_inhibitors();
}

wf::signal::connection_t<wf::view_pre_map_signal> on_view_pre_map = [=] (wf::view_pre_map_signal *ev)
{
auto view = wf::toplevel_cast(ev->view);
pid_t pid = 0;
pid_t x_pid = 0;
pid_t wl_pid = 0;
auto view = ev->view;
if (!view)
{
return;
}

wlr_xwayland_surface *xwayland_surface = nullptr;
wlr_surface *wlr_surface = ev->surface;
#if WF_HAS_XWAYLAND
if (wlr_surface && wlr_surface_is_xwayland_surface(wlr_surface))
{
xwayland_surface = wlr_xwayland_surface_from_wlr_surface(wlr_surface);
}

#endif
if (xwayland_surface)
{
#if WF_HAS_XWAYLAND
/* Get pid for xwayland view */
x_pid = xwayland_surface->pid;
#endif
} else if (ev->surface)
{
/* Get pid for native view */
wl_client_get_credentials(wl_resource_get_client(ev->surface->resource), &wl_pid, 0, 0);
}

for (auto& o : wf::get_core().output_layout->get_outputs())
{
if (views.count(o))
if (views[o].view)
{
continue;
}

/* Try to match application identifier */
/* First try to match pid */
if ((views[o].pid == wl_pid) || (views[o].pid == x_pid))
{
set_view_for_output(wf::toplevel_cast(view), ev->surface, o);
ev->override_implementation = true;
return;
}

if (xwayland_surface)
{
pid = get_parent_pid(x_pid);
} else
{
pid = get_parent_pid(wl_pid);
}

do {
if (views[o].pid == pid)
{
set_view_for_output(wf::toplevel_cast(view), ev->surface, o);
ev->override_implementation = true;
return;
}
} while ((pid = get_parent_pid(pid)) != -1);

/* Next, try to match application identifier */
if (std::string(app_id) == view->get_app_id())
{
set_view_for_output(view, ev->surface, o);
set_view_for_output(wf::toplevel_cast(view), ev->surface, o);
ev->override_implementation = true;
break;
return;
}
}
};
Expand All @@ -348,9 +458,9 @@ class wayfire_background_view : public wf::plugin_interface_t

wl_list_for_each(inhibitor, &mgr->inhibitors, link)
{
for (auto& [_, view] : views)
for (auto& [_, bg_view] : views)
{
if (inhibitor->surface == view->get_wlr_surface())
if (bg_view.view && (inhibitor->surface == bg_view.view->get_wlr_surface()))
{
// Tell core that the inhibitor was destroyed. It was not really destroyed, but core
// will adjust its internal state as if the inhibitor wasn't there.
Expand All @@ -365,6 +475,7 @@ class wayfire_background_view : public wf::plugin_interface_t
void fini() override
{
close_all_views();
wf::get_core().disconnect(&on_view_pre_map);
}
};

Expand Down

0 comments on commit 540a020

Please sign in to comment.