ryanandClaude
b6769abbe9
- Move Elixir code to server/ subdirectory for monorepo structure - Update flake.nix to provide packages and apps outputs for nix run support - Update nix/package.nix to accept src parameter instead of fetchgit - Add NixOS module export for easy consumption Now supports: nix run, nix build, and nix develop from git repo 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
115 lines
3.2 KiB
Elixir
115 lines
3.2 KiB
Elixir
defmodule Tortoise.Package.Publish do
|
|
@moduledoc false
|
|
|
|
@opcode 3
|
|
|
|
alias Tortoise.Package
|
|
|
|
@type t :: %__MODULE__{
|
|
__META__: Package.Meta.t(),
|
|
topic: Tortoise.topic() | nil,
|
|
qos: Tortoise.qos(),
|
|
payload: Tortoise.payload(),
|
|
identifier: Tortoise.package_identifier(),
|
|
dup: boolean(),
|
|
retain: boolean()
|
|
}
|
|
defstruct __META__: %Package.Meta{opcode: @opcode, flags: 0},
|
|
identifier: nil,
|
|
topic: nil,
|
|
payload: nil,
|
|
qos: 0,
|
|
dup: false,
|
|
retain: false
|
|
|
|
@spec decode(binary()) :: t
|
|
def decode(<<@opcode::4, 0::1, 0::2, retain::1, length_prefixed_payload::binary>>) do
|
|
payload = drop_length_prefix(length_prefixed_payload)
|
|
{topic, payload} = decode_message(payload)
|
|
|
|
%__MODULE__{
|
|
qos: 0,
|
|
identifier: nil,
|
|
dup: false,
|
|
retain: retain == 1,
|
|
topic: topic,
|
|
payload: payload
|
|
}
|
|
end
|
|
|
|
def decode(
|
|
<<@opcode::4, dup::1, qos::integer-size(2), retain::1, length_prefixed_payload::binary>>
|
|
) do
|
|
payload = drop_length_prefix(length_prefixed_payload)
|
|
{topic, identifier, payload} = decode_message_with_id(payload)
|
|
|
|
%__MODULE__{
|
|
qos: qos,
|
|
identifier: identifier,
|
|
dup: dup == 1,
|
|
retain: retain == 1,
|
|
topic: topic,
|
|
payload: payload
|
|
}
|
|
end
|
|
|
|
defp drop_length_prefix(payload) do
|
|
case payload do
|
|
<<0::1, _::7, r::binary>> -> r
|
|
<<1::1, _::7, 0::1, _::7, r::binary>> -> r
|
|
<<1::1, _::7, 1::1, _::7, 0::1, _::7, r::binary>> -> r
|
|
<<1::1, _::7, 1::1, _::7, 1::1, _::7, 0::1, _::7, r::binary>> -> r
|
|
end
|
|
end
|
|
|
|
defp decode_message(<<topic_length::big-integer-size(16), msg::binary>>) do
|
|
<<topic::binary-size(topic_length), payload::binary>> = msg
|
|
{topic, nullify(payload)}
|
|
end
|
|
|
|
defp decode_message_with_id(<<topic_length::big-integer-size(16), msg::binary>>) do
|
|
<<topic::binary-size(topic_length), identifier::big-integer-size(16), payload::binary>> = msg
|
|
{topic, identifier, nullify(payload)}
|
|
end
|
|
|
|
defp nullify(""), do: nil
|
|
defp nullify(payload), do: payload
|
|
|
|
# Protocols ----------------------------------------------------------
|
|
defimpl Tortoise.Encodable do
|
|
def encode(%Package.Publish{identifier: nil, qos: 0} = t) do
|
|
[
|
|
Package.Meta.encode(%{t.__META__ | flags: encode_flags(t)}),
|
|
Package.variable_length_encode([
|
|
Package.length_encode(t.topic),
|
|
encode_payload(t)
|
|
])
|
|
]
|
|
end
|
|
|
|
def encode(%Package.Publish{identifier: identifier, qos: qos} = t)
|
|
when identifier in 0x0001..0xFFFF and qos in 1..2 do
|
|
[
|
|
Package.Meta.encode(%{t.__META__ | flags: encode_flags(t)}),
|
|
Package.variable_length_encode([
|
|
Package.length_encode(t.topic),
|
|
<<identifier::big-integer-size(16)>>,
|
|
encode_payload(t)
|
|
])
|
|
]
|
|
end
|
|
|
|
defp encode_flags(%{dup: dup, qos: qos, retain: retain}) do
|
|
<<flags::4>> = <<flag(dup)::1, qos::integer-size(2), flag(retain)::1>>
|
|
flags
|
|
end
|
|
|
|
defp encode_payload(%{payload: nil}), do: ""
|
|
|
|
defp encode_payload(%{payload: payload}), do: payload
|
|
|
|
defp flag(f) when f in [0, nil, false], do: 0
|
|
defp flag(_), do: 1
|
|
end
|
|
end
|