posh-wakatime.psm1

# wakatime for powershell
#
# include this file in your `$profile`
#
# Don't forget to create and configure your "~/.wakatime.cfg" file.
#
# based on posh-git, bash-wakatime, powershell-wakatime and conda(powershell)

if (Get-Module posh-wakatime) { return }

Import-Module posh-git

## PROMPT MANAGEMENT ###########################################################

<#
    .SYNOPSIS
        Modifies the current prompt to send wakatime a tick.
    .EXAMPLE
        Send-HeartbeatAtPrompt

        Causes a WakaTime heartbeat sent at the current session's prompt.
#>

# Use the same procedure as in conda to nest prompts.
if (Test-Path Function:\prompt) {
    Rename-Item Function:\prompt WakaTimePromptBackup
} else {
    function WakaTimePromptBackup() {
        # Restore a basic prompt if the definition is missing.
        "PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) ";
    }
}

function Test-Wakatime {return $(if($(where.exe wakatime)) {$True} else {$False})}

if (!$env:wakaDebug) {$env:wakaDebug = $True}

function Send-WakaTimeHeartbeat(){
    if (!$(where.exe wakatime)) {return;}
    $command = "";
    try {
        $historyItem = (Get-History | Select-Object -Last 1)
        $commandObj = ($historyItem | Select-Object -Property CommandLine).CommandLine
        $commandText = ([regex]::split($commandObj,"[ |;:]")[0])
        $command = $commandText.Replace("(","")
    } catch [Exception] {
        if($command -eq "") {
            $command = "error"
        }
    }
    $gitFolder = (Get-GitDirectory);
    Get-Job -State Completed|Where-Object{$_.Name.Contains("WakaJob")}|Remove-Job
    $job = Start-Job -Name "WakaJob" -ScriptBlock {
        param($command, $gitFolder)

        Write-Host "Sending wakatime"
        Write-Host "Waka command: $command"
        if($command -eq "") {return;}

        $PLUGIN_VERSION = "0.1";

        $wakaCommand = 'wakatime --write'
        $wakaCommand+= " --plugin `"powershell-wakatime-gravifer-plugin/$PLUGIN_VERSION`""
        $wakaCommand+= ' --entity-type app '
        $wakaCommand+= ' --entity "'
        $wakaCommand+=  $command
        $wakaCommand+= '" '
        $wakaCommand+= ' --language "PowerShell" '

        if($null -eq $gitFolder){
            $wakaCommand =$wakaCommand + ' --project '
            $wakaCommand =$wakaCommand + 'Terminal'
        } else {
            $gitFolder = (Get-Item ($gitFolder).Replace(".git",""))
            $wakaCommand =$wakaCommand + ' --project "'
            $wakaCommand =$wakaCommand + $gitFolder.Name
            $wakaCommand =$wakaCommand + '" '
        }
        Write-Host "wakaDebug: $Env:wakaDebug"
        $wakaCommand
        if($Env:wakaDebug){
            $wakaCommand | Out-File ~/.posh-wakatime.log -Append
        }
        Invoke-Expression $wakaCommand
    } -ArgumentList $command, $gitFolder
}

function Send-HeartbeatAtPrompt() {
    function global:prompt() {
        Send-WakaTimeHeartbeat
        # If (Test-Wakatime) {
        #     $wakaBgColor = "Gray"
        #     $wakaFgColor = "White"
        #     $openSymbol = "{"
        #     $closeSymbol = "}"
    
        #     $Env:WAKATIME_PROMPT_MODIFIER += Write-Prompt "$openSymbol " `
        #     -BackgroundColor $wakaBgColor
    
        #     $Env:WAKATIME_PROMPT_MODIFIER +=Write-Prompt ("W") `
        #     -ForegroundColor $wakaFgColor `
        #     -BackgroundColor $wakaBgColor
    
        #     if (Get-GitDirectory) {
        #         $Env:WAKATIME_PROMPT_MODIFIER +=Write-Prompt ("P") `
        #         -ForegroundColor $wakaFgColor `
        #         -BackgroundColor $wakaBgColor
        #     }
        #     $Env:WAKATIME_PROMPT_MODIFIER += Write-Prompt -Object "$closeSymbol " `
        #     -BackgroundColor $wakaBgColor
        # }
        # if ($Env:WAKATIME_PROMPT_MODIFIER) {
        #     $Env:WAKATIME_PROMPT_MODIFIER | Write-Host -NoNewline
        # }
        WakaTimePromptBackup;
    }
}

Send-HeartbeatAtPrompt