2020-05-19 11:25:16 +00:00
|
|
|
// This file is part of river, a dynamic tiling wayland compositor.
|
|
|
|
//
|
|
|
|
// Copyright 2020 Isaac Freund
|
|
|
|
//
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
// GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
const Self = @This();
|
|
|
|
|
|
|
|
const std = @import("std");
|
|
|
|
|
|
|
|
const c = @import("c.zig");
|
2020-05-26 20:55:07 +00:00
|
|
|
const command = @import("command.zig");
|
2020-05-19 11:25:16 +00:00
|
|
|
|
|
|
|
const Log = @import("log.zig").Log;
|
2020-06-15 20:24:54 +00:00
|
|
|
const Seat = @import("Seat.zig");
|
2020-05-19 11:25:16 +00:00
|
|
|
const Server = @import("Server.zig");
|
|
|
|
|
|
|
|
const protocol_version = 1;
|
|
|
|
|
2020-05-24 18:58:39 +00:00
|
|
|
const implementation = c.struct_zriver_control_v1_interface{
|
2020-06-15 20:24:54 +00:00
|
|
|
.destroy = destroy,
|
|
|
|
.add_argument = addArgument,
|
2020-05-19 11:25:16 +00:00
|
|
|
.run_command = runCommand,
|
|
|
|
};
|
|
|
|
|
|
|
|
server: *Server,
|
|
|
|
wl_global: *c.wl_global,
|
|
|
|
|
2020-06-15 20:24:54 +00:00
|
|
|
args_map: std.AutoHashMap(u32, std.ArrayList([]const u8)),
|
|
|
|
|
2020-05-19 11:25:16 +00:00
|
|
|
listen_display_destroy: c.wl_listener,
|
|
|
|
|
|
|
|
pub fn init(self: *Self, server: *Server) !void {
|
2020-05-19 16:22:22 +00:00
|
|
|
self.server = server;
|
2020-05-19 11:25:16 +00:00
|
|
|
self.wl_global = c.wl_global_create(
|
|
|
|
server.wl_display,
|
2020-05-24 18:58:39 +00:00
|
|
|
&c.zriver_control_v1_interface,
|
2020-05-19 11:25:16 +00:00
|
|
|
protocol_version,
|
|
|
|
self,
|
|
|
|
bind,
|
2020-06-04 14:56:58 +00:00
|
|
|
) orelse return error.CantCreateWlGlobal;
|
2020-05-19 11:25:16 +00:00
|
|
|
|
2020-06-15 20:24:54 +00:00
|
|
|
self.args_map = std.AutoHashMap(u32, std.ArrayList([]const u8)).init(server.allocator);
|
|
|
|
|
2020-05-19 11:25:16 +00:00
|
|
|
self.listen_display_destroy.notify = handleDisplayDestroy;
|
|
|
|
c.wl_display_add_destroy_listener(server.wl_display, &self.listen_display_destroy);
|
|
|
|
}
|
|
|
|
|
|
|
|
fn handleDisplayDestroy(wl_listener: ?*c.wl_listener, data: ?*c_void) callconv(.C) void {
|
|
|
|
const self = @fieldParentPtr(Self, "listen_display_destroy", wl_listener.?);
|
|
|
|
c.wl_global_destroy(self.wl_global);
|
2020-06-15 20:24:54 +00:00
|
|
|
self.args_map.deinit();
|
2020-05-19 11:25:16 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Called when a client binds our global
|
|
|
|
fn bind(wl_client: ?*c.wl_client, data: ?*c_void, version: u32, id: u32) callconv(.C) void {
|
|
|
|
const self = @ptrCast(*Self, @alignCast(@alignOf(*Self), data));
|
|
|
|
const wl_resource = c.wl_resource_create(
|
|
|
|
wl_client,
|
2020-05-24 18:58:39 +00:00
|
|
|
&c.zriver_control_v1_interface,
|
2020-05-19 11:25:16 +00:00
|
|
|
@intCast(c_int, version),
|
|
|
|
id,
|
|
|
|
) orelse {
|
|
|
|
c.wl_client_post_no_memory(wl_client);
|
|
|
|
return;
|
|
|
|
};
|
2020-06-15 20:24:54 +00:00
|
|
|
self.args_map.putNoClobber(id, std.ArrayList([]const u8).init(self.server.allocator)) catch {
|
|
|
|
c.wl_resource_destroy(wl_resource);
|
|
|
|
c.wl_client_post_no_memory(wl_client);
|
|
|
|
return;
|
|
|
|
};
|
|
|
|
c.wl_resource_set_implementation(wl_resource, &implementation, self, handleResourceDestroy);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Remove the resource from the hash map and free all stored args
|
|
|
|
fn handleResourceDestroy(wl_resource: ?*c.wl_resource) callconv(.C) void {
|
|
|
|
const self = @ptrCast(*Self, @alignCast(@alignOf(*Self), c.wl_resource_get_user_data(wl_resource)));
|
|
|
|
const id = c.wl_resource_get_id(wl_resource);
|
|
|
|
const list = self.args_map.remove(id).?.value;
|
|
|
|
for (list.items) |arg| list.allocator.free(arg);
|
|
|
|
list.deinit();
|
|
|
|
}
|
|
|
|
|
|
|
|
fn destroy(wl_client: ?*c.wl_client, wl_resource: ?*c.wl_resource) callconv(.C) void {
|
|
|
|
c.wl_resource_destroy(wl_resource);
|
|
|
|
}
|
|
|
|
|
|
|
|
fn addArgument(wl_client: ?*c.wl_client, wl_resource: ?*c.wl_resource, arg: ?[*:0]const u8) callconv(.C) void {
|
|
|
|
const self = @ptrCast(*Self, @alignCast(@alignOf(*Self), c.wl_resource_get_user_data(wl_resource)));
|
|
|
|
const id = c.wl_resource_get_id(wl_resource);
|
|
|
|
const allocator = self.server.allocator;
|
|
|
|
|
|
|
|
const owned_slice = std.mem.dupe(allocator, u8, std.mem.span(arg.?)) catch {
|
|
|
|
c.wl_client_post_no_memory(wl_client);
|
|
|
|
return;
|
|
|
|
};
|
|
|
|
|
|
|
|
self.args_map.get(id).?.value.append(owned_slice) catch {
|
|
|
|
c.wl_client_post_no_memory(wl_client);
|
|
|
|
allocator.free(owned_slice);
|
|
|
|
return;
|
|
|
|
};
|
2020-05-19 11:25:16 +00:00
|
|
|
}
|
|
|
|
|
2020-05-24 13:18:57 +00:00
|
|
|
fn runCommand(
|
|
|
|
wl_client: ?*c.wl_client,
|
|
|
|
wl_resource: ?*c.wl_resource,
|
2020-06-15 20:24:54 +00:00
|
|
|
seat_wl_resource: ?*c.wl_resource,
|
2020-05-24 13:18:57 +00:00
|
|
|
callback_id: u32,
|
|
|
|
) callconv(.C) void {
|
2020-05-19 16:22:22 +00:00
|
|
|
const self = @ptrCast(*Self, @alignCast(@alignOf(*Self), c.wl_resource_get_user_data(wl_resource)));
|
2020-06-15 20:24:54 +00:00
|
|
|
// This can be null if the seat is inert, in which case we ignore the request
|
|
|
|
const wlr_seat_client = c.wlr_seat_client_from_resource(seat_wl_resource) orelse return;
|
|
|
|
const seat = @ptrCast(*Seat, @alignCast(@alignOf(*Seat), wlr_seat_client.*.seat.*.data));
|
2020-05-19 16:22:22 +00:00
|
|
|
const allocator = self.server.allocator;
|
|
|
|
|
2020-05-24 13:18:57 +00:00
|
|
|
const callback_resource = c.wl_resource_create(
|
|
|
|
wl_client,
|
|
|
|
&c.zriver_command_callback_v1_interface,
|
|
|
|
protocol_version,
|
|
|
|
callback_id,
|
|
|
|
) orelse {
|
|
|
|
c.wl_client_post_no_memory(wl_client);
|
|
|
|
return;
|
|
|
|
};
|
|
|
|
c.wl_resource_set_implementation(callback_resource, null, null, null);
|
|
|
|
|
2020-06-15 20:24:54 +00:00
|
|
|
const args = self.args_map.get(c.wl_resource_get_id(wl_resource)).?.value.items;
|
|
|
|
|
2020-05-26 20:55:07 +00:00
|
|
|
var failure_message: []const u8 = undefined;
|
2020-06-15 20:24:54 +00:00
|
|
|
command.run(allocator, seat, args, &failure_message) catch |err| {
|
2020-05-26 20:55:07 +00:00
|
|
|
if (err == command.Error.CommandFailed) {
|
2020-06-01 12:41:44 +00:00
|
|
|
defer allocator.free(failure_message);
|
|
|
|
const out = std.cstr.addNullByte(allocator, failure_message) catch {
|
|
|
|
c.zriver_command_callback_v1_send_failure(callback_resource, "out of memory");
|
|
|
|
return;
|
|
|
|
};
|
2020-05-26 20:55:07 +00:00
|
|
|
defer allocator.free(out);
|
|
|
|
c.zriver_command_callback_v1_send_failure(callback_resource, out);
|
|
|
|
} else {
|
|
|
|
c.zriver_command_callback_v1_send_failure(
|
|
|
|
callback_resource,
|
|
|
|
switch (err) {
|
|
|
|
command.Error.NoCommand => "no command given",
|
|
|
|
command.Error.UnknownCommand => "unknown command",
|
2020-06-09 15:28:48 +00:00
|
|
|
command.Error.UnknownOption => "unknown option",
|
2020-05-26 20:55:07 +00:00
|
|
|
command.Error.NotEnoughArguments => "not enough arguments",
|
|
|
|
command.Error.TooManyArguments => "too many arguments",
|
|
|
|
command.Error.Overflow => "value out of bounds",
|
|
|
|
command.Error.InvalidCharacter => "invalid character in argument",
|
|
|
|
command.Error.InvalidDirection => "invalid direction. Must be 'next' or 'previous'",
|
2020-06-13 11:09:43 +00:00
|
|
|
command.Error.InvalidRgba => "invalid color format, must be #RRGGBB or #RRGGBBAA",
|
2020-05-26 20:55:07 +00:00
|
|
|
command.Error.OutOfMemory => "out of memory",
|
|
|
|
command.Error.CommandFailed => unreachable,
|
|
|
|
},
|
|
|
|
);
|
|
|
|
}
|
2020-05-24 13:18:57 +00:00
|
|
|
return;
|
|
|
|
};
|
2020-06-15 20:24:54 +00:00
|
|
|
c.zriver_command_callback_v1_send_success(callback_resource, "");
|
2020-05-19 11:25:16 +00:00
|
|
|
}
|