presence.nvim/lua/presence/discord.lua

local Discord = {}

Discord.opcodes = {
	auth = 0,
	frame = 1,
	closed = 2,
}

-- Discord RPC Subscription events
-- https://discord.com/developers/docs/topics/rpc#commands-and-events-rpc-events
-- Ready: https://discord.com/developers/docs/topics/rpc#ready
-- Error: https://discord.com/developers/docs/topics/rpc#error
Discord.events = {
	READY = "READY",
	ERROR = "ERROR",
}

local struct = require("deps.struct")

-- Initialize a new Discord RPC client
function Discord:init(options)
	self.log = options.logger
	self.client_id = options.client_id
	self.ipc_socket = options.ipc_socket

	self.pipe = vim.loop.new_pipe(false)

	return self
end

-- Connect to the local Discord RPC socket
-- TODO Might need to check for pipes ranging from discord-ipc-0 to discord-ipc-9:
-- https://github.com/discord/discord-rpc/blob/master/documentation/hard-mode.md#notes
function Discord:connect(on_connect)
	if self.pipe:is_closing() then
		self.pipe = vim.loop.new_pipe(false)
	end

	self.pipe:connect(self.ipc_socket, on_connect)
end

function Discord:is_connected()
	return self.pipe:is_active()
end

-- Disconnect from the local Discord RPC socket
function Discord:disconnect(on_close)
	self.pipe:shutdown()
	if not self.pipe:is_closing() then
		self.pipe:close(on_close)
	end
end

-- Make a remote procedure call to Discord
-- Callback argument in format: on_response(error[, response_table])
function Discord:call(opcode, payload, on_response)
	self.encode_json(payload, function(success, body)
		if not success then
			self.log:warn(string.format("Failed to encode payload: %s", vim.inspect(body)))
			return
		end

		-- Start reading for the response
		self.pipe:read_start(function(...)
			self:read_message(payload.nonce, on_response, ...)
		end)

		-- Construct message denoting little endian, auth opcode, msg length
		local message = struct.pack("<ii", opcode, #body) .. body

		-- Write the message to the pipe
		self.pipe:write(message, function(err)
			if err then
				local err_format = "Pipe write error - %s"
				local err_message = string.format(err_format, err)

				on_response(err_message)
			else
				self.log:debug("Wrote message to pipe")
			end
		end)
	end)
end

-- Read and handle socket messages
function Discord:read_message(nonce, on_response, err, chunk)
	if err then
		local err_format = "Pipe read error - %s"
		local err_message = string.format(err_format, err)

		on_response(err_message)
	elseif chunk then
		-- Strip header from the chunk
		local message = chunk:match("({.+)")
		local response_opcode = struct.unpack("<ii", chunk)

		self.decode_json(message, function(success, response)
			-- Check for a non-frame opcode in the response
			if response_opcode ~= self.opcodes.frame then
				local err_format = "Received unexpected opcode - %s (code %s)"
				local err_message = string.format(err_format, response.message, response.code)

				return on_response(err_message)
			end

			-- Unable to decode the response
			if not success then
				-- Indetermine state at this point, no choice but to simply warn on the parse failure
				-- but invoke empty response callback as request may still have succeeded
				self.log:warn(string.format("Failed to decode payload: %s", vim.inspect(message)))
				return on_response()
			end

			-- Check for an error event response
			if response.evt == self.events.ERROR then
				local data = response.data
				local err_format = "Received error event - %s (code %s)"
				local err_message = string.format(err_format, data.message, data.code)

				return on_response(err_message)
			end

			-- Check for a valid nonce value
			if response.nonce and response.nonce ~= vim.NIL and response.nonce ~= nonce then
				local err_format = "Received unexpected nonce - %s (expected %s)"
				local err_message = string.format(err_format, response.nonce, nonce)

				return on_response(err_message)
			end

			on_response(nil, response)
		end)
	else
		-- TODO: Handle when pipe is closed
		self.log:warn("Pipe was closed")
	end
end

-- Call to authorize the client connection with Discord
-- Callback argument in format: on_authorize(error[, response_table])
function Discord:authorize(on_authorize)
	local payload = {
		client_id = self.client_id,
		v = 1,
		nonce = "initial-ready",
	}

	self:call(self.opcodes.auth, payload, on_authorize)
end

-- Call to set the Neovim activity to Discord
function Discord:set_activity(activity, on_response)
	local payload = {
		cmd = "SET_ACTIVITY",
		nonce = self.generate_uuid(),
		args = {
			activity = activity,
			pid = vim.loop:os_getpid(),
		},
	}

	self:call(self.opcodes.frame, payload, on_response)
end

function Discord.generate_uuid(seed)
	local index = 0
	local template = "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx"

	local uuid = template:gsub("[xy]", function(char)
		-- Increment an index to seed per char
		index = index + 1
		math.randomseed((seed or os.clock()) / index)

		local n = char == "x" and math.random(0, 0xf) or math.random(8, 0xb)

		return string.format("%x", n)
	end)

	return uuid
end

function Discord.decode_json(t, on_done)
	vim.schedule(function()
		on_done(pcall(function()
			return vim.fn.json_decode(t)
		end))
	end)
end

function Discord.encode_json(t, on_done)
	vim.schedule(function()
		on_done(pcall(function()
			return vim.fn.json_encode(t)
		end))
	end)
end

return Discord