add functionality to respond to pings
parent
fe23666fc0
commit
4fe1cdddcb
@ -1,70 +1,224 @@
|
|||||||
param (
|
param (
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$HomeServer,
|
$HomeServer,
|
||||||
|
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$User,
|
$User,
|
||||||
|
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[securestring]
|
[securestring]
|
||||||
$AccessToken
|
$AccessToken
|
||||||
)
|
)
|
||||||
|
|
||||||
function Send-MatrixEvent {
|
function Send-MatrixEvent {
|
||||||
param (
|
param (
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$RoomId,
|
$RoomId,
|
||||||
|
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$Event,
|
$Event,
|
||||||
|
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$EventType
|
$EventType
|
||||||
)
|
)
|
||||||
|
|
||||||
#txn_id should be unique per client, so we use timestamp+random
|
#txn_id should be unique per client, so we use timestamp+random
|
||||||
$txn_id = '{0}{1}' -f (Get-Date -UFormat '%s'),(Get-Random)
|
$txn_id = '{0}{1}' -f (Get-Date -UFormat '%s'),(Get-Random)
|
||||||
|
|
||||||
$uri = '{0}/_matrix/client/r0/rooms/{1}/send/{2}/{3}' -f $HomeServer,$RoomId,$EventType,$txn_id
|
$uri = '{0}/_matrix/client/r0/rooms/{1}/send/{2}/{3}' -f $HomeServer,$RoomId,$EventType,$txn_id
|
||||||
|
|
||||||
$header_splat = @{
|
$header_splat = @{
|
||||||
Authentication = 'Bearer'
|
Authentication = 'Bearer'
|
||||||
Token = $AccessToken
|
Token = $AccessToken
|
||||||
ContentType = 'application/json'
|
ContentType = 'application/json'
|
||||||
}
|
}
|
||||||
$http_splat = @{
|
$http_splat = @{
|
||||||
Uri = $uri
|
Uri = $uri
|
||||||
Method = 'Put'
|
Method = 'Put'
|
||||||
Body = $Event
|
Body = $Event
|
||||||
}
|
}
|
||||||
|
|
||||||
$response = Invoke-RestMethod @header_splat @http_splat
|
$response = Invoke-RestMethod @header_splat @http_splat
|
||||||
if($response.event_id) {
|
if($response.event_id) {
|
||||||
Write-Output ('Event {0} sent to room {1}' -f $response.event_id,$RoomId)
|
Write-Host ('Event {0} sent to room {1}' -f $response.event_id,$RoomId)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
function Send-MatrixNotice {
|
function Send-MatrixNotice {
|
||||||
param (
|
param (
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$RoomId,
|
$RoomId,
|
||||||
|
|
||||||
[Parameter(Mandatory=$true)]
|
[Parameter(Mandatory=$true)]
|
||||||
[string]
|
[string]
|
||||||
$Message
|
$Body,
|
||||||
)
|
|
||||||
|
[Parameter(Mandatory=$false)]
|
||||||
$event = @{
|
[string]
|
||||||
msgtype = 'm.notice'
|
$FormattedBody
|
||||||
body = $Message
|
)
|
||||||
} | ConvertTo-Json -Compress
|
|
||||||
|
$event = @{
|
||||||
Send-MatrixEvent -RoomId $RoomId -Event $event -EventType 'm.room.message'
|
msgtype = 'm.notice'
|
||||||
}
|
body = $Body
|
||||||
|
}
|
||||||
Send-MatrixNotice -RoomId '!amKUIaaKdERatZsJgB:matrix.org' -Message ('asef :) {0}' -f (Get-Date))
|
|
||||||
|
if($FormattedBody) {
|
||||||
|
$event += @{
|
||||||
|
format = 'org.matrix.custom.html'
|
||||||
|
formatted_body = $FormattedBody
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
$event_json = $event | ConvertTo-Json -Compress
|
||||||
|
|
||||||
|
Send-MatrixEvent -RoomId $RoomId -Event $event_json -EventType 'm.room.message'
|
||||||
|
}
|
||||||
|
function Compare-Timestamps {
|
||||||
|
param (
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
[Int64]
|
||||||
|
$OriginServerTs
|
||||||
|
)
|
||||||
|
|
||||||
|
$current_time = (Get-Date).ToUniversalTime()
|
||||||
|
|
||||||
|
#$OriginServerTs is milliseconds since 1970-01-01 (epoch time in milliseconds)
|
||||||
|
$original_time = Get-Date -Date ((Get-Date -Date '1970-01-01') + [timespan]::FromMilliseconds($OriginServerTs))
|
||||||
|
|
||||||
|
#return the difference
|
||||||
|
Write-Output ($current_time - $original_time)
|
||||||
|
}
|
||||||
|
function ConvertTo-HumanReadableTimespan {
|
||||||
|
param (
|
||||||
|
[parameter(Mandatory=$true)]
|
||||||
|
[timespan]
|
||||||
|
$TimeSpan
|
||||||
|
)
|
||||||
|
|
||||||
|
switch($TimeSpan) {
|
||||||
|
Default {
|
||||||
|
$output = '{0} days' -f [System.Math]::Round($TimeSpan.TotalDays, 2)
|
||||||
|
}
|
||||||
|
{$PSItem.TotalHours -lt 24} {
|
||||||
|
$output = '{0} h' -f [System.Math]::Round($TimeSpan.TotalHours, 2)
|
||||||
|
}
|
||||||
|
{$PSItem.TotalMinutes -lt 120} {
|
||||||
|
$output = '{0} min' -f [System.Math]::Round($TimeSpan.TotalMinutes, 2)
|
||||||
|
}
|
||||||
|
{$PSItem.TotalSeconds -lt 120} {
|
||||||
|
$output = '{0} s' -f [System.Math]::Round($TimeSpan.TotalSeconds, 2)
|
||||||
|
}
|
||||||
|
{$PSItem.TotalMilliseconds -lt 10000} {
|
||||||
|
$output = '{0} ms' -f [System.Math]::Round($TimeSpan.TotalMilliseconds, 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Write-Output $output
|
||||||
|
}
|
||||||
|
function Join-Pong {
|
||||||
|
param (
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
[string]
|
||||||
|
$RoomId,
|
||||||
|
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
[string]
|
||||||
|
$PingEventId,
|
||||||
|
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
[string]
|
||||||
|
$SenderMxId,
|
||||||
|
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
[string]
|
||||||
|
$ReadableTimespan,
|
||||||
|
|
||||||
|
[Parameter(Mandatory=$false)]
|
||||||
|
[string]
|
||||||
|
$Ball
|
||||||
|
)
|
||||||
|
|
||||||
|
if($Ball) {
|
||||||
|
$padded_ball = $Ball+' '
|
||||||
|
}
|
||||||
|
|
||||||
|
$body = '{0}: Pong! (ping {1}took {2} to arrive)' -f $SenderMxId,$padded_ball,$ReadableTimespan
|
||||||
|
$formatted_body = '<a href=''https://matrix.to/#/{0}''>{0}</a>: Pong! ' -f $SenderMxId
|
||||||
|
$formatted_body += '(<a href="https://matrix.to/#/{0}/{1}">ping</a> {2}took {3} to arrive)' -f $RoomId,$PingEventId,$padded_ball,$ReadableTimespan
|
||||||
|
|
||||||
|
return @{
|
||||||
|
Body = $body
|
||||||
|
FormattedBody = $formatted_body
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function Open-MatrixEvent {
|
||||||
|
param (
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
$Event,
|
||||||
|
|
||||||
|
[Parameter(Mandatory=$true)]
|
||||||
|
$RoomId
|
||||||
|
)
|
||||||
|
|
||||||
|
#the "ball" is a string returned by the bot
|
||||||
|
if($Event.content.msgtype -eq 'm.text' -and $Event.content.body -match '!ping( (?<ball>.*))?') {
|
||||||
|
$difference = Compare-Timestamps -OriginServerTs $Event.origin_server_ts
|
||||||
|
$readable_timespan = ConvertTo-HumanReadableTimespan -TimeSpan $difference
|
||||||
|
|
||||||
|
#$bodies contains a hashtable with keys Body and FormattedBody
|
||||||
|
$bodies = Join-Pong -RoomId $RoomId -PingEventId $Event.event_id -SenderMxId $Event.sender -ReadableTimespan $readable_timespan -Ball $Matches.ball
|
||||||
|
|
||||||
|
Send-MatrixNotice -RoomId $RoomId @bodies
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function Start-MatrixSync {
|
||||||
|
param (
|
||||||
|
[string]
|
||||||
|
$Token,
|
||||||
|
|
||||||
|
[int]
|
||||||
|
$Timeout = 30000
|
||||||
|
)
|
||||||
|
|
||||||
|
$uri = '{0}/_matrix/client/r0/sync?timeout={1}' -f $HomeServer,$Timeout
|
||||||
|
if($Token) {
|
||||||
|
$uri += '&since={0}' -f $Token
|
||||||
|
}
|
||||||
|
|
||||||
|
$header_splat = @{
|
||||||
|
Authentication = 'Bearer'
|
||||||
|
Token = $AccessToken
|
||||||
|
ContentType = 'application/json'
|
||||||
|
}
|
||||||
|
$http_splat = @{
|
||||||
|
Uri = $uri
|
||||||
|
}
|
||||||
|
|
||||||
|
$response = Invoke-RestMethod @header_splat @http_splat
|
||||||
|
|
||||||
|
#.PSObject.Properties because the rooms under .join are [NoteProperty]
|
||||||
|
$room_ids = $response.rooms.join.PSObject.Properties.Name
|
||||||
|
foreach($room_id in $room_ids) {
|
||||||
|
$events = $response.rooms.join.$room_id.timeline.events
|
||||||
|
foreach($event in $events) {
|
||||||
|
Open-MatrixEvent -Event $event -RoomId $room_id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Write-Output $response.next_batch
|
||||||
|
}
|
||||||
|
|
||||||
|
#Send-MatrixNotice -RoomId '!amKUIaaKdERatZsJgB:matrix.org' -Message ('asef :) {0}' -f (Get-Date))
|
||||||
|
|
||||||
|
#start sync loop
|
||||||
|
while($true) {
|
||||||
|
#use the token of the last sync
|
||||||
|
#initial token is $null
|
||||||
|
$token = Start-MatrixSync -Token $token
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue