setup.iss

; Inno Setup script for Infrastructure Agent
#define MyAppName "Infrastructure Agent"
#define MyAppVersion "1.0"
#define MyAppPublisher "thien.dinh-infra"
#define MyAppURL "https://10.1.32.66"
#define MyAppExeName "InfraAgent.exe"

[Setup]
AppId={{8B4D8F1D-4F4E-4A26-B8E4-6A7D8E1D4F2A}}
AppName={#MyAppName}
AppVersion={#MyAppVersion}
AppPublisher={#MyAppPublisher}
AppPublisherURL={#MyAppURL}
AppSupportURL={#MyAppURL}
AppUpdatesURL={#MyAppURL}

DefaultDirName={autopf}\{#MyAppName}
DisableProgramGroupPage=yes
PrivilegesRequired=admin
OutputDir=.
OutputBaseFilename=InfraAgentSetup
SetupIconFile=agentlogo.ico
UninstallDisplayIcon={app}\agentlogo.ico
Compression=lzma
SolidCompression=yes
WizardStyle=modern

[Languages]
Name: "english"; MessagesFile: "compiler:Default.isl"

[Tasks]
Name: "startup"; Description: "Run at Windows startup for all users"; GroupDescription: "Startup settings:"
Name: "desktopicon"; Description: "Create a desktop icon"; GroupDescription: "Additional icons:"

[Files]
Source: "dist\*"; DestDir: "{app}"; Flags: ignoreversion recursesubdirs createallsubdirs
Source: "agentlogo.ico"; DestDir: "{app}"; Flags: ignoreversion

[Icons]
Name: "{autoprograms}\{#MyAppName}"; Filename: "{app}\{#MyAppExeName}"
Name: "{autodesktop}\{#MyAppName}"; Filename: "{app}\{#MyAppExeName}"; Tasks: desktopicon

[Run]
Filename: "{app}\{#MyAppExeName}"; Description: "{cm:LaunchProgram,{#MyAppName}}"; Flags: nowait postinstall skipifsilent

[UninstallDelete]
Type: filesandordirs; Name: "{app}"

[Code]
var
  ServerPage: TInputQueryWizardPage;
  CredentialsPage: TInputQueryWizardPage;

procedure InitializeWizard;
begin
  ServerPage := CreateInputQueryPage(wpSelectTasks,
    'Server Configuration', 'Enter server connection details',
    'Please specify the server URL where the agent will send data.');
  ServerPage.Add('Server URL:', False);

  CredentialsPage := CreateInputQueryPage(ServerPage.ID,
    'Login Credentials', 'Enter authentication details',
    'Please specify the email and password for agent authentication.');
  CredentialsPage.Add('Email:', False);
  CredentialsPage.Add('Password:', True);
  
  ServerPage.Values[0] := 'https://10.1.32.66';
  CredentialsPage.Values[0] := 'infraagent@localhost.com';
  CredentialsPage.Values[1] := 'Infraagent@2025';
end;

function NextButtonClick(CurPageID: Integer): Boolean;
begin
  Result := True;
end;

function IsAppRunning(const FileName : string): Boolean;
var
  FSWbemLocator: Variant;
  FWMIService: Variant;
  FWbemObjectSet: Variant;
begin
  Result := false;
  try
    FSWbemLocator := CreateOleObject('WbemScripting.SWbemLocator');
    FWMIService := FSWbemLocator.ConnectServer('', 'root\CIMV2', '', '');
    FWbemObjectSet := FWMIService.ExecQuery(Format('SELECT Name FROM Win32_Process Where Name="%s"', [FileName]));
    Result := (FWbemObjectSet.Count > 0);
    FWbemObjectSet := Unassigned;
    FWMIService := Unassigned;
    FSWbemLocator := Unassigned;
  except
    Result := false;
  end;
end;

function StopApp(const FileName : string): Boolean;
var
  FSWbemLocator: Variant;
  FWMIService: Variant;
  FWbemObject: Variant;
  FWbemObjectSet: Variant;
begin
  Result := false;
  try
    FSWbemLocator := CreateOleObject('WbemScripting.SWbemLocator');
    FWMIService := FSWbemLocator.ConnectServer('', 'root\CIMV2', '', '');
    FWbemObjectSet := FWMIService.ExecQuery(Format('SELECT * FROM Win32_Process Where Name="%s"', [FileName]));
    if FWbemObjectSet.Count > 0 then
    begin
      FWbemObject := FWbemObjectSet.ItemIndex(0);
      Result := (FWbemObject.Terminate(0) = 0);
    end;
    FWbemObjectSet := Unassigned;
    FWbemObject := Unassigned;
    FWMIService := Unassigned;
    FSWbemLocator := Unassigned;
  except
    Result := false;
  end;
end;

function UpdateConfigFile(const ConfigPath, ServerURL, Email, Password: String): Boolean;
var
  Lines: TArrayOfString;
  Content: String;
  I: Integer;
  p: Integer;
begin
  Result := False;
  
  // Load file content using the correct function
  if LoadStringsFromFile(ConfigPath, Lines) then
  begin
    // Join all lines into a single string
    Content := '';
    for I := 0 to GetArrayLength(Lines) - 1 do
    begin
      Content := Content + Lines[I] + #13#10;
    end;
    
    // Simple string replacement without using StringReplace function
    // Replace server URL
    p := Pos('"https://10.1.32.66"', Content);
    while p > 0 do
    begin
      Delete(Content, p, 21); // Length of '"https://10.1.32.66"'
      Insert('"' + ServerURL + '"', Content, p);
      p := Pos('"https://10.1.32.66"', Content);
    end;
    
    // Replace email
    p := Pos('"infraagent@localhost.com"', Content);
    while p > 0 do
    begin
      Delete(Content, p, 27); // Length of '"infraagent@localhost.com"'
      Insert('"' + Email + '"', Content, p);
      p := Pos('"infraagent@localhost.com"', Content);
    end;
    
    // Replace password
    p := Pos('"Infraagent@2025"', Content);
    while p > 0 do
    begin
      Delete(Content, p, 18); // Length of '"Infraagent@2025"'
      Insert('"' + Password + '"', Content, p);
      p := Pos('"Infraagent@2025"', Content);
    end;
    
    // Save updated content
    if SaveStringToFile(ConfigPath, Content, False) then
    begin
      Result := True;
    end;
  end;
end;

procedure CurStepChanged(CurStep: TSetupStep);
var
  ConfigFile: String;
  ServerURL, Email, Password: String;
begin
  if CurStep = ssInstall then
  begin
    // Stop any running instances of the agent before installation
    if IsAppRunning('InfraAgent.exe') then
    begin
      // Try to stop the application gracefully
      StopApp('InfraAgent.exe');
      // Wait a moment for the process to terminate
      Sleep(1000);
    end;
  end
  else if CurStep = ssPostInstall then
  begin
    ServerURL := ServerPage.Values[0];
    Email := CredentialsPage.Values[0];
    Password := CredentialsPage.Values[1];
    
    // Update config file with user-provided values
    ConfigFile := ExpandConstant('{app}\config.json');
    UpdateConfigFile(ConfigFile, ServerURL, Email, Password);
    
    // Add startup shortcut if selected
    if WizardIsTaskSelected('startup') then
    begin
      CreateShellLink(
        ExpandConstant('{commonstartup}\Infrastructure Agent.lnk'),
        'Infrastructure Agent',
        ExpandConstant('{app}\InfraAgent.exe'),
        '',
        ExpandConstant('{app}'),
        ExpandConstant('{app}\agentlogo.ico'),
        0,
        SW_SHOWNORMAL);
    end;
  end;
end;

procedure CurUninstallStepChanged(CurUninstallStep: TUninstallStep);
var
  AppPath: String;
  ResultCode: Integer;
begin
  case CurUninstallStep of
    usUninstall:
      begin
        // Stop any running instances of the agent before uninstallation
        if IsAppRunning('InfraAgent.exe') then
        begin
          StopApp('InfraAgent.exe');
          Sleep(3000); // Give more time for the process to terminate
        end;
        
        // Force kill any remaining processes
        try
          Exec('taskkill', '/F /IM InfraAgent.exe', '', SW_HIDE, ewWaitUntilTerminated, ResultCode);
        except
          // Ignore any errors
        end;
        
        // Remove startup shortcut
        DeleteFile(ExpandConstant('{commonstartup}\Infrastructure Agent.lnk'));
        
        // Remove desktop shortcut
        DeleteFile(ExpandConstant('{autodesktop}\Infrastructure Agent.lnk'));
        
        // Remove programs menu shortcut
        DeleteFile(ExpandConstant('{autoprograms}\Infrastructure Agent.lnk'));
      end;
    usPostUninstall:
      begin
        // Additional cleanup can be done here if needed
        // Force removal of application directory
        AppPath := ExpandConstant('{app}');
        if DirExists(AppPath) then
        begin
          DelTree(AppPath, True, True, True);
        end;
        
        // Remove any remaining registry entries
        try
          RegDeleteKeyIncludingSubkeys(HKLM, 'SOFTWARE\{#MyAppName}');
        except
          // Ignore any errors
        end;
        
        try
          RegDeleteKeyIncludingSubkeys(HKCU, 'SOFTWARE\{#MyAppName}');
        except
          // Ignore any errors
        end;
      end;
  end;
end;

function InitializeUninstall(): Boolean;
var
  ResultCode: Integer;
begin
  // Stop any running instances of the agent before uninstallation
  if IsAppRunning('InfraAgent.exe') then
  begin
    StopApp('InfraAgent.exe');
    Sleep(3000); // Give more time for the process to terminate
  end;
  
  // Force kill any remaining processes
  try
    Exec('taskkill', '/F /IM InfraAgent.exe', '', SW_HIDE, ewWaitUntilTerminated, ResultCode);
  except
    // Ignore any errors
  end;
  
  Result := True;
end;