lucky.erl 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782
  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 :: dict(), %% {TableId, PlayersIds}
  49. reg_requests :: dict(), %% {PlayerId, From}
  50. tab_requests :: dict() %% {RequestId, RequestContext}
  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 = IsBot}] ->
  259. case real_players_at_table(TableId, Seats) of
  260. 1 when not IsBot -> %% Last real player gone
  261. gas:info(?MODULE,"TRN_LUCKY <~p> Last real player gone from "
  262. "table <~p>. Closing the table.", [GameId, TableId]),
  263. unreg_player_and_eliminate_table(PlayerId, TableId, StateData);
  264. _ ->
  265. gas:info(?MODULE,"TRN_LUCKY <~p> Al least one real player is at table <~p>. "
  266. "Starting a bot to replace free seat.", [GameId, TableId]),
  267. replace_player_by_bot(PlayerId, TableId, SeatNum, StateData)
  268. end;
  269. [] -> %% Ignoring the message
  270. {next_state, ?STATE_PROCESSING, StateData}
  271. end;
  272. handle_table_message(TableId, {table_created, Relay}, ?STATE_PROCESSING,
  273. #state{game_id = GameId, tables = Tables, seats = Seats,
  274. cr_tab_requests = TCrRequests, table_module = TableModule,
  275. reg_requests = RegRequests} = StateData)
  276. when is_integer(TableId) ->
  277. gas:info(?MODULE,"TRN_LUCKY <~p> The <table_created> notification received from table: ~p.",
  278. [GameId, TableId]),
  279. TabInitPlayers = dict:fetch(TableId, TCrRequests),
  280. %% Update status of players
  281. TabSeats = find_seats_by_table_id(TableId, Seats),
  282. F = fun(#seat{player_id = PlayerId} = S, Acc) ->
  283. case lists:member(PlayerId, TabInitPlayers) of
  284. true -> store_seat(S#seat{registered_by_table = true}, Acc);
  285. false -> Acc
  286. end
  287. end,
  288. NewSeats = lists:foldl(F, Seats, TabSeats),
  289. %% Process delayed registration requests
  290. TablePid = get_table_pid(TableId, Tables),
  291. F2 = fun(PlayerId, Acc) ->
  292. case dict:find(PlayerId, Acc) of
  293. {ok, From} ->
  294. gen_fsm:reply(From, {ok, {PlayerId, Relay, {TableModule, TablePid}}}),
  295. dict:erase(PlayerId, Acc);
  296. error -> Acc
  297. end
  298. end,
  299. NewRegRequests = lists:foldl(F2, RegRequests, TabInitPlayers),
  300. NewTCrRequests = dict:erase(TableId, TCrRequests),
  301. NewTables = update_created_table(TableId, Relay, Tables),
  302. {next_state, ?STATE_PROCESSING, StateData#state{tables = NewTables,
  303. seats = NewSeats,
  304. cr_tab_requests = NewTCrRequests,
  305. reg_requests = NewRegRequests}};
  306. handle_table_message(TableId, {round_finished, NewScoringState, RoundScore, _TotalScore},
  307. ?STATE_PROCESSING,
  308. #state{game_id = GameId, tables = Tables, table_module = TableModule,
  309. game_mode = GameMode, game = GameType, players = Players} = StateData)
  310. when is_integer(TableId) ->
  311. gas:info(?MODULE,"TRN_LUCKY <~p> The <round_finished> received from table: ~p~nScore: ~p.",
  312. [GameId, TableId,RoundScore]),
  313. %% Add score per round
  314. UsersPoints = lists:flatten(
  315. [ begin
  316. case midict:find(PlayerId, Players) of
  317. {ok, #player{id = UserId, is_bot = false}} -> {UserId, Points};
  318. _Error -> gas:info(?MODULE, "get_player_info_by_user_id ~p", [_Error]), [] end
  319. end|| {PlayerId, Points} <- RoundScore]),
  320. add_points_to_accounts(UsersPoints, GameId, GameType, GameMode),
  321. #table{pid = TablePid} = Table = fetch_table(TableId, Tables),
  322. TRef = erlang:send_after(?REST_TIMEOUT, self(), {rest_timeout, TableId}),
  323. NewTable = Table#table{scoring_state = NewScoringState, state = ?TABLE_STATE_FINISHED, timer = TRef},
  324. NewTables = store_table(NewTable, Tables),
  325. send_to_table(TableModule, TablePid, show_round_result),
  326. {next_state, ?STATE_PROCESSING, StateData#state{tables = NewTables}};
  327. handle_table_message(TableId, {response, RequestId, Response},
  328. ?STATE_PROCESSING,
  329. #state{game_id = GameId, tab_requests = TabRequests} = StateData)
  330. when is_integer(TableId) ->
  331. NewTabRequests = dict:erase(RequestId, TabRequests),
  332. case dict:find(RequestId, TabRequests) of
  333. {ok, ReqContext} ->
  334. gas:info(?MODULE,"TRN_LUCKY <~p> A response received from table <~p>. "
  335. "RequestId: ~p. Request context: ~p. Response: ~p",
  336. [GameId, TableId, RequestId, ReqContext, Response]),
  337. handle_table_response(ReqContext, Response, ?STATE_PROCESSING,
  338. StateData#state{tab_requests = NewTabRequests});
  339. error ->
  340. gas:error(?MODULE,"TRN_LUCKY <~p> Table <~p> sent a response for unknown request. "
  341. "RequestId: ~p. Response", []),
  342. {next_state, ?STATE_PROCESSING, StateData#state{tab_requests = NewTabRequests}}
  343. end;
  344. handle_table_message(_TableId, _Event, StateName, StateData) ->
  345. {next_state, StateName, StateData}.
  346. %%===================================================================
  347. %% handle_table_response(RequestContext, Response, StateName, StateData)
  348. handle_table_response({replace_player, PlayerId, TableId, SeatNum}, ok = _Response,
  349. ?STATE_PROCESSING,
  350. #state{reg_requests = RegRequests, seats = Seats,
  351. tables = Tables, table_module = TableModule} = StateData) ->
  352. Seat = fetch_seat(TableId, SeatNum, Seats),
  353. NewSeats = store_seat(Seat#seat{registered_by_table = true}, Seats),
  354. %% Send response to a client for a delayed request
  355. NewRegRequests =
  356. case dict:find(PlayerId, RegRequests) of
  357. {ok, From} ->
  358. #table{relay = Relay, pid = TablePid} = fetch_table(TableId, Tables),
  359. gen_fsm:reply(From, {ok, {PlayerId, Relay, {TableModule, TablePid}}}),
  360. dict:erase(PlayerId, RegRequests);
  361. error -> RegRequests
  362. end,
  363. {next_state, ?STATE_PROCESSING, StateData#state{seats = NewSeats,
  364. reg_requests = NewRegRequests}}.
  365. %%===================================================================
  366. handle_client_request({reg, User}, From, ?STATE_PROCESSING,
  367. #state{game_id = GameId, reg_requests = RegRequests,
  368. seats = Seats, players=Players, tables = Tables,
  369. table_module = TableModule} = StateData) ->
  370. #'PlayerInfo'{id = UserId, robot = IsBot} = User,
  371. gas:info(?MODULE,"TRN_LUCKY <~p> The Register request received from user: ~p.", [GameId, UserId]),
  372. case IsBot of
  373. true -> %% Bots can't initiate a registration
  374. case get_player_id_by_user_id(UserId, Players) of
  375. {ok, PlayerId} -> %% Already registered. Send table requsites.
  376. [#seat{table = TableId, registered_by_table = RegByTable}] = find_seats_by_player_id(PlayerId, Seats),
  377. case RegByTable of
  378. false -> %% Store delayed request
  379. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  380. {next_state, ?STATE_PROCESSING, StateData#state{reg_requests = NewRegRequests}};
  381. _ ->
  382. #table{relay = Relay, pid = TPid} = fetch_table(TableId, Tables),
  383. {reply, {ok, {PlayerId, Relay, {TableModule, TPid}}}, ?STATE_PROCESSING, StateData}
  384. end;
  385. error -> %% Not registered
  386. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p is a bot. The user not registered. "
  387. "Rejecting registration.", [GameId, UserId]),
  388. {reply, {error, indy_bots_not_allowed}, ?STATE_PROCESSING, StateData}
  389. end;
  390. false -> %% Normal user
  391. IgnoredPlayers = [Id || #player{id = Id} <- midict:geti(UserId, user_id, Players)],
  392. gas:info(?MODULE,"TRN_LUCKY <~p> There are no table with free seats.", [GameId]),
  393. case find_bot_seat_without_players(Seats, IgnoredPlayers) of
  394. #seat{table = TabId, seat_num = SeatNum, player_id = OldPlayerId} ->
  395. gas:info(?MODULE,"TRN_LUCKY <~p> Found a seat with a bot. Replacing by the user. "
  396. "UserId:~p TabId: ~p SeatNum: ~p.", [GameId, UserId, TabId, SeatNum]),
  397. reg_player_with_replace(User, TabId, SeatNum, OldPlayerId, From, StateData);
  398. not_found ->
  399. gas:info(?MODULE,"TRN_LUCKY <~p> There are no seats with bots. "
  400. "Creating new table for user: ~p.", [GameId, UserId]),
  401. reg_player_at_new_table(User, From, StateData)
  402. end
  403. end;
  404. handle_client_request(Request, From, StateName, #state{game_id = GameId} = StateData) ->
  405. gas:info(?MODULE,"TRN_LUCKY <~p> Unhandled client request received from ~p in "
  406. "state <~p>: ~p.", [GameId, From, StateName, Request]),
  407. {reply, {error, unexpected_request}, StateName, StateData}.
  408. %%===================================================================
  409. reg_player_with_replace(UserInfo, TableId, SeatNum, OldPlayerId, From,
  410. #state{game_id = GameId, players = Players, tables = Tables,
  411. seats = Seats, player_id_counter = PlayerId,
  412. tab_requests = TabRequests, reg_requests = RegRequests,
  413. table_module = TableModule} = StateData) ->
  414. #'PlayerInfo'{id = UserId, robot = IsBot} = UserInfo,
  415. NewPlayers = del_player(OldPlayerId, Players),
  416. NewPlayers2 = reg_player(#player{id = PlayerId, user_id = UserId, is_bot = IsBot}, NewPlayers),
  417. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p registered as player <~p>.", [GameId, UserId, PlayerId]),
  418. NewSeats = assign_seat(TableId, SeatNum, PlayerId, IsBot, false, false, Seats),
  419. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p assigned to seat <~p> of table <~p>.", [GameId, UserId, SeatNum, TableId]),
  420. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  421. TablePid = get_table_pid(TableId, Tables),
  422. NewTabRequests = table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests),
  423. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  424. seats = NewSeats,
  425. player_id_counter = PlayerId + 1,
  426. tab_requests = NewTabRequests,
  427. reg_requests = NewRegRequests}}.
  428. reg_player_at_new_table(User, From,
  429. #state{game_id = GameId, players = Players,
  430. tables = Tables, seats = Seats, seats_per_table = SeatsNum,
  431. player_id_counter = PlayerIdCounter,
  432. table_id_counter = TableId, table_module = TableModule,
  433. bot_module = BotModule, table_params = TableParams,
  434. reg_requests = RegRequests, cr_tab_requests = TCrRequests
  435. } = StateData) ->
  436. #'PlayerInfo'{id = UserId, robot = IsBot} = User,
  437. RobotsInfo = spawn_bots(GameId, BotModule, SeatsNum - 1),
  438. gas:info(?MODULE,"TRN_LUCKY <~p> Bots for table <~p> are spawned.", [GameId, TableId]),
  439. F = fun(BotInfo, {PlId,SNum}) -> {{PlId, BotInfo, SNum, _Points = 0}, {PlId + 1, SNum + 1}} end,
  440. {RobotsRegData, {PlayerId, SeatNum}} = lists:mapfoldl(F, {PlayerIdCounter, 1}, RobotsInfo),
  441. TPlayers = [{PlayerId, User, SeatNum, 0} | RobotsRegData],
  442. TableParams2 = [{players, TPlayers}, {table_name, "I'm filling lucky"} | TableParams],
  443. {ok, TabPid} = spawn_table(TableModule, GameId, TableId, TableParams2),
  444. MonRef = erlang:monitor(process, TabPid),
  445. %% FIXME: Table global id should use a persistent counter
  446. NewTables = reg_table(TableId, TabPid, MonRef, 0, undefined, Tables),
  447. gas:info(?MODULE,"TRN_LUCKY <~p> New table created: ~p.", [GameId, TableId]),
  448. NewPlayers = reg_player(#player{id = PlayerId, user_id = UserId, is_bot = IsBot}, Players),
  449. F2 = fun({PlId, #'PlayerInfo'{id = UId, robot=Bot}, _SNum, _Points}, Acc) ->
  450. reg_player(#player{id = PlId, user_id = UId, is_bot = Bot}, Acc)
  451. end,
  452. NewPlayers2 = lists:foldl(F2, NewPlayers, RobotsRegData),
  453. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p registered as player <~p>.", [GameId, UserId, PlayerId]),
  454. NewSeats = assign_seat(TableId, SeatNum, PlayerId, IsBot, false, false, Seats),
  455. F3 = fun({PlId, _UserInfo, SNum, _Points}, Acc) ->
  456. assign_seat(TableId, SNum, PlId, true, false, false, Acc)
  457. end,
  458. NewSeats2 = lists:foldl(F3, NewSeats, RobotsRegData),
  459. gas:info(?MODULE,"TRN_LUCKY <~p> User ~p assigned to seat <~p> of table <~p>.", [GameId, UserId, SeatNum, TableId]),
  460. NewRegRequests = dict:store(PlayerId, From, RegRequests),
  461. PlayersIds = [PlayerId | [PlId || {PlId, _, _, _} <- RobotsRegData]],
  462. NewTCrRequests = dict:store(TableId, PlayersIds, TCrRequests),
  463. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  464. seats = NewSeats2,
  465. tables = NewTables,
  466. player_id_counter = PlayerId + 1,
  467. table_id_counter = TableId + 1,
  468. reg_requests = NewRegRequests,
  469. cr_tab_requests = NewTCrRequests}}.
  470. unreg_player_and_eliminate_table(PlayerId, TableId,
  471. #state{players = Players, tables = Tables,
  472. table_module = TableModule, seats = Seats} = StateData) ->
  473. NewPlayers = del_player(PlayerId, Players),
  474. TablePid = get_table_pid(TableId, Tables),
  475. NewSeats = del_seats_by_table_id(TableId, Seats),
  476. NewTables = del_table(TableId, Tables),
  477. send_to_table(TableModule, TablePid, stop),
  478. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers,
  479. seats = NewSeats,
  480. tables = NewTables}}.
  481. replace_player_by_bot(PlayerId, TableId, SeatNum,
  482. #state{players = Players, seats = Seats, game_id = GameId,
  483. bot_module = BotModule, table_module = TableModule,
  484. player_id_counter = NewPlayerId, tables = Tables,
  485. tab_requests = Requests} = StateData) ->
  486. NewPlayers = del_player(PlayerId, Players),
  487. [#'PlayerInfo'{id = UserId} = UserInfo] = spawn_bots(GameId, BotModule, 1),
  488. NewPlayers2 = reg_player(#player{id = NewPlayerId, user_id = UserId, is_bot = true}, NewPlayers),
  489. NewSeats = assign_seat(TableId, SeatNum, NewPlayerId, true, false, false, Seats),
  490. TablePid = get_table_pid(TableId, Tables),
  491. NewRequests = table_req_replace_player(TableModule, TablePid, NewPlayerId, UserInfo, TableId, SeatNum, Requests),
  492. {next_state, ?STATE_PROCESSING, StateData#state{players = NewPlayers2,
  493. seats = NewSeats,
  494. player_id_counter = NewPlayerId + 1,
  495. tab_requests = NewRequests}}.
  496. %% table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests) -> NewRequests
  497. table_req_replace_player(TableModule, TablePid, PlayerId, UserInfo, TableId, SeatNum, TabRequests) ->
  498. RequestId = make_ref(),
  499. NewRequests = dict:store(RequestId, {replace_player, PlayerId, TableId, SeatNum}, TabRequests),
  500. send_to_table(TableModule, TablePid, {replace_player, RequestId, UserInfo, PlayerId, SeatNum}),
  501. NewRequests.
  502. %% players_init() -> players()
  503. players_init() ->
  504. midict:new().
  505. %% reg_player(#player{}, Players) -> NewPlayers
  506. reg_player(#player{id =Id, user_id = UserId} = Player, Players) ->
  507. midict:store(Id, Player, [{user_id, UserId}], Players).
  508. get_player_id_by_user_id(UserId, Players) ->
  509. case midict:geti(UserId, user_id, Players) of
  510. [#player{id = PlayerId}] -> {ok, PlayerId};
  511. [] -> error
  512. end.
  513. %% del_player(PlayerId, Players) -> NewPlayers
  514. del_player(PlayerId, Players) ->
  515. midict:erase(PlayerId, Players).
  516. %% del_player(PlayersIds, Players) -> NewPlayers
  517. del_players([], Players) -> Players;
  518. del_players([PlayerId | Rest], Players) ->
  519. del_players(Rest, del_player(PlayerId, Players)).
  520. tables_init() ->
  521. midict:new().
  522. reg_table(TableId, Pid, MonRef, GlobalId, Scoring, Tables) ->
  523. Table = #table{id = TableId, pid = Pid, mon_ref = MonRef, global_id = GlobalId,
  524. state = initializing, scoring_state = Scoring},
  525. store_table(Table, Tables).
  526. update_created_table(TableId, Relay, Tables) ->
  527. Table = midict:fetch(TableId, Tables),
  528. NewTable = Table#table{relay = Relay, state = ?TABLE_STATE_READY},
  529. store_table(NewTable, Tables).
  530. store_table(#table{id = TableId, pid = Pid, mon_ref = MonRef, global_id = GlobalId} = Table, Tables) ->
  531. midict:store(TableId, Table, [{pid, Pid}, {global_id, GlobalId}, {mon_ref, MonRef}], Tables).
  532. fetch_table(TableId, Tables) ->
  533. midict:fetch(TableId, Tables).
  534. get_table(TableId, Tables) ->
  535. midict:find(TableId, Tables).
  536. get_table_pid(TabId, Tables) ->
  537. {ok, #table{pid = TabPid}} = midict:find(TabId, Tables),
  538. TabPid.
  539. del_table(TabId, Tables) ->
  540. midict:erase(TabId, Tables).
  541. get_table_by_mon_ref(MonRef, Tables) ->
  542. case midict:geti(MonRef, mon_ref, Tables) of
  543. [Table] -> Table;
  544. [] -> not_found
  545. end.
  546. set_table_state(TableId, State, Tables) ->
  547. Table = midict:fetch(TableId, Tables),
  548. store_table(Table#table{state = State}, Tables).
  549. seats_init() ->
  550. midict:new().
  551. find_bot_seat_without_players(Seats, PlayersList) ->
  552. case midict:geti(true, is_bot, Seats) of
  553. [] -> not_found;
  554. List ->
  555. TabList = lists:usort([TabId || #seat{table = TabId} <- List]),
  556. lookup_bot_seat_without_players(TabList, PlayersList, Seats)
  557. end.
  558. lookup_bot_seat_without_players([], _, _) -> not_found;
  559. lookup_bot_seat_without_players([TabId | Rest], PlayersList, Seats) ->
  560. TabPlayers = [Id || #seat{player_id=Id} <-
  561. midict:geti(TabId, non_free_at_tab, Seats), lists:member(Id, PlayersList)],
  562. if TabPlayers == [] ->
  563. gas:info(?MODULE,"TRN_LUCKY Seats:~p", [midict:geti(TabId, table_id, Seats)]),
  564. hd(midict:geti(TabId, bot_at_tab, Seats));
  565. true -> lookup_bot_seat_without_players(Rest, PlayersList, Seats)
  566. end.
  567. find_seats_with_players_for_table_id(TabId, Seats) ->
  568. midict:geti(TabId, non_free_at_tab, Seats).
  569. find_seats_by_player_id(PlayerId, Seats) ->
  570. midict:geti(PlayerId, player_id, Seats).
  571. find_seats_by_table_id(TabId, Seats) ->
  572. midict:geti(TabId, table_id, Seats).
  573. %% real_players_at_table(TabId, Seats) -> Num
  574. real_players_at_table(TabId, Seats) ->
  575. length(find_real_players_seats_at_tab(TabId, Seats)).
  576. is_all_players_connected(TableId, TableSeatsNum, Seats) ->
  577. TableSeatsNum == length(midict:geti(true, {connected, TableId}, Seats)).
  578. find_real_players_seats_at_tab(TabId, Seats) ->
  579. midict:geti(TabId, real_player_at_tab, Seats).
  580. fetch_seat(TableId, SeatNum, Seats) ->
  581. midict:fetch({TableId, SeatNum}, Seats).
  582. %% assign_seat(TabId, SeatNum, PlayerId, IsBot, RegByTable, Connected, Seats) -> NewSeats
  583. %% PlayerId = integer()
  584. %% IsBot = RegByTable = Connected = undefined | boolean()
  585. assign_seat(TabId, SeatNum, PlayerId, IsBot, RegByTable, Connected, Seats) ->
  586. Seat = #seat{table = TabId, seat_num = SeatNum, player_id = PlayerId,
  587. is_bot = IsBot, registered_by_table = RegByTable, connected = Connected},
  588. store_seat(Seat, Seats).
  589. update_seat_connect_status(TableId, SeatNum, ConnStatus, Seats) ->
  590. Seat = midict:fetch({TableId, SeatNum}, Seats),
  591. NewSeat = Seat#seat{connected = ConnStatus},
  592. store_seat(NewSeat, Seats).
  593. store_seat(#seat{table = TabId, seat_num = SeatNum, player_id = PlayerId,
  594. is_bot = IsBot, registered_by_table = _RegByTable,
  595. connected = Connected} = Seat, Seats) ->
  596. Indices = if PlayerId == undefined ->
  597. [{table_id, TabId}, {free, true}, {free_at_tab, TabId}];
  598. true ->
  599. I = [{table_id, TabId}, {free, false}, {non_free_at_tab, TabId},
  600. {player_id, PlayerId}, {is_bot, IsBot},
  601. {{connected, TabId}, Connected}],
  602. if IsBot -> [{bot_at_tab, TabId} | I];
  603. true -> [{real_player_at_tab, TabId} | I]
  604. end
  605. end,
  606. midict:store({TabId, SeatNum}, Seat, Indices, Seats).
  607. create_seats(_TabId, 0, Seats) -> Seats;
  608. create_seats(TabId, SeatNum, Seats) ->
  609. NewSeats = assign_seat(TabId, SeatNum, undefined, undefined, undefined, undefined, Seats),
  610. create_seats(TabId, SeatNum - 1, NewSeats).
  611. del_seats_by_table_id(TabId, Seats) ->
  612. F = fun(#seat{seat_num = SeatNum}, Acc) ->
  613. midict:erase({TabId, SeatNum}, Acc)
  614. end,
  615. lists:foldl(F, Seats, find_seats_by_table_id(TabId, Seats)).
  616. spawn_bots(GameId, BotModule, 1) ->
  617. [ spawn_bot(BotModule, GameId, auth_server:robot_credentials()) ];
  618. spawn_bots(GameId, BotModule, BotsNum) ->
  619. [ spawn_bot(BotModule, GameId, PlayerInfo)
  620. || PlayerInfo <- lists:sublist(auth_server:spare(),BotsNum) ].
  621. spawn_bot(BotModule, GameId, PlayerInfo) ->
  622. {NPid, UserInfo} = create_robot(BotModule, GameId, PlayerInfo),
  623. BotModule:join_game(NPid),
  624. PlayerInfo.
  625. create_robot(BotModule, GameId, UserInfo) ->
  626. {ok, NPid} = BotModule:start(self(), UserInfo, GameId),
  627. BotModule:get_session(NPid), %% Hack for the game_tavla_bot. Creates a game session process.
  628. {NPid, UserInfo}.
  629. spawn_table(TabMod, GameId, TableId, Params) ->
  630. Pid = TabMod:start(GameId, TableId, Params),
  631. Pid.
  632. send_to_table(TabMod, TabPid, Message) ->
  633. TabMod:parent_message(TabPid, Message).
  634. get_param(ParamId, Params) ->
  635. {_, Value} = lists:keyfind(ParamId, 1, Params),
  636. Value.
  637. %% add_points_to_accounts(Points, GameId, GameType, GameMode) -> ok
  638. %% Types: Points = [{UserId, GamePoints}]
  639. add_points_to_accounts(Points, GameId, GameType, GameMode) ->
  640. TI = #ti_game_event{game_name = GameType, game_mode = GameMode,
  641. id = GameId, double_points = 1,
  642. type = game_end, tournament_type = lucky},
  643. [begin
  644. if GamePoints =/= 0 ->
  645. %{UserId,game_points,GameType,GameMode,TournamentType}
  646. kvs:add(#transaction{
  647. id=kvs:next_id(transaction,1),
  648. feed_id={game_points,UserId},
  649. amount=GamePoints,
  650. comment=TI});
  651. true -> do_nothing
  652. end
  653. end || {UserId, GamePoints} <- Points],
  654. ok.