wxerlangtest.erl

-module(test).
-export([start/0]).
-include_lib("/usr/local/Cellar/erlang/21.1.2/lib/erlang/lib/wx-1.8.5/include/wx.hrl").
 
-define(Connect, 150).
-define(Send, 151).
 
start() -> spawn_link(fun() -> init() end).
 
init() ->
  wx:new(),
  Frame = wxFrame:new(wx:null(), ?wxID_ANY, "Simple chat client", [{size, {400, 400}},
    {style, ?wxDEFAULT_FRAME_STYLE band (bnot (?wxMAXIMIZE_BOX bor ?wxRESIZE_BORDER))}]),
  Panel = wxPanel:new(Frame),
  MainSizer = wxBoxSizer:new(?wxHORIZONTAL),
  wxSizer:addSpacer(MainSizer, 15),
  VSizer1 = wxBoxSizer:new(?wxVERTICAL),
  wxSizer:addSpacer(VSizer1, 15),
 
  OutputText = wxTextCtrl:new(Panel, ?wxID_ANY,
    [{size, {250, 250}},
      {style, ?wxTE_MULTILINE}]),
  wxTextCtrl:setEditable(OutputText, false),
  wxSizer:add(VSizer1, OutputText, []),
  wxSizer:addSpacer(VSizer1, 25),
  InputText = wxTextCtrl:new(Panel, ?wxID_ANY,
    [{size, {250, 50}}, {style, ?wxTE_MULTILINE}]),
  wxSizer:add(VSizer1, InputText, []),
  wxSizer:addSpacer(VSizer1, 10),
  wxSizer:add(MainSizer, VSizer1, []),
  wxSizer:addSpacer(MainSizer, 20),
 
  VSizer2 = wxBoxSizer:new(?wxVERTICAL),
  wxSizer:addSpacer(VSizer2, 15),
  BConnect = wxButton:new(Panel, ?Connect, [{label, "Connect"}, {size, {80, 30}}]),
  BSend = wxButton:new(Panel, ?Send, [{label, "Send"}, {size, {80, 50}}]),
  NickText = wxTextCtrl:new(Panel, ?wxID_ANY, [{value, "NickName"}, {size, {80, 25}}]),
  Ip = wxTextCtrl:new(Panel, ?wxID_ANY, [{value, "127.0.0.1"}, {size, {80, 25}}]),
  Port = wxTextCtrl:new(Panel, ?wxID_ANY, [{value, "6268"}, {size, {80, 25}}]),
  wxSizer:add(VSizer2, BConnect, []),
  wxSizer:addSpacer(VSizer2, 10),
  wxSizer:add(VSizer2, Ip, []),
  wxSizer:addSpacer(VSizer2, 10),
  wxSizer:add(VSizer2, Port, []),
  wxSizer:addSpacer(VSizer2, 50),
  wxSizer:add(VSizer2, NickText, []),
  wxSizer:addSpacer(VSizer2, 100),
  wxSizer:add(VSizer2, BSend),
 
  wxSizer:add(MainSizer, VSizer2, []),
  wxPanel:setSizer(Panel, MainSizer),
  wxPanel:connect(Panel, command_button_clicked),
  wxFrame:connect(Frame, close_window),
  wxFrame:show(Frame),
  Pid = self(),
  loop([Frame, OutputText, InputText, BConnect], {NickText, Ip, Port}, [], Pid).
 
loop([Frame, OutputText, InputText, B1] = Widgets, {NickText, Ip, Port} = Params, ClientSocket, Pid) ->
  receive
 
    {add_message, Answer} ->
      wxTextCtrl:appendText(OutputText, "\n" ++ Answer),
      loop(Widgets, Params, ClientSocket, Pid);
 
    #wx{id = ?Connect, event = #wxCommand{type = command_button_clicked}} ->
      Answer = client(
        iptotuple(wxTextCtrl:getValue(Ip)),
        get_port(wxTextCtrl:getValue(Port)),
        Pid),
      case Answer of
        {ok, Socket} ->
          wxTextCtrl:setEditable(NickText, false),
          wxTextCtrl:setEditable(Ip, false),
          wxTextCtrl:setEditable(Port, false),
          wxWindow:enable(B1, [{enable, false}]),
          send(Socket, term_to_binary({add, new})),
          loop(Widgets, Params, Socket, Pid);
        {error, Reason} ->
          MD = wxMessageDialog:new(Frame, "Невозможно установить соединение : " ++ atom_to_list(Reason),
            [{style, ?wxOK}, {caption, "Warning"}]),
          wxDialog:showModal(MD),
          wxDialog:destroy(MD),
          loop(Widgets, Params, ClientSocket, Pid)
      end;
 
    #wx{id = ?Send, event = #wxCommand{type = command_button_clicked}} ->
      case wxWindow:isEnabled(B1) of
        false ->
          send(
            ClientSocket,
            create_message(
              wxTextCtrl:getValue(NickText),
              wxTextCtrl:getValue(InputText))),
          wxTextCtrl:setValue(InputText,"");
        true ->
          MD = wxMessageDialog:new(Frame, "Вначале установите соединение!",
            [{style, ?wxOK}, {caption, "Warning"}]),
          wxDialog:showModal(MD),
          wxDialog:destroy(MD)
      end,
      loop(Widgets, Params, ClientSocket, Pid);
 
    #wx{event = #wxClose{}} ->
      case is_list(ClientSocket) of
        true -> ok;
        false -> disconnect(ClientSocket)
      end,
      wx:destroy()
 
  end.
 
client(Host, Port, Pid) ->
  case gen_tcp:connect(Host, Port, [binary, {active, false}, {packet, 0}]) of
    {ok, Socket} -> spawn(fun() -> recv(Socket, Pid) end), {ok, Socket};
    Other -> Other
  end.
 
recv(Socket, Pid) ->
  case gen_tcp:recv(Socket, 0) of
    {ok, Data} -> Pid ! {add_message, Data}, recv(Socket, Pid);
    {error, Reason} -> Reason
  end.
 
send(Socket, Message) ->
  gen_tcp:send(Socket, Message).
 
disconnect(Socket) ->
  gen_tcp:close(Socket).
 
create_message(Nick, Text) -> unicode:characters_to_binary(Nick ++ " >> " ++ Text).
 
iptotuple(Ip) ->
  list_to_tuple([fun() -> {N, _} = string:to_integer(X), N end()
    || X <- string:tokens(Ip, ".")]).
 
get_port(Str) -> element(1, string:to_integer(Str)).