lucky.erl 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785
  1. %%% -------------------------------------------------------------------
  2. %%% Author : Sergii Polkovnikov <serge.polkovnikov@gmail.com>
  3. %%% Description : The "Quick play" logic
  4. %%%
  5. %%% Created : Oct 16, 2012
  6. %%% -------------------------------------------------------------------
  7. %%% Terms explanation:
  8. %%% GameId - uniq identifier of the tournament. Type: integer().
  9. %%% PlayerId - registration number of a player in the tournament. Type: integer()
  10. %%% UserId - cross system identifier of a physical user. Type: binary() (or string()?).
  11. %%% TableId - uniq identifier of a table in the tournament. Used by the
  12. %%% tournament logic. Type: integer().
  13. %%% TableGlobalId - uniq identifier of a table in the system. Can be used
  14. %%% to refer to a table directly - without pointing to a tournament.
  15. %%% Type: integer()
  16. -module(lucky).
  17. -behaviour(gen_fsm).
  18. %% --------------------------------------------------------------------
  19. %% Include files
  20. %% --------------------------------------------------------------------
  21. -include_lib("server/include/basic_types.hrl").
  22. -include_lib("db/include/table.hrl").
  23. -include_lib("db/include/scoring.hrl").
  24. -include_lib("db/include/transaction.hrl").
  25. %% --------------------------------------------------------------------
  26. %% External exports
  27. -export([start/1, start/2, start_link/2, reg/2]).
  28. %% gen_fsm callbacks
  29. -export([init/1, handle_event/3, handle_sync_event/4, handle_info/3, terminate/3, code_change/4]).
  30. -export([table_message/3, client_message/2, client_request/2, client_request/3]).
  31. -record(state,
  32. {%% Static values
  33. game_id :: pos_integer(),
  34. game :: atom(),
  35. game_mode :: atom(),
  36. table_params :: proplists:proplist(),
  37. table_module :: atom(),
  38. bot_module :: atom(),
  39. seats_per_table :: integer(),
  40. game_name :: string(),
  41. mode :: normal | exclusive,
  42. %% Dynamic values
  43. players, %% The register of tournament players
  44. tables, %% The register of tournament tables
  45. seats, %% Stores relation between players and tables seats
  46. player_id_counter :: pos_integer(),
  47. table_id_counter :: pos_integer(),
  48. cr_tab_requests,
  49. reg_requests,
  50. tab_requests
  51. }).
  52. -record(player,
  53. {
  54. id :: pos_integer(),
  55. user_id,
  56. is_bot :: boolean()
  57. }).
  58. -record(table,
  59. {
  60. id :: pos_integer(),
  61. global_id :: pos_integer(),
  62. pid,
  63. relay :: {atom(), pid()}, %%{RelayMod, RelayPid}
  64. mon_ref,
  65. state :: initializing | ready | in_progress | finished,
  66. scoring_state,
  67. timer :: reference()
  68. }).
  69. -record(seat,
  70. {
  71. table :: pos_integer(),
  72. seat_num :: integer(),
  73. player_id :: undefined | pos_integer(),
  74. is_bot :: undefined | boolean(),
  75. registered_by_table :: undefined | boolean(),
  76. connected :: undefined | boolean()
  77. }).
  78. -define(STATE_INIT, state_init).
  79. -define(STATE_PROCESSING, state_processing).
  80. -define(TOURNAMENT_TYPE, lucky).
  81. -define(TABLE_STATE_INITIALIZING, initializing).
  82. -define(TABLE_STATE_READY, ready).
  83. -define(TABLE_STATE_IN_PROGRESS, in_progress).
  84. -define(TABLE_STATE_FINISHED, finished).
  85. -define(REST_TIMEOUT, 5000). %% Time between game finsh and start of new round
  86. %% ====================================================================
  87. %% External functions
  88. %% ====================================================================
  89. start([GameId, Params]) -> %% XXX WTF?
  90. gas:info(?MODULE," +++ START LUCKY"),
  91. start(GameId,Params).
  92. start(GameId, Params) ->
  93. gen_fsm:start(?MODULE, [GameId, Params, self()], []).
  94. start_link(GameId, Params) ->
  95. gen_fsm:start_link(?MODULE, [GameId, Params, self()], []).
  96. reg(Pid, User) ->
  97. client_request(Pid, {reg, User}, 10000).
  98. table_message(Pid, TableId, Message) ->
  99. gen_fsm:send_all_state_event(Pid, {table_message, TableId, Message}).
  100. client_message(Pid, Message) ->
  101. gen_fsm:send_all_state_event(Pid, {client_message, Message}).
  102. client_request(Pid, Message) ->
  103. client_request(Pid, Message, 5000).
  104. client_request(Pid, Message, Timeout) ->
  105. gen_fsm:sync_send_all_state_event(Pid, {client_request, Message}, Timeout).
  106. %% ====================================================================
  107. %% Server functions
  108. %% ====================================================================
  109. init([GameId, Params, _Manager]) ->
  110. SeatsPerTable = get_param(seats, Params),
  111. Game = get_param(game, Params),
  112. GameMode = get_param(game_mode, Params),
  113. GameName = get_param(game_name, Params),
  114. %%% XXX QuotaPerRound = get_param(quota_per_round, Params),
  115. TableParams = get_param(table_params, Params),
  116. TableModule = get_param(table_module, Params),
  117. BotModule = get_param(bot_module, Params),
  118. gas:info(?MODULE,"TRN_LUCKY <~p> All parameteres are read. Send the directive to start the game.", [GameId]),
  119. gen_fsm:send_all_state_event(self(), go),
  120. {ok, ?STATE_INIT,
  121. #state{game_id = GameId,
  122. game = Game,
  123. game_mode = GameMode,
  124. game_name = GameName,
  125. seats_per_table = SeatsPerTable,
  126. table_params = [{parent, {?MODULE, self()}} | TableParams],
  127. table_module = TableModule,
  128. bot_module = BotModule
  129. }}.
  130. %%===================================================================
  131. handle_event(go, ?STATE_INIT, #state{game_id = GameId, game = Game, game_mode = GameMode,
  132. game_name = GameName} = StateData) ->
  133. gas:info(?MODULE,"TRN_LUCKY <~p> Received the directive to start the game.", [GameId]),
  134. DeclRec = #game_table{id = GameId,
  135. game_type = Game,
  136. game_mode = GameMode,
  137. game_process = self(),
  138. game_module = ?MODULE,
  139. name = GameName,
  140. age_limit = 100,
  141. game_speed = undefined,
  142. feel_lucky = true,
  143. owner = undefined,
  144. creator = undefined,
  145. rounds = undefined,
  146. pointing_rules = [],
  147. pointing_rules_ex = [],
  148. users = []
  149. },
  150. gproc:reg({p,l,self()}, DeclRec),
  151. {next_state, ?STATE_PROCESSING,
  152. StateData#state{players = players_init(),
  153. tables = tables_init(),
  154. seats = seats_init(),
  155. player_id_counter = 1,
  156. table_id_counter = 1,
  157. cr_tab_requests = dict:new(),
  158. reg_requests = dict:new(),
  159. tab_requests = dict:new()
  160. }};
  161. handle_event({client_message, Message}, StateName, StateData) ->
  162. handle_client_message(Message, StateName, StateData);
  163. handle_event({table_message, TableId, Message}, StateName, StateData) ->
  164. handle_table_message(TableId, Message, StateName, StateData);
  165. handle_event(_Event, StateName, StateData) ->
  166. {next_state, StateName, StateData}.
  167. handle_sync_event({client_request, Request}, From, StateName, StateData) ->
  168. handle_client_request(Request, From, StateName, StateData);
  169. handle_sync_event(_Event, _From, StateName, StateData) ->
  170. Reply = ok,
  171. {reply, Reply, StateName, StateData}.
  172. %%===================================================================
  173. handle_info({'DOWN', MonRef, process, _Pid, _}, StateName,
  174. #state{game_id = GameId, tables = Tables,
  175. seats = Seats, players = Players} = StateData) ->
  176. case get_table_by_mon_ref(MonRef, Tables) of
  177. #table{id = TabId, timer = TRef} ->
  178. gas:info(?MODULE,"TRN_LUCKY <~p> Table <~p> is down. Cleaning up registeres.", [GameId, TabId]),
  179. case TRef == undefined of false -> erlang:cancel_timer(TRef); true -> skip end,
  180. PlayersIds =
  181. [PlayerId || #seat{player_id = PlayerId} <- find_seats_with_players_for_table_id(TabId, Seats)],
  182. NewTables = del_table(TabId, Tables),
  183. NewSeats = del_seats_by_table_id(TabId, Seats),
  184. NewPlayers = del_players(PlayersIds, Players),
  185. {next_state, StateName, StateData#state{tables = NewTables,
  186. seats = NewSeats,
  187. players = NewPlayers}};
  188. not_found ->
  189. {next_state, StateName, StateData}
  190. end;
  191. handle_info({rest_timeout, TableId}, StateName,
  192. #state{game_id = GameId, tables = Tables, table_module = TableModule} = StateData) ->
  193. gas:info(?MODULE,"TRN_LUCKY <~p> Time to start new round for table <~p>.", [GameId, TableId]),
  194. case get_table(TableId, Tables) of
  195. {ok, #table{pid = TablePid}} ->
  196. gas:info(?MODULE,"TRN_LUCKY <~p> Initiating new round at table <~p>.", [GameId, TableId]),
  197. NewTables = set_table_state(TableId, ?TABLE_STATE_IN_PROGRESS, Tables),
  198. send_to_table(TableModule, TablePid, start_round),
  199. {next_state, StateName, StateData#state{tables = NewTables}};
  200. error -> %% If no such table ignore the timeout
  201. gas:info(?MODULE,"TRN_LUCKY <~p> There is no table <~p>. Can't start new round for it.", [GameId, TableId]),
  202. {next_state, StateName, StateData}
  203. end;
  204. handle_info(Message, StateName, #state{game_id = GameId} = StateData) ->
  205. gas:info(?MODULE,"TRN_STANDALONE <~p> Unhandled message(info) received in state <~p>: ~p.",
  206. [GameId, StateName, Message]),
  207. {next_state, StateName, StateData}.
  208. %%===================================================================
  209. terminate(_Reason, _StateName, #state{game_id=GameId}=_StatData) ->
  210. gas:info(?MODULE,"TRN_LUCKY <~p> Shutting down at state: <~p>. Reason: ~p",
  211. [GameId, _StateName, _Reason]),
  212. ok.
  213. %%===================================================================
  214. code_change(_OldVsn, StateName, StateData, _Extra) ->
  215. {ok, StateName, StateData}.
  216. %% --------------------------------------------------------------------
  217. %%% Internal functions
  218. %% --------------------------------------------------------------------
  219. handle_client_message(_Msg, StateName, StateData) ->
  220. {next_state, StateName, StateData}.
  221. %%===================================================================
  222. %% handle_table_message(TableId, Message, StateName, StateData)
  223. handle_table_message(TableId, {player_connected, PlayerId},
  224. ?STATE_PROCESSING,
  225. #state{game_id = GameId, seats = Seats, seats_per_table = SeatsNum,
  226. tables = Tables, table_module = TableModule} = StateData)
  227. when is_integer(TableId), is_integer(PlayerId) ->
  228. gas:info(?MODULE,"TRN_LUCKY <~p> The player_connected notification received from "
  229. "table <~p>. PlayerId: <~p>", [GameId, TableId, PlayerId]),
  230. case find_seats_by_player_id(PlayerId, Seats) of
  231. [#seat{seat_num = SeatNum}] ->
  232. NewSeats = update_seat_connect_status(TableId, SeatNum, true, Seats),
  233. case fetch_table(TableId, Tables) of
  234. #table{state = ?TABLE_STATE_READY, pid = TabPid} ->
  235. case is_all_players_connected(TableId, SeatsNum, NewSeats) of
  236. true ->
  237. gas:info(?MODULE,"TRN_LUCKY <~p> All clients connected. Starting a game.",
  238. [GameId]),
  239. NewTables = set_table_state(TableId, ?TABLE_STATE_IN_PROGRESS, Tables),
  240. send_to_table(TableModule, TabPid, start_round),
  241. {next_state, ?STATE_PROCESSING, StateData#state{seats = NewSeats,
  242. tables = NewTables}};
  243. false ->
  244. {next_state, ?STATE_PROCESSING, StateData#state{seats = NewSeats}}
  245. end;
  246. _ ->
  247. {next_state, ?STATE_PROCESSING, StateData#state{seats = NewSeats}}
  248. end;
  249. [] -> %% Ignoring the message
  250. {next_state, ?STATE_PROCESSING, StateData}
  251. end;
  252. handle_table_message(TableId, {player_disconnected, PlayerId},
  253. ?STATE_PROCESSING, #state{game_id = GameId, seats = Seats} = StateData)
  254. when is_integer(TableId), is_integer(PlayerId) ->
  255. gas:info(?MODULE,"TRN_LUCKY <~p> The player_disconnected notification received from "
  256. "table <~p>. PlayerId: <~p>", [GameId, TableId, PlayerId]),
  257. case find_seats_by_player_id(PlayerId, Seats) of
  258. [#seat{seat_num = SeatNum, is_bot = true}] ->
  259. gas:info(?MODULE,"TRN_LUCKY <~p> Bot Replaces the Bot?", [GameId, TableId]),
  260. {next_state, ?STATE_PROCESSING, StateData};
  261. [#seat{seat_num = SeatNum, is_bot = IsBot}] ->
  262. case real_players_at_table(TableId, Seats) of
  263. 1 when not IsBot -> %% Last real player gone
  264. gas:info(?MODULE,"TRN_LUCKY <~p> Last real player gone from "
  265. "table <~p>. Closing the table.", [GameId, TableId]),
  266. unreg_player_and_eliminate_table(PlayerId, TableId, StateData);
  267. _ ->
  268. gas:info(?MODULE,"TRN_LUCKY <~p> Al least one real player is at table <~p>. "
  269. "Starting a bot to replace free seat.", [GameId, TableId]),
  270. replace_player_by_bot(PlayerId, TableId, SeatNum, StateData)
  271. end;
  272. [] -> %% Ignoring the message
  273. {next_state, ?STATE_PROCESSING, StateData}
  274. end;
  275. handle_table_message(TableId, {table_created, Relay}, ?STATE_PROCESSING,
  276. #state{game_id = GameId, tables = Tables, seats = Seats,
  277. cr_tab_requests = TCrRequests, table_module = TableModule,
  278. reg_requests = RegRequests} = StateData)
  279. when is_integer(TableId) ->
  280. gas:info(?MODULE,"TRN_LUCKY <~p> The <table_created> notification received from table: ~p.",
  281. [GameId, TableId]),
  282. TabInitPlayers = dict:fetch(TableId, TCrRequests),
  283. %% Update status of players
  284. TabSeats = find_seats_by_table_id(TableId, Seats),
  285. F = fun(#seat{player_id = PlayerId} = S, Acc) ->
  286. case lists:member(PlayerId, TabInitPlayers) of
  287. true -> store_seat(S#seat{registered_by_table = true}, Acc);
  288. false -> Acc
  289. end
  290. end,
  291. NewSeats = lists:foldl(F, Seats, TabSeats),
  292. %% Process delayed registration requests
  293. TablePid = get_table_pid(TableId, Tables),
  294. F2 = fun(PlayerId, Acc) ->
  295. case dict:find(PlayerId, Acc) of
  296. {ok, From} ->
  297. gen_fsm:reply(From, {ok, {PlayerId, Relay, {TableModule, TablePid}}}),
  298. dict:erase(PlayerId, Acc);
  299. error -> Acc
  300. end
  301. end,
  302. NewRegRequests = lists:foldl(F2, RegRequests, TabInitPlayers),
  303. NewTCrRequests = dict:erase(TableId, TCrRequests),
  304. NewTables = update_created_table(TableId, Relay, Tables),
  305. {next_state, ?STATE_PROCESSING, StateData#state{tables = NewTables,
  306. seats = NewSeats,
  307. cr_tab_requests = NewTCrRequests,
  308. reg_requests = NewRegRequests}};
  309. handle_table_message(TableId, {round_finished, NewScoringState, RoundScore, _TotalScore},
  310. ?STATE_PROCESSING,
  311. #state{game_id = GameId, tables = Tables, table_module = TableModule,
  312. game_mode = GameMode, game = GameType, players = Players} = StateData)
  313. when is_integer(TableId) ->
  314. gas:info(?MODULE,"TRN_LUCKY <~p> The <round_finished> received from table: ~p~nScore: ~p.",
  315. [GameId, TableId,RoundScore]),
  316. %% Add score per round
  317. UsersPoints = lists:flatten(
  318. [ begin
  319. case midict:find(PlayerId, Players) of
  320. {ok, #player{id = UserId, is_bot = false}} -> {UserId, Points};
  321. _Error -> gas:info(?MODULE, "get_player_info_by_user_id ~p", [_Error]), [] end
  322. end|| {PlayerId, Points} <- RoundScore]),
  323. add_points_to_accounts(UsersPoints, GameId, GameType, GameMode),
  324. #table{pid = TablePid} = Table = fetch_table(TableId, Tables),
  325. TRef = erlang:send_after(?REST_TIMEOUT, self(), {rest_timeout, TableId}),
  326. NewTable = Table#table{scoring_state = NewScoringState, state = ?TABLE_STATE_FINISHED, timer = TRef},
  327. NewTables = store_table(NewTable, Tables),
  328. send_to_table(TableModule, TablePid, show_round_result),
  329. {next_state, ?STATE_PROCESSING, StateData#state{tables = NewTables}};
  330. handle_table_message(TableId, {response, RequestId, Response},
  331. ?STATE_PROCESSING,
  332. #state{game_id = GameId, tab_requests = TabRequests} = StateData)
  333. when is_integer(TableId) ->
  334. NewTabRequests = dict:erase(RequestId, TabRequests),
  335. case dict:find(RequestId, TabRequests) of
  336. {ok, ReqContext} ->
  337. gas:info(?MODULE,"TRN_LUCKY <~p> A response received from table <~p>. "
  338. "RequestId: ~p. Request context: ~p. Response: ~p",
  339. [GameId, TableId, RequestId, ReqContext, Response]),
  340. handle_table_response(ReqContext, Response, ?STATE_PROCESSING,
  341. StateData#state{tab_requests = NewTabRequests});
  342. error ->
  343. gas:error(?MODULE,"TRN_LUCKY <~p> Table <~p> sent a response for unknown request. "
  344. "RequestId: ~p. Response", []),
  345. {next_state, ?STATE_PROCESSING, StateData#state{tab_requests = NewTabRequests}}
  346. end;
  347. handle_table_message(_TableId, _Event, StateName, StateData) ->
  348. {next_state, StateName, StateData}.
  349. %%===================================================================
  350. %% handle_table_response(RequestContext, Response, StateName, StateData)
  351. handle_table_response({replace_player, PlayerId, TableId, SeatNum}, ok = _Response,
  352. ?STATE_PROCESSING,
  353. #state{reg_requests = RegRequests, seats = Seats,
  354. tables = Tables, table_module = TableModule} = StateData) ->
  355. Seat = fetch_seat(TableId, SeatNum, Seats),
  356. NewSeats = store_seat(Seat#seat{registered_by_table = true}, Seats),
  357. %% Send response to a client for a delayed request
  358. NewRegRequests =
  359. case dict:find(PlayerId, RegRequests) of
  360. {ok, From} ->
  361. #table{relay = Relay, pid = TablePid} = fetch_table(TableId, Tables),
  362. gen_fsm:reply(From, {ok, {PlayerId, Relay, {TableModule, TablePid}}}),
  363. dict:erase(PlayerId, RegRequests);
  364. error -> RegRequests
  365. end,
  366. {next_state, ?STATE_PROCESSING, StateData#state{seats = NewSeats,
  367. reg_requests = NewRegRequests}}.
  368. %%===================================================================
  369. handle_client_request({reg, User}, From, ?STATE_PROCESSING,
  370. #state{game_id = GameId, reg_requests = RegRequests,
  371. seats = Seats, players=Players, tables = Tables,
  372. table_module = TableModule} = StateData) ->
  373. #'PlayerInfo'{id = UserId, robot = IsBot} = User,
  374. gas:info(?MODULE,"TRN_LUCKY <~p> The Register request received from user: ~p.", [GameId, UserId]),
  375. case IsBot of
  376. true -> %% Bots can't initiate a registration
  377. case get_player_id_by_user_id(UserId, Players) of
  378. {ok, PlayerId} -> %% Already registered. Send table requsites.
  379. [#seat{table = TableId, registered_by_table = RegByTable}] = find_seats_by_player_id(PlayerId, Seats),
  380. case RegByTable of
  381. false -> %% Store delayed request
  382. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  383. {next_state, ?STATE_PROCESSING, StateData#state{reg_requests = NewRegRequests}};
  384. _ ->
  385. #table{relay = Relay, pid = TPid} = fetch_table(TableId, Tables),
  386. {reply, {ok, {PlayerId, Relay, {TableModule, TPid}}}, ?STATE_PROCESSING, StateData}
  387. end;
  388. error -> %% Not registered
  389. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p is a bot. The user not registered. "
  390. "Rejecting registration.", [GameId, UserId]),
  391. {reply, {error, indy_bots_not_allowed}, ?STATE_PROCESSING, StateData}
  392. end;
  393. false -> %% Normal user
  394. IgnoredPlayers = [Id || #player{id = Id} <- midict:geti(UserId, user_id, Players)],
  395. gas:info(?MODULE,"TRN_LUCKY <~p> There are no table with free seats.", [GameId]),
  396. case find_bot_seat_without_players(Seats, IgnoredPlayers) of
  397. #seat{table = TabId, seat_num = SeatNum, player_id = OldPlayerId} ->
  398. gas:info(?MODULE,"TRN_LUCKY <~p> Found a seat with a bot. Replacing by the user. "
  399. "UserId:~p TabId: ~p SeatNum: ~p.", [GameId, UserId, TabId, SeatNum]),
  400. reg_player_with_replace(User, TabId, SeatNum, OldPlayerId, From, StateData);
  401. not_found ->
  402. gas:info(?MODULE,"TRN_LUCKY <~p> There are no seats with bots. "
  403. "Creating new table for user: ~p.", [GameId, UserId]),
  404. reg_player_at_new_table(User, From, StateData)
  405. end
  406. end;
  407. handle_client_request(Request, From, StateName, #state{game_id = GameId} = StateData) ->
  408. gas:info(?MODULE,"TRN_LUCKY <~p> Unhandled client request received from ~p in "
  409. "state <~p>: ~p.", [GameId, From, StateName, Request]),
  410. {reply, {error, unexpected_request}, StateName, StateData}.
  411. %%===================================================================
  412. reg_player_with_replace(UserInfo, TableId, SeatNum, OldPlayerId, From,
  413. #state{game_id = GameId, players = Players, tables = Tables,
  414. seats = Seats, player_id_counter = PlayerId,
  415. tab_requests = TabRequests, reg_requests = RegRequests,
  416. table_module = TableModule} = StateData) ->
  417. #'PlayerInfo'{id = UserId, robot = IsBot} = UserInfo,
  418. NewPlayers = del_player(OldPlayerId, Players),
  419. NewPlayers2 = reg_player(#player{id = PlayerId, user_id = UserId, is_bot = IsBot}, NewPlayers),
  420. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p registered as player <~p>.", [GameId, UserId, PlayerId]),
  421. NewSeats = assign_seat(TableId, SeatNum, PlayerId, IsBot, false, false, Seats),
  422. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p assigned to seat <~p> of table <~p>.", [GameId, UserId, SeatNum, TableId]),
  423. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  424. TablePid = get_table_pid(TableId, Tables),
  425. NewTabRequests = table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests),
  426. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  427. seats = NewSeats,
  428. player_id_counter = PlayerId + 1,
  429. tab_requests = NewTabRequests,
  430. reg_requests = NewRegRequests}}.
  431. reg_player_at_new_table(User, From,
  432. #state{game_id = GameId, players = Players,
  433. tables = Tables, seats = Seats, seats_per_table = SeatsNum,
  434. player_id_counter = PlayerIdCounter,
  435. table_id_counter = TableId, table_module = TableModule,
  436. bot_module = BotModule, table_params = TableParams,
  437. reg_requests = RegRequests, cr_tab_requests = TCrRequests
  438. } = StateData) ->
  439. #'PlayerInfo'{id = UserId, robot = IsBot} = User,
  440. RobotsInfo = spawn_bots(GameId, BotModule, SeatsNum - 1),
  441. gas:info(?MODULE,"TRN_LUCKY <~p> Bots for table <~p> are spawned.", [GameId, TableId]),
  442. F = fun(BotInfo, {PlId,SNum}) -> {{PlId, BotInfo, SNum, _Points = 0}, {PlId + 1, SNum + 1}} end,
  443. {RobotsRegData, {PlayerId, SeatNum}} = lists:mapfoldl(F, {PlayerIdCounter, 1}, RobotsInfo),
  444. TPlayers = [{PlayerId, User, SeatNum, 0} | RobotsRegData],
  445. TableParams2 = [{players, TPlayers}, {table_name, "I'm filling lucky"} | TableParams],
  446. {ok, TabPid} = spawn_table(TableModule, GameId, TableId, TableParams2),
  447. MonRef = erlang:monitor(process, TabPid),
  448. %% FIXME: Table global id should use a persistent counter
  449. NewTables = reg_table(TableId, TabPid, MonRef, 0, undefined, Tables),
  450. gas:info(?MODULE,"TRN_LUCKY <~p> New table created: ~p.", [GameId, TableId]),
  451. NewPlayers = reg_player(#player{id = PlayerId, user_id = UserId, is_bot = IsBot}, Players),
  452. F2 = fun({PlId, #'PlayerInfo'{id = UId, robot=Bot}, _SNum, _Points}, Acc) ->
  453. reg_player(#player{id = PlId, user_id = UId, is_bot = Bot}, Acc)
  454. end,
  455. NewPlayers2 = lists:foldl(F2, NewPlayers, RobotsRegData),
  456. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p registered as player <~p>.", [GameId, UserId, PlayerId]),
  457. NewSeats = assign_seat(TableId, SeatNum, PlayerId, IsBot, false, false, Seats),
  458. F3 = fun({PlId, _UserInfo, SNum, _Points}, Acc) ->
  459. assign_seat(TableId, SNum, PlId, true, false, false, Acc)
  460. end,
  461. NewSeats2 = lists:foldl(F3, NewSeats, RobotsRegData),
  462. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p assigned to seat <~p> of table <~p>.", [GameId, UserId, SeatNum, TableId]),
  463. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  464. PlayersIds = [PlayerId | [PlId || {PlId, _, _, _} <- RobotsRegData]],
  465. NewTCrRequests = dict:store(TableId, PlayersIds, TCrRequests),
  466. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  467. seats = NewSeats2,
  468. tables = NewTables,
  469. player_id_counter = PlayerId + 1,
  470. table_id_counter = TableId + 1,
  471. reg_requests = NewRegRequests,
  472. cr_tab_requests = NewTCrRequests}}.
  473. unreg_player_and_eliminate_table(PlayerId, TableId,
  474. #state{players = Players, tables = Tables,
  475. table_module = TableModule, seats = Seats} = StateData) ->
  476. NewPlayers = del_player(PlayerId, Players),
  477. TablePid = get_table_pid(TableId, Tables),
  478. NewSeats = del_seats_by_table_id(TableId, Seats),
  479. NewTables = del_table(TableId, Tables),
  480. send_to_table(TableModule, TablePid, stop),
  481. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers,
  482. seats = NewSeats,
  483. tables = NewTables}}.
  484. replace_player_by_bot(PlayerId, TableId, SeatNum,
  485. #state{players = Players, seats = Seats, game_id = GameId,
  486. bot_module = BotModule, table_module = TableModule,
  487. player_id_counter = NewPlayerId, tables = Tables,
  488. tab_requests = Requests} = StateData) ->
  489. NewPlayers = del_player(PlayerId, Players),
  490. [#'PlayerInfo'{id = UserId} = UserInfo] = spawn_bots(GameId, BotModule, 1),
  491. NewPlayers2 = reg_player(#player{id = NewPlayerId, user_id = UserId, is_bot = true}, NewPlayers),
  492. NewSeats = assign_seat(TableId, SeatNum, NewPlayerId, true, false, false, Seats),
  493. TablePid = get_table_pid(TableId, Tables),
  494. NewRequests = table_req_replace_player(TableModule, TablePid, NewPlayerId, UserInfo, TableId, SeatNum, Requests),
  495. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  496. seats = NewSeats,
  497. player_id_counter = NewPlayerId + 1,
  498. tab_requests = NewRequests}}.
  499. %% table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests) -> NewRequests
  500. table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests) ->
  501. RequestId = make_ref(),
  502. NewRequests = dict:store(RequestId, {replace_player, PlayerId, TableId, SeatNum}, TabRequests),
  503. send_to_table(TableModule, TablePid, {replace_player, RequestId, UserInfo, PlayerId, SeatNum}),
  504. NewRequests.
  505. %% players_init() -> players()
  506. players_init() ->
  507. midict:new().
  508. %% reg_player(#player{}, Players) -> NewPlayers
  509. reg_player(#player{id =Id, user_id = UserId} = Player, Players) ->
  510. midict:store(Id, Player, [{user_id, UserId}], Players).
  511. get_player_id_by_user_id(UserId, Players) ->
  512. case midict:geti(UserId, user_id, Players) of
  513. [#player{id = PlayerId}] -> {ok, PlayerId};
  514. [] -> error
  515. end.
  516. %% del_player(PlayerId, Players) -> NewPlayers
  517. del_player(PlayerId, Players) ->
  518. midict:erase(PlayerId, Players).
  519. %% del_player(PlayersIds, Players) -> NewPlayers
  520. del_players([], Players) -> Players;
  521. del_players([PlayerId | Rest], Players) ->
  522. del_players(Rest, del_player(PlayerId, Players)).
  523. tables_init() ->
  524. midict:new().
  525. reg_table(TableId, Pid, MonRef, GlobalId, Scoring, Tables) ->
  526. Table = #table{id = TableId, pid = Pid, mon_ref = MonRef, global_id = GlobalId,
  527. state = initializing, scoring_state = Scoring},
  528. store_table(Table, Tables).
  529. update_created_table(TableId, Relay, Tables) ->
  530. Table = midict:fetch(TableId, Tables),
  531. NewTable = Table#table{relay = Relay, state = ?TABLE_STATE_READY},
  532. store_table(NewTable, Tables).
  533. store_table(#table{id = TableId, pid = Pid, mon_ref = MonRef, global_id = GlobalId} = Table, Tables) ->
  534. midict:store(TableId, Table, [{pid, Pid}, {global_id, GlobalId}, {mon_ref, MonRef}], Tables).
  535. fetch_table(TableId, Tables) ->
  536. midict:fetch(TableId, Tables).
  537. get_table(TableId, Tables) ->
  538. midict:find(TableId, Tables).
  539. get_table_pid(TabId, Tables) ->
  540. {ok, #table{pid = TabPid}} = midict:find(TabId, Tables),
  541. TabPid.
  542. del_table(TabId, Tables) ->
  543. midict:erase(TabId, Tables).
  544. get_table_by_mon_ref(MonRef, Tables) ->
  545. case midict:geti(MonRef, mon_ref, Tables) of
  546. [Table] -> Table;
  547. [] -> not_found
  548. end.
  549. set_table_state(TableId, State, Tables) ->
  550. Table = midict:fetch(TableId, Tables),
  551. store_table(Table#table{state = State}, Tables).
  552. seats_init() ->
  553. midict:new().
  554. find_bot_seat_without_players(Seats, PlayersList) ->
  555. case midict:geti(true, is_bot, Seats) of
  556. [] -> not_found;
  557. List ->
  558. TabList = lists:usort([TabId || #seat{table = TabId} <- List]),
  559. lookup_bot_seat_without_players(TabList, PlayersList, Seats)
  560. end.
  561. lookup_bot_seat_without_players([], _, _) -> not_found;
  562. lookup_bot_seat_without_players([TabId | Rest], PlayersList, Seats) ->
  563. TabPlayers = [Id || #seat{player_id=Id} <-
  564. midict:geti(TabId, non_free_at_tab, Seats), lists:member(Id, PlayersList)],
  565. if TabPlayers == [] ->
  566. gas:info(?MODULE,"TRN_LUCKY Seats:~p", [midict:geti(TabId, table_id, Seats)]),
  567. hd(midict:geti(TabId, bot_at_tab, Seats));
  568. true -> lookup_bot_seat_without_players(Rest, PlayersList, Seats)
  569. end.
  570. find_seats_with_players_for_table_id(TabId, Seats) ->
  571. midict:geti(TabId, non_free_at_tab, Seats).
  572. find_seats_by_player_id(PlayerId, Seats) ->
  573. midict:geti(PlayerId, player_id, Seats).
  574. find_seats_by_table_id(TabId, Seats) ->
  575. midict:geti(TabId, table_id, Seats).
  576. %% real_players_at_table(TabId, Seats) -> Num
  577. real_players_at_table(TabId, Seats) ->
  578. length(find_real_players_seats_at_tab(TabId, Seats)).
  579. is_all_players_connected(TableId, TableSeatsNum, Seats) ->
  580. TableSeatsNum == length(midict:geti(true, {connected, TableId}, Seats)).
  581. find_real_players_seats_at_tab(TabId, Seats) ->
  582. midict:geti(TabId, real_player_at_tab, Seats).
  583. fetch_seat(TableId, SeatNum, Seats) ->
  584. midict:fetch({TableId, SeatNum}, Seats).
  585. %% assign_seat(TabId, SeatNum, PlayerId, IsBot, RegByTable, Connected, Seats) -> NewSeats
  586. %% PlayerId = integer()
  587. %% IsBot = RegByTable = Connected = undefined | boolean()
  588. assign_seat(TabId, SeatNum, PlayerId, IsBot, RegByTable, Connected, Seats) ->
  589. Seat = #seat{table = TabId, seat_num = SeatNum, player_id = PlayerId,
  590. is_bot = IsBot, registered_by_table = RegByTable, connected = Connected},
  591. store_seat(Seat, Seats).
  592. update_seat_connect_status(TableId, SeatNum, ConnStatus, Seats) ->
  593. Seat = midict:fetch({TableId, SeatNum}, Seats),
  594. NewSeat = Seat#seat{connected = ConnStatus},
  595. store_seat(NewSeat, Seats).
  596. store_seat(#seat{table = TabId, seat_num = SeatNum, player_id = PlayerId,
  597. is_bot = IsBot, registered_by_table = _RegByTable,
  598. connected = Connected} = Seat, Seats) ->
  599. Indices = if PlayerId == undefined ->
  600. [{table_id, TabId}, {free, true}, {free_at_tab, TabId}];
  601. true ->
  602. I = [{table_id, TabId}, {free, false}, {non_free_at_tab, TabId},
  603. {player_id, PlayerId}, {is_bot, IsBot},
  604. {{connected, TabId}, Connected}],
  605. if IsBot -> [{bot_at_tab, TabId} | I];
  606. true -> [{real_player_at_tab, TabId} | I]
  607. end
  608. end,
  609. midict:store({TabId, SeatNum}, Seat, Indices, Seats).
  610. create_seats(_TabId, 0, Seats) -> Seats;
  611. create_seats(TabId, SeatNum, Seats) ->
  612. NewSeats = assign_seat(TabId, SeatNum, undefined, undefined, undefined, undefined, Seats),
  613. create_seats(TabId, SeatNum - 1, NewSeats).
  614. del_seats_by_table_id(TabId, Seats) ->
  615. F = fun(#seat{seat_num = SeatNum}, Acc) ->
  616. midict:erase({TabId, SeatNum}, Acc)
  617. end,
  618. lists:foldl(F, Seats, find_seats_by_table_id(TabId, Seats)).
  619. spawn_bots(GameId, BotModule, 1) ->
  620. [ spawn_bot(BotModule, GameId, auth_server:robot_credentials()) ];
  621. spawn_bots(GameId, BotModule, BotsNum) ->
  622. [ spawn_bot(BotModule, GameId, PlayerInfo)
  623. || PlayerInfo <- lists:sublist(auth_server:spare(),BotsNum) ].
  624. spawn_bot(BotModule, GameId, PlayerInfo) ->
  625. {NPid, UserInfo} = create_robot(BotModule, GameId, PlayerInfo),
  626. BotModule:join_game(NPid),
  627. PlayerInfo.
  628. create_robot(BotModule, GameId, UserInfo) ->
  629. {ok, NPid} = BotModule:start(self(), UserInfo, GameId),
  630. BotModule:get_session(NPid), %% Hack for the tavla_bot. Creates a game session process.
  631. {NPid, UserInfo}.
  632. spawn_table(TabMod, GameId, TableId, Params) ->
  633. Pid = TabMod:start(GameId, TableId, Params),
  634. Pid.
  635. send_to_table(TabMod, TabPid, Message) ->
  636. TabMod:parent_message(TabPid, Message).
  637. get_param(ParamId, Params) ->
  638. {_, Value} = lists:keyfind(ParamId, 1, Params),
  639. Value.
  640. %% add_points_to_accounts(Points, GameId, GameType, GameMode) -> ok
  641. %% Types: Points = [{UserId, GamePoints}]
  642. add_points_to_accounts(Points, GameId, GameType, GameMode) ->
  643. TI = #ti_game_event{game_name = GameType, game_mode = GameMode,
  644. id = GameId, double_points = 1,
  645. type = game_end, tournament_type = lucky},
  646. [begin
  647. if GamePoints =/= 0 ->
  648. %{UserId,game_points,GameType,GameMode,TournamentType}
  649. kvs:add(#transaction{
  650. id=kvs:next_id(transaction,1),
  651. feed_id={game_points,UserId},
  652. amount=GamePoints,
  653. comment=TI});
  654. true -> do_nothing
  655. end
  656. end || {UserId, GamePoints} <- Points],
  657. ok.