mirror of
https://github.com/zap-proto/erlang.git
synced 2026-10-30 15:05:14 +00:00
Andreas Stenius
cdf0f53e20
also keep a single format for the result term
when reading union fields as: {TagName, Value}
where Value may be 'void' for tags of type :Void.
127 lines
3.9 KiB
Bash
Executable File
127 lines
3.9 KiB
Bash
Executable File
#!/usr/bin/env escript
|
|
%% -*- mode: erlang -*-
|
|
|
|
main(Args) ->
|
|
try process(Args)
|
|
catch
|
|
C:E ->
|
|
io:format(standard_error,
|
|
"~s ~p: ~s:~p~n"
|
|
"~p~n",
|
|
[escript:script_name(), Args, C, E,
|
|
erlang:get_stacktrace()]),
|
|
halt(2)
|
|
end.
|
|
|
|
process(["read"|Args]) ->
|
|
read(Args);
|
|
process(["write"|Args]) ->
|
|
write(Args);
|
|
process(_) ->
|
|
io:format(standard_error,
|
|
"Usage: ~s {read [filename] | write}~n",
|
|
[escript:script_name()]),
|
|
halt(1).
|
|
|
|
read([FileName]) ->
|
|
{ok, Data} = file:read_file(FileName),
|
|
dump_message(Data);
|
|
read([]) ->
|
|
io:format(standard_error,
|
|
"Reading message from stdin~n"
|
|
" (note, this will likely fail)~n"
|
|
" (if you are on windows, )~n"
|
|
" (see README. )~n~n",
|
|
[]),
|
|
dump_message(read_stdin()).
|
|
|
|
write([]) ->
|
|
{ok, Root} = ecapnp:set_root(addressbook_capnp:'AddressBook'()),
|
|
[Alice, Bob, Steve] = ecapnp:set(people, 3, Root),
|
|
[AlicePhone] = ecapnp:set(phones, 1, Alice),
|
|
[BobPhone1, BobPhone2] = ecapnp:set(phones, 2, Bob),
|
|
[ecapnp:set(Field, Value, Obj)
|
|
|| {Obj, FieldValues} <-
|
|
[{Alice,
|
|
[{id, 123},
|
|
{name, <<"Alice">>},
|
|
{email, <<"[email protected]">>},
|
|
{employment, {school, <<"MIT">>}}
|
|
]},
|
|
{AlicePhone,
|
|
[{number, <<"555-1212">>},
|
|
{type, mobile}]},
|
|
{Bob,
|
|
[{id, 456},
|
|
{name, <<"Bob">>},
|
|
{email, <<"[email protected]">>},
|
|
{employment, unemployed}
|
|
]},
|
|
{BobPhone1,
|
|
[{number, <<"555-4567">>},
|
|
{type, home}]},
|
|
{BobPhone2,
|
|
[{number, <<"555-7654">>},
|
|
{type, work}]},
|
|
{Steve,
|
|
[{id, 123456},
|
|
{name, <<"Steve">>},
|
|
{email, <<"[email protected]">>},
|
|
{employment, selfEmployed},
|
|
{phones, 3},
|
|
{phones, {0, {number, <<"555-1234">>}}},
|
|
{phones, {0, {type, home}}},
|
|
{phones, {1, {number, <<"555-4321">>}}},
|
|
{phones, {1, {type, work}}},
|
|
{phones, {2, {number, <<"070-5555">>}}},
|
|
{phones, {2, {type, mobile}}}
|
|
]}
|
|
],
|
|
{Field, Value} <- FieldValues],
|
|
|
|
%% Get message data and pack it
|
|
Data1 = ecapnp_serialize:pack(
|
|
ecapnp_message:write(Root)),
|
|
io:setopts([{encoding, unicode}]),
|
|
io:put_chars(Data1).
|
|
|
|
|
|
dump_message(Data) ->
|
|
%% unpack and read message data
|
|
{ok, Message, <<>>} = ecapnp_message:read(
|
|
ecapnp_serialize:unpack(Data)),
|
|
{ok, Root} = ecapnp:get_root(
|
|
addressbook_capnp:'AddressBook'(), Message),
|
|
People = ecapnp:get(people, Root),
|
|
[dump_person(Person) || Person <- People].
|
|
|
|
dump_person(Person) ->
|
|
io:format("#~p ", [ecapnp:get(id, Person)]),
|
|
io:format("~s: ~s~n", [ecapnp:get(name, Person),
|
|
ecapnp:get(email, Person)]),
|
|
Phones = ecapnp:get(phones, Person),
|
|
[io:format(" ~s phone: ~s~n", [ecapnp:get(type, P),
|
|
ecapnp:get(number, P)])
|
|
|| P <- Phones],
|
|
case ecapnp:get(employment, Person) of
|
|
{unemployed, void} -> io:format(" unemployed~n");
|
|
{employer, Employer} ->
|
|
io:format(" employer: ~s~n", [Employer]);
|
|
{school, School} ->
|
|
io:format(" student at: ~s~n", [School]);
|
|
{selfEmployed, void} -> io:format(" self-employed~n")
|
|
end.
|
|
|
|
read_stdin() ->
|
|
read_stdin([]).
|
|
|
|
read_stdin(Acc)
|
|
when is_list(Acc) ->
|
|
read_stdin(file:read(standard_io, 1024), Acc).
|
|
|
|
read_stdin(eof, Acc) ->
|
|
list_to_binary(
|
|
lists:reverse(Acc));
|
|
read_stdin({ok, Data}, Acc) ->
|
|
read_stdin([Data|Acc]).
|