Changeset 12 for trunk/UCore.pas

Timestamp:

Mar 15, 2013, 11:42:38 PM (11 years ago)

Author:

chronos

Message:

• Přidáno: Podpora pro více účtů.

File:

• trunk/UCore.pas (modified) (17 diffs)

trunk/UCore.pas

@@ -8 +8 @@
   Classes, SysUtils, FileUtil, ExtCtrls, Controls, ActnList, Menus, UFioAPI,
   URegistry, UApplicationInfo, UCoolTranslator, Registry, DateUtils, Forms,
-  Dialogs;
+  Dialogs, SpecializedList, dom, XMLRead, XMLWrite;
 
 type
+
+  { TAccount }
+
+  TAccount = class
+    Token: string;
+    Name: string;
+    Number: string;
+    Balance: Double;
+    Time: TDateTime;
+    procedure Assign(Source: TAccount);
+    procedure LoadFromRegistry(Context: TRegistryContext);
+    procedure SaveToRegistry(Context: TRegistryContext);
+  end;
+
+  { TAccountList }
+
+  TAccountList = class(TListObject)
+    procedure LoadFromRegistry(Context: TRegistryContext);
+    procedure SaveToRegistry(Context: TRegistryContext);
+    procedure LoadToStrings(Strings: TStrings);
+    procedure Assign(Source: TAccountList);
+  end;
 
   { TCore }
@@ -16 +38 @@
   TCore = class(TDataModule)
     AAbout: TAction;
+    AAccounts: TAction;
     ActionList1: TActionList;
     ADownloadInterval: TAction;
@@ -36 +59 @@
     PopupMenuTray: TPopupMenu;
     TrayIcon1: TTrayIcon;
+    procedure AAccountsExecute(Sender: TObject);
     procedure ADownloadIntervalExecute(Sender: TObject);
     procedure ADownloadMonthlyExecute(Sender: TObject);
@@ -49 +73 @@
   public
     RegistryContext: TRegistryContext;
-    Token: string;
+    Accounts: TAccountList;
+    CurrentAccount: TAccount;
     DataFormat: TFioDataFormat;
     TargetDirectory: string;
@@ -56 +81 @@
     ReportTimeFrom: TDateTime;
     ReportTimeTo: TDateTime;
+    procedure LoadAccount(Account: TAccount);
+    procedure LoadAccounts;
     procedure SaveToRegistry(Context: TRegistryContext);
     procedure LoadFromRegistry(Context: TRegistryContext);
@@ -68 +95 @@
 
 uses
-  UFormAbout, UFormMain, UFormSettings;
+  UFormAbout, UFormMain, UFormSettings, UFormAccounts;
 
 resourcestring
@@ -75 +102 @@
   SSavedToFile = 'Dump saved to file %s';
 
