CentrED

(*

 * CDDL HEADER START

 *

 * The contents of this file are subject to the terms of the

 * Common Development and Distribution License, Version 1.0 only

 * (the "License").  You may not use this file except in compliance

 * with the License.

 *

 * You can obtain a copy of the license at

 * http://www.opensource.org/licenses/cddl1.php.

 * See the License for the specific language governing permissions

 * and limitations under the License.

 *

 * When distributing Covered Code, include this CDDL HEADER in each

 * file and include the License file at

 * http://www.opensource.org/licenses/cddl1.php.  If applicable,

 * add the following below this CDDL HEADER, with the fields enclosed

 * by brackets "[]" replaced with your own identifying * information:

 *      Portions Copyright [yyyy] [name of copyright owner]

 *

 * CDDL HEADER END

 *

 *

 *      Portions Copyright 2007 Andreas Schneider

 *)

unit UfrmAccountControl;

{$mode objfpc}{$H+}

interface

uses

  Classes, SysUtils, LResources, Forms, Controls, Graphics, Dialogs, ComCtrls,

  VirtualTrees, VTHeaderPopup, UEnhancedMemoryStream, UEnums;

type

  { TfrmAccountControl }

  TfrmAccountControl = class(TForm)

    ilToolbar: TImageList;

    ilAccesslevel: TImageList;

    tbMain: TToolBar;

    tbRefresh: TToolButton;

    tbAddUser: TToolButton;

    tbEditUser: TToolButton;

    tbDeleteUser: TToolButton;

    tbSeparator1: TToolButton;

    vstAccounts: TVirtualStringTree;

    procedure tbEditUserClick(Sender: TObject);

    procedure FormClose(Sender: TObject; var CloseAction: TCloseAction);

    procedure FormCreate(Sender: TObject);

    procedure FormDestroy(Sender: TObject);

    procedure FormShow(Sender: TObject);

    procedure tbAddUserClick(Sender: TObject);

    procedure tbDeleteUserClick(Sender: TObject);

    procedure tbRefreshClick(Sender: TObject);

    procedure vstAccountsDblClick(Sender: TObject);

    procedure vstAccountsGetImageIndex(Sender: TBaseVirtualTree;

      Node: PVirtualNode; Kind: TVTImageKind; Column: TColumnIndex;

      var Ghosted: Boolean; var ImageIndex: Integer);

    procedure vstAccountsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode;

      Column: TColumnIndex; TextType: TVSTTextType; var CellText: WideString);

  protected

    procedure OnModifyUserResponse(ABuffer: TEnhancedMemoryStream);

    procedure OnDeleteUserResponse(ABuffer: TEnhancedMemoryStream);

    procedure OnListUsersPacket(ABuffer: TEnhancedMemoryStream);

    function FindNode(AUsername: string): PVirtualNode;

  public

    { public declarations }

  end;

var

  frmAccountControl: TfrmAccountControl;

implementation

uses

  UdmNetwork, UPacket, UPacketHandlers, UAdminHandling, UfrmEditAccount;

type

  PAccountInfo = ^TAccountInfo;

  TAccountInfo = record

    Username: string;

    AccessLevel: TAccessLevel;

  end;

  { TModifyUserPacket }

  TModifyUserPacket = class(TPacket)

    constructor Create(AUsername, APassword: string; AAccessLevel: TAccessLevel);

  end;

  { TDeleteUserPacket }

  TDeleteUserPacket = class(TPacket)

    constructor Create(AUsername: string);

  end;

  { TRequestUserListPacket }

  TRequestUserListPacket = class(TPacket)

    constructor Create;

  end;

{ TModifyUserPacket }

constructor TModifyUserPacket.Create(AUsername, APassword: string;

  AAccessLevel: TAccessLevel);

begin

  inherited Create($03, 0);

  FStream.WriteByte($05);

  FStream.WriteStringNull(AUsername);

  FStream.WriteStringNull(APassword);

  FStream.WriteByte(Byte(AAccessLevel));

end;

{ TDeleteUserPacket }

constructor TDeleteUserPacket.Create(AUsername: string);

