FastKeyboardScreenshot.ahk

; FastKeyboardScreenshot - Main Entry Point
; Create screenshots by selecting regions with keyboard navigation

#Include %A_ScriptDir%\config.ahk
#Include github_modules/Gdip/Gdip.ahk

; Include library modules (functions only - no hotkeys/labels)
#Include %A_ScriptDir%\lib\utils.ahk
#Include %A_ScriptDir%\lib\gui.ahk
#Include %A_ScriptDir%\lib\image.ahk
#Include %A_ScriptDir%\github_modules\RapidOCR-AutoHotkey\RapidOCR\RapidOCR.ahk
#Include %A_ScriptDir%\lib\capture.ahk
#Include %A_ScriptDir%\lib\crop.ahk
#Include %A_ScriptDir%\lib\arrow.ahk
#Include %A_ScriptDir%\lib\number.ahk
#Include %A_ScriptDir%\lib\rectangle.ahk
#Include %A_ScriptDir%\lib\status_bar.ahk
#Include %A_ScriptDir%\lib\FTP_Upload.ahk
#Include %A_ScriptDir%\lib\SingleInstance.ahk

; Class for single instance management
class FastKeyboardScreenshotApp {
    IsActive() {
        return 1
    }
    Quit() {
        ExitApp
    }
}

; Ensure single instance - quit if another instance is already running
CheckSingleInstance("{B8F4C2E1-7A3D-4F9E-8B5C-1D2E3F4A5B6C}", "FastKeyboardScreenshotApp")

; Initialize GDI+ once at startup (keep running for entire script lifetime)
pGdipToken := Gdip_Startup()
if (!pGdipToken) {
    MsgBox, 48, Error!, GDI+ failed to initialize
    ExitApp
}
OnExit("CleanupAll")

CoordMode, Mouse, Screen

; Initialize RapidOCR for English text
rapidOcr := new RapidOCR({model: "en"})

; Global variables for ShareX
sharexPath := ""
sharexSearched := false

; Global variables for mouse/screenshot state
mouseSpeed := 50
mouseSPeedSlow := 10
interactiveMode := 0
state := 0
delayedScreenShot := 0
screenshotTimerIndex := 0
screenShotStartX := -1
screenShotStartY := -1
screenShotEndX := -1
screenShotEndY := -1
resizeNextScreenshotBy := 1
saveToFile := 0
uploadAfterCapture := 0
editWithShareX := 0
ocrScreenshot := 0
captureCursor := 0
showWindow := 0

; Register message handler for resolution changes
OnMessage(0x007E, "HandleResolutionChange")

; Global variables for preview window
previewImagePath := ""
previewImageWidth := 0
previewImageHeight := 0
previewPBitmap := 0
previewHwnd := 0
previewTempFile := ""
previewSavedFilePath := ""  ; Track saved file for overwrite

; Global variables for crop mode
previewMode := "viewing"  ; "viewing", "crop", "arrow", "number", or "rectangle"
cropSettingStart := 0  ; 0 = setting first corner, 1 = setting second corner
cropStartX := 0
cropStartY := 0
cropEndX := 0
cropEndY := 0

; Global variables for arrow mode
arrowCursorX := 0
arrowCursorY := 0
arrowStartX := 0
arrowStartY := 0
arrowSettingStart := 0  ; 0 = not setting, 1 = setting start point
arrowSize := 3
arrowColorIndex := 0
arrowColors := [0xFFFF0000, 0xFF0000FF, 0xFF00FF00, 0xFFFFFF00, 0xFF000000]  ; red, blue, green, yellow, black
arrowColorNames := ["Red", "Blue", "Green", "Yellow", "Black"]
arrows := []  ; Array of drawn arrows
arrowMoveStep := 10  ; Pixels per keypress

; Global variables for number mode
numbers := []  ; Array of placed numbers
numberSize := 24  ; Circle diameter
numberColorIndex := 0  ; Color index (shares arrowColors palette)
nextNumber := 1  ; Next number to place on click (auto-increments)

; Global variables for rectangle mode
rectangles := []  ; Array of drawn rectangles
rectSize := 3  ; Line thickness (1-20)
rectColorIndex := 0  ; Color index (shares arrowColors palette)
rectSettingStart := 0  ; 0 = not setting, 1 = setting first corner
rectStartX := 0
rectStartY := 0