+{ TAccount }
+
+procedure TAccount.Assign(Source: TAccount);
+begin
+  Token := Source.Token;
+  Name := Source.Name;
+  Balance := Source.Balance;
+  Time := Source.Time;
+  Number := Source.Number;
+end;
+
+procedure TAccount.LoadFromRegistry(Context: TRegistryContext);
+begin
+  with TRegistryEx.Create do
+  try
+    RootKey := Context.RootKey;
+    OpenKey(Context.Key, True);
+    Name := ReadStringWithDefault('Name', '');
+    Time := ReadDateTimeWithDefault('Time', 0);
+    Token := ReadStringWithDefault('Token', '');
+    Balance := ReadFloatWithDefault('Balance', 0);
+    Number := ReadStringWithDefault('Number', '');
+  finally
+    Free;
+  end;
+end;
+
+procedure TAccount.SaveToRegistry(Context: TRegistryContext);
+begin
+  with TRegistryEx.Create do
+  try
+    RootKey := Context.RootKey;
+    OpenKey(Context.Key, True);
+    WriteString('Number', Number);
+    WriteString('Name', Name);
+    WriteDateTime('Time', Time);
+    WriteString('Token', Token);
+    WriteFloat('Balance', Balance);
+  finally
+    Free;
+  end;
+end;
+
+{ TAccountList }
+
+procedure TAccountList.LoadFromRegistry(Context: TRegistryContext);
+var
+  I: Integer;
+begin
+  with TRegistryEx.Create do
+  try
+    RootKey := Context.RootKey;
+    OpenKey(Context.Key, True);
+    Count := ReadIntegerWithDefault('Count', 0);
+  finally
+    Free;
+  end;
+  for I := 0 to Count - 1 do begin
+    if not Assigned(Items[I]) then Items[I] := TAccount.Create;
+    TAccount(Items[I]).LoadFromRegistry(RegContext(Context.RootKey, Context.Key + '\' + IntToStr(I)));
+  end;
+end;
+
+procedure TAccountList.SaveToRegistry(Context: TRegistryContext);
+var
+  I: Integer;
+begin
+  with TRegistryEx.Create do
+  try
+    RootKey := Context.RootKey;
+    OpenKey(Context.Key, True);
+    WriteInteger('Count', Count);
+  finally
+    Free;
+  end;
+  for I := 0 to Count - 1 do
+    TAccount(Items[I]).SaveToRegistry(RegContext(Context.RootKey, Context.Key + '\' + IntToStr(I)));
+end;
+
+procedure TAccountList.LoadToStrings(Strings: TStrings);
+var
+  I: Integer;
+begin
+  while Strings.Count < Count do
+    Strings.Add('');
+  while Strings.Count > Count do
+    Strings.Delete(Strings.Count - 1);
+  for I := 0 to Count - 1 do
+    Strings.Strings[I] := TAccount(Items[I]).Name;
+end;
+
+procedure TAccountList.Assign(Source: TAccountList);
+var
+  I: Integer;
+begin
+  //inherited Assign(Source);
+  Count := Source.Count;
+  for I := 0 to Count - 1 do begin
+    if not Assigned(Items[I]) then Items[I] := TAccount.Create;
+    TAccount(Items[I]).Assign(TAccount(Source.Items[I]));
+  end;
+end;
+
 { TCore }
 
@@ -84 +214 @@
 procedure TCore.DataModuleCreate(Sender: TObject);
 begin
+  Accounts := TAccountList.Create;
   RegistryContext := RegContext(HKEY(ApplicationInfo1.RegistryRoot),
     ApplicationInfo1.RegistryKey);
   LoadFromRegistry(RegistryContext);
+  //LoadAccounts;
 end;
 
@@ -92 +224 @@
 begin
   SaveToRegistry(RegistryContext);
+  FreeAndNil(Accounts);
+end;
+
+procedure TCore.LoadAccount(Account: TAccount);
+var
+  FioAPI: TFioAPI;
+  List: TStringList;
+  FileName: string;
+  XMLDocument: TXMLDocument;
+  Mem: TMemoryStream;
+  RootNode: TDOMNode;
+begin
+  FioAPI := TFioAPI.Create;
+  List := TStringList.Create;
+  XMLDocument := TXMLDocument.Create;
+  Mem := TMemoryStream.Create;
+  try
+    FioAPI.Format := DataFormat;
+    FioAPI.Token := Account.Token;
+    Account.Time := Now;
+    if FioAPI.DownloadInterval(Now, Now, List) then begin
+      List.SaveToStream(Mem);
+      ReadXMLFile(XMLDocument, Mem);
+      RootNode := XMLDocument.DocumentElement;
+      ShowMessage(RootNode.NodeName);
+    end else begin
+      Account.Number := '';
+      Account.Balance := 0;
+    end;
+  finally
+    Mem.Free;
+    XMLDocument.Free;
+    FioAPI.Free;
+    List.Free;
+  end;
+end;
+
+procedure TCore.LoadAccounts;
+var
+  I: Integer;
+begin
+  for I := 0 to Accounts.Count - 1 do
+    LoadAccount(TAccount(Accounts[I]));
 end;
 
@@ -108 +283 @@
     WriteString('TargetDir', TargetDirectory);
     WriteInteger('DataFormat', Integer(DataFormat));
-    WriteString('Token', Token);
     WriteDate('ReportTimeFrom', ReportTimeFrom);
     WriteDateTime('ReportTimeTo', ReportTimeTo);
@@ -117 +291 @@
     Free;
   end;
+  Accounts.SaveToRegistry(RegContext(Context.RootKey, Context.Key + '\Account'));
 end;
 
@@ -127 +302 @@
     TargetDirectory := ReadStringWithDefault('TargetDir', ExtractFileDir(Application.ExeName));
     DataFormat := TFioDataFormat(ReadIntegerWithDefault('DataFormat', Integer(dfXML)));
-    Token := ReadStringWithDefault('Token', '');
     ReportTimeFrom := ReadDateTimeWithDefault('ReportTimeFrom', Now);
     ReportTimeTo := ReadDateTimeWithDefault('ReportTimeTo', Now);
@@ -136 +310 @@
     Free;
   end;
+  Accounts.LoadFromRegistry(RegContext(Context.RootKey, Context.Key + '\Account'));
 end;
 
@@ -162 +337 @@
   try
     FioAPI.Format := DataFormat;
-    FioAPI.Token := Token;
+    FioAPI.Token := CurrentAccount.Token;
     if FioAPI.DownloadLast(List) then begin
       FileName := TargetDirectory + DirectorySeparator + Format(SReport, [FormatDateTime('yyyy-mm-dd hh-nn-ss', Now) +
@@ -185 +360 @@
   try
     FioAPI.Format := DataFormat;
-    FioAPI.Token := Token;
+    FioAPI.Token := CurrentAccount.Token;
     if FioAPI.DownloadMonthly(ReportYear, ReportId, List) then begin
       FileName := TargetDirectory + DirectorySeparator + Format(SReport, [IntToStr(ReportYear) +
@@ -209 +384 @@
   try
     FioAPI.Format := DataFormat;
-    FioAPI.Token := Token;
+    FioAPI.Token := CurrentAccount.Token;
     if FioAPI.DownloadInterval(ReportTimeFrom, ReportTimeTo, List) then begin
       FileName := TargetDirectory + DirectorySeparator + Format(SReport, [FormatDateTime('yyyy-mm-dd', ReportTimeFrom) +
@@ -223 +398 @@
 end;
 
+procedure TCore.AAccountsExecute(Sender: TObject);
+begin
+  TAccountList(FormAccounts.Accounts).Assign(Accounts);
+  if FormAccounts.ShowModal = mrOk then begin
+    Accounts.Assign(TAccountList(FormAccounts.Accounts));
+  end;
+end;
+
 end.