begin

  inherited Create($03, 0);

  FStream.WriteByte($06);

  FStream.WriteStringNull(AUsername);

end;

{ TRequestUserListPacket }

constructor TRequestUserListPacket.Create;

begin

  inherited Create($03, 0);

  FStream.WriteByte($07);

end;

{ TfrmAccountControl }

procedure TfrmAccountControl.FormCreate(Sender: TObject);

begin

  vstAccounts.NodeDataSize := SizeOf(TAccountInfo);

  AdminPacketHandlers[$05] := TPacketHandler.Create(0, @OnModifyUserResponse);

  AdminPacketHandlers[$06] := TPacketHandler.Create(0, @OnDeleteUserResponse);

  AdminPacketHandlers[$07] := TPacketHandler.Create(0, @OnListUsersPacket);

end;

procedure TfrmAccountControl.FormClose(Sender: TObject;

  var CloseAction: TCloseAction);

begin

  CloseAction := caHide;

end;

procedure TfrmAccountControl.tbEditUserClick(Sender: TObject);

var

  selected: PVirtualNode;

  accountInfo: PAccountInfo;

begin

  selected := vstAccounts.GetFirstSelected;

  if selected <> nil then

  begin

    accountInfo := vstAccounts.GetNodeData(selected);

    with frmEditAccount do

    begin

      edUsername.Text := accountInfo^.Username;

      edUsername.Color := clBtnFace;

      edUsername.ReadOnly := True;

      edPassword.Text := '';

      lblPasswordHint.Visible := True;

      SetAccessLevel(accountInfo^.AccessLevel);

      if ShowModal = mrOK then

        dmNetwork.Send(TModifyUserPacket.Create(edUsername.Text, edPassword.Text, GetAccessLevel));

    end;

  end;

end;

procedure TfrmAccountControl.FormDestroy(Sender: TObject);

begin

  if AdminPacketHandlers[$05] <> nil then FreeAndNil(AdminPacketHandlers[$05]);

  if AdminPacketHandlers[$06] <> nil then FreeAndNil(AdminPacketHandlers[$06]);

  if AdminPacketHandlers[$07] <> nil then FreeAndNil(AdminPacketHandlers[$07]);

end;

procedure TfrmAccountControl.FormShow(Sender: TObject);

begin

  tbRefreshClick(Sender);

end;

procedure TfrmAccountControl.tbAddUserClick(Sender: TObject);

begin

  with frmEditAccount do

  begin

    edUsername.Text := '';

    edUsername.Color := clWindow;

    edUsername.ReadOnly := False;

    edPassword.Text := '';

    lblPasswordHint.Visible := False;

    cbAccessLevel.ItemIndex := 2;

    if ShowModal = mrOK then

      dmNetwork.Send(TModifyUserPacket.Create(edUsername.Text, edPassword.Text, GetAccessLevel));

  end;

end;

procedure TfrmAccountControl.tbDeleteUserClick(Sender: TObject);

var

  selected: PVirtualNode;

  accountInfo: PAccountInfo;

begin

  selected := vstAccounts.GetFirstSelected;

  if selected <> nil then

  begin

    accountInfo := vstAccounts.GetNodeData(selected);

    if MessageDlg('Confirmation', Format('Do you really want to delete "%s"?', [accountInfo^.Username]), mtConfirmation, [mbYes, mbNo], 0) = mrYes then

      dmNetwork.Send(TDeleteUserPacket.Create(accountInfo^.Username));

  end;

end;

procedure TfrmAccountControl.tbRefreshClick(Sender: TObject);

begin

  dmNetwork.Send(TRequestUserListPacket.Create);

end;

procedure TfrmAccountControl.vstAccountsDblClick(Sender: TObject);

begin

  tbEditUserClick(Sender);

end;

procedure TfrmAccountControl.vstAccountsGetImageIndex(Sender: TBaseVirtualTree;

  Node: PVirtualNode; Kind: TVTImageKind; Column: TColumnIndex;

  var Ghosted: Boolean; var ImageIndex: Integer);

var

  accountInfo: PAccountInfo;