; Global variables for text preview window
textPreviewHwnd := 0

; Global variable for preview help window
previewHelpOpen := 0

; Global variables for action tooltips (FTP upload and OCR)
lastUploadedUrl := ""
pendingOcrText := ""

; Initialize ShareX path from settings or by searching
IniRead, sharexPath, %settingsFile%, General, ShareXPath, NOT_FOUND
IniRead, sharexNotFound, %settingsFile%, General, ShareXNotFound, 0

; If path is not in settings or marked as not found, search for it
if (sharexPath = "NOT_FOUND" && sharexNotFound = 0) {
    sharexPath := FindShareX()
    sharexSearched := true

    ; Store the result in settings.ini
    if (sharexPath = "") {
        ; ShareX was not found, mark it as not found
        IniWrite, 1, %settingsFile%, General, ShareXNotFound
    } else {
        ; ShareX was found, store the path
        IniWrite, %sharexPath%, %settingsFile%, General, ShareXPath
        IniWrite, 0, %settingsFile%, General, ShareXNotFound
    }
} else if (sharexNotFound = 1) {
    ; ShareX was previously not found, set path to empty
    sharexPath := ""
}

; Load arrow preferences from settings
IniRead, arrowColorIndex, %settingsFile%, Arrow, ColorIndex, 0
if (arrowColorIndex >= arrowColors.Length())
    arrowColorIndex := 0
IniRead, arrowSize, %settingsFile%, Arrow, Size, 3
if (arrowSize < 1 || arrowSize > 20)
    arrowSize := 3

; Load number preferences from settings
IniRead, numberColorIndex, %settingsFile%, Number, ColorIndex, 0
if (numberColorIndex >= arrowColors.Length())
    numberColorIndex := 0
IniRead, numberSize, %settingsFile%, Number, Size, 24
if (numberSize < 12 || numberSize > 60)
    numberSize := 24

; Load rectangle preferences from settings
IniRead, rectColorIndex, %settingsFile%, Rectangle, ColorIndex, 0
if (rectColorIndex >= arrowColors.Length())
    rectColorIndex := 0
IniRead, rectSize, %settingsFile%, Rectangle, Size, 3
if (rectSize < 1 || rectSize > 20)
    rectSize := 3

; Set tray icon
if (!a_iscompiled) {
	Menu, tray, icon, icon.ico,0,1
}

; Setup tray menu
Menu, tray, NoStandard
Menu, tray, add  ; Creates a separator line.
Menu, tray, add, Reload
Menu, tray, add, Exit

return

; Include files with hotkeys/labels AFTER auto-execute section (they end auto-execute)
#Include %A_ScriptDir%\lib\help_window.ahk
#Include %A_ScriptDir%\lib\screenshot.ahk
#Include %A_ScriptDir%\lib\preview_window.ahk
#Include %A_ScriptDir%\lib\hotkeys.ahk

Reload:
	Reload
return

Exit:
	ExitApp
return

; Action tooltip hotkeys (for FTP upload and OCR)
OpenLastUrl:
    global lastUploadedUrl, pendingOcrText
    if(lastUploadedUrl != "") {
        Run, %lastUploadedUrl%
    } else if(pendingOcrText != "") {
        ; Create temp file on-demand and open in default editor
        FormatTime, ts, , yyyy_MM_dd_HH_mm_ss
        tempTxt := A_Temp . "\ocr_" . ts . ".txt"
        FileAppend, %pendingOcrText%, %tempTxt%
        Run, %tempTxt%
        pendingOcrText := ""
    }
    GoSub, ClearActionTooltip
return

CancelActionTooltip:
    GoSub, ClearActionTooltip
return

ClearActionTooltip:
    global pendingOcrText
    ToolTip
    Hotkey, o, OpenLastUrl, Off
    Hotkey, Escape, CancelActionTooltip, Off
    SetTimer, ClearActionTooltip, Off
    pendingOcrText := ""  ; Clear pending text if user dismissed
return

; Cleanup GDI+ and RapidOCR on script exit
CleanupAll() {
    global pGdipToken, rapidOcr
    ; Terminate RapidOCR process
    if (IsObject(rapidOcr)) {
        try rapidOcr.exec.Terminate()
    }
    ; Shutdown GDI+
    if (pGdipToken)
        Gdip_Shutdown(pGdipToken)
}