1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
-module(rst_load).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-include("../../../include/yaws_api.hrl").
-record
(
input,
{
player_id :: binary(),
session_token :: binary()
}
).
-record
(
query_state,
{
player :: shr_player:type(),
inventory :: shr_inventory:type(),
roster :: rst_roster:type()
}
).
-type input() :: #input{}.
-type query_state() :: #query_state{}.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% EXPORTS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-export([out/1]).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-spec parse_input (binary()) -> input().
parse_input (Req) ->
JSONReqMap = jiffy:decode(Req, [return_maps]),
PlayerID = maps:get(<<"pid">>, JSONReqMap),
SessionToken = maps:get(<<"stk">>, JSONReqMap),
#input
{
player_id = PlayerID,
session_token = SessionToken
}.
-spec authenticate_user (input()) -> ({'ok', shr_player:type()} | 'error').
authenticate_user (Input) ->
PlayerID = Input#input.player_id,
SessionToken = Input#input.session_token,
Player = shr_timed_cache:fetch(player_db, ataxia_security:any(), PlayerID),
case shr_security:credentials_match(SessionToken, Player) of
true -> {ok, Player};
_ -> error
end.
-spec fetch_data (shr_player:type(), input()) -> query_state().
fetch_data (Player, Input) ->
PlayerID = Input#input.player_id,
RosterID = shr_player:get_roster_id(Player),
% InventoryID = shr_player:get_inventory_id(Player),
Roster =
shr_timed_cache:fetch
(
roster_db,
ataxia_security:user_from_id(PlayerID),
RosterID
),
%% TODO
%% Inventory = shr_timed_cache:fetch(inventory_db, PlayerID, InventoryID),
io:format("[W] Using shim inventory.~n"),
Inventory = shr_inventory:new(PlayerID),
#query_state
{
player = Player,
roster = Roster,
inventory = Inventory
}.
-spec generate_reply(query_state()) -> binary().
generate_reply (QueryState) ->
Roster = QueryState#query_state.roster,
Inventory = QueryState#query_state.inventory,
RosterCharacters = rst_roster:get_characters(Roster),
SetInventory = shr_set_inventory:generate(Inventory),
EncodedRoster =
lists:map
(
fun ({IX, Char}) ->
rst_add_char:generate(IX, Char)
end,
orddict:to_list(RosterCharacters)
),
Output = jiffy:encode([SetInventory|EncodedRoster]),
Output.
-spec handle (binary()) -> binary().
handle (Req) ->
Input = parse_input(Req),
case authenticate_user(Input) of
{ok, Player} ->
shr_security:lock_queries(Input#input.player_id),
QueryState = fetch_data(Player, Input),
shr_security:unlock_queries(Input#input.player_id),
generate_reply(QueryState);
error -> jiffy:encode([shr_disconnected:generate()])
end.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
out(A) ->
{
content,
"application/json; charset=UTF-8",
handle(A#arg.clidata)
}.
|