begin

  if Column = 0 then

  begin

    accountInfo := Sender.GetNodeData(Node);

    case accountInfo^.AccessLevel of

      alNone: ImageIndex := 0;

      alView: ImageIndex := 1;

      alNormal: ImageIndex := 2;

      alAdministrator: ImageIndex := 3;

    end;

  end;

end;

procedure TfrmAccountControl.vstAccountsGetText(Sender: TBaseVirtualTree;

  Node: PVirtualNode; Column: TColumnIndex; TextType: TVSTTextType;

  var CellText: WideString);

var

  accountInfo: PAccountInfo;

begin

  accountInfo := Sender.GetNodeData(Node);

  case Column of

    1: CellText := accountInfo^.Username;

    2: CellText := GetAccessLevelString(accountInfo^.AccessLevel);

  else

    CellText := '';

  end;

end;

procedure TfrmAccountControl.OnModifyUserResponse(ABuffer: TEnhancedMemoryStream);

var

  node: PVirtualNode;

  modifyStatus: TModifyUserStatus;

  username: string;

  accountInfo: PAccountInfo;

begin

  modifyStatus := TModifyUserStatus(ABuffer.ReadByte);

  username := ABuffer.ReadStringNull;

  case modifyStatus of

    muAdded:

      begin

        node := vstAccounts.AddChild(nil);

        accountInfo := vstAccounts.GetNodeData(node);

        accountInfo^.Username := username;

        accountInfo^.AccessLevel := TAccessLevel(ABuffer.ReadByte);

        Messagedlg('Success', Format('The user "%s" has been added.', [username]), mtInformation, [mbOK], 0);

      end;

    muModified:

      begin

        node := FindNode(username);

        if node <> nil then

        begin

          accountInfo := vstAccounts.GetNodeData(node);

          accountInfo^.AccessLevel := TAccessLevel(ABuffer.ReadByte);

          Messagedlg('Success', Format('The user "%s" has been modified.', [username]), mtInformation, [mbOK], 0);

        end;

      end;

    muInvalidUsername:

      MessageDlg('Error', Format('The username "%s" is not valid.', [username]), mtError, [mbOK], 0);

  end;

end;

procedure TfrmAccountControl.OnDeleteUserResponse(ABuffer: TEnhancedMemoryStream);

var

  node: PVirtualNode;

  deleteStatus: TDeleteUserStatus;

  username: string;

begin

  deleteStatus := TDeleteUserStatus(ABuffer.ReadByte);

  username := ABuffer.ReadStringNull;

  case deleteStatus of

    duDeleted:

      begin

        node := FindNode(username);

        if node <> nil then

        begin

          vstAccounts.DeleteNode(node);

          Messagedlg('Success', Format('The user "%s" has been deleted.', [username]), mtInformation, [mbOK], 0);

        end;

      end;

    duNotFound:

      MessageDlg('Error', Format('The user "%s" could not be deleted. Maybe your list is out of date or you tried to delete yourself.', [username]), mtError, [mbOK], 0);

  end;

end;

procedure TfrmAccountControl.OnListUsersPacket(ABuffer: TEnhancedMemoryStream);

var

  node: PVirtualNode;

  accountInfo: PAccountInfo;

  i, count: Word;

begin

  vstAccounts.BeginUpdate;

  vstAccounts.Clear;

  count := ABuffer.ReadWord;

  for i := 1 to count do

  begin

    node := vstAccounts.AddChild(nil);

    accountInfo := vstAccounts.GetNodeData(node);

    accountInfo^.Username := ABuffer.ReadStringNull;

    accountInfo^.AccessLevel := TAccessLevel(ABuffer.ReadByte);

  end;

  vstAccounts.EndUpdate;

end;

function TfrmAccountControl.FindNode(AUsername: string): PVirtualNode;

var

  node: PVirtualNode;

  accountInfo: PAccountInfo;

begin

  Result := nil;

  node := vstAccounts.GetFirst;

  while (node <> nil) and (Result = nil) do

  begin

    accountInfo := vstAccounts.GetNodeData(node);

    if accountInfo^.Username = AUsername then

      Result := node;

    node := vstAccounts.GetNext(node);

  end;

end;

initialization

  {$I UfrmAccountControl.lrs}

end.