initial commit

ModpackUpdater/UpdateInstaller.vb

@@ -0,0 +1,153 @@
+Imports System.IO
+Imports System.Net
+
+Imports WebDav
+
+Public Class UpdateInstaller
+
+    Public Event InstallProgessUpdated(result As UpdateCheckResult, processedSyncs As Integer)
+    Public Event CheckingProgressUpdated(toCheck As Integer, processed As Integer)
+
+    Private ReadOnly updateConfig As UpdateConfig
+    Private ReadOnly localPath As String
+    Private webdavClient As WebDavClient = Nothing
+
+    Public Sub New(updateConfig As UpdateConfig, localPath As String)
+        Me.updateConfig = updateConfig
+        Me.localPath = localPath
+    End Sub
+
+    Private Function CreateClient() As WebDavClient
+        If webdavClient Is Nothing Then
+            Dim params As New WebDavClientParams With {
+                .BaseAddress = New Uri(updateConfig.WebdavURL),
+                .Credentials = New NetworkCredential(updateConfig.WebdavUsername, updateConfig.WebdavPassword)
+            }
+            webdavClient = New WebDavClient(params)
+        End If
+
+        Return webdavClient
+    End Function
+
+    Public Async Function CheckForUpdates(ignoreRevmoedFiles As Boolean) As Task(Of UpdateCheckResult)
+        Dim result As New UpdateCheckResult
+        Dim client As WebDavClient = CreateClient()
+        Dim resTopFolder As WebDavResource = Nothing
+        Dim checkedFiles = New List(Of String)()
+        Dim responseSuccessfull As Boolean = False
+
+        'Check for new & updated files
+        Dim response = Await client.Propfind(String.Empty, New PropfindParameters() With {.ApplyTo = ApplyTo.Propfind.ResourceAndAncestors})
+        If resTopFolder Is Nothing AndAlso response.IsSuccessful AndAlso response.Resources.Any() Then
+            resTopFolder = response.Resources.ElementAtOrDefault(0)
+            responseSuccessfull = True
+        End If
+
+        If responseSuccessfull AndAlso response.Resources.Count > 1 Then
+            Dim i As Integer = 1
+            Dim loopTo As Integer = response.Resources.Count - 1
+
+            Do While i <= loopTo
+                Dim res = response.Resources.ElementAtOrDefault(i)
+                Dim isFolder As Boolean = res.Uri.EndsWith("/")
+
+                If Not isFolder Then
+                    Dim syncAction? As UpdateSyncAction = Nothing
+                    Dim localFile As String
+                    Dim remoteFile As String
+
+                    'Get remote file path
+                    remoteFile = res.Uri
+
+                    'Get local file path
+                    localFile = Path.Combine(localPath, Uri.UnescapeDataString(res.Uri.Substring(resTopFolder.Uri.Length)).Replace("/", "\"))
+
+                    'Check action
+                    If File.Exists(localFile) Then
+                        If File.GetLastWriteTime(localFile) < res.LastModifiedDate = True Then
+                            syncAction = UpdateSyncAction.UpdatedFile
+                        End If
+                    Else
+                        syncAction = UpdateSyncAction.NewFile
+                    End If
+
+                    'Add to list
+                    checkedFiles.Add(localFile)
+                    If syncAction IsNot Nothing Then
+                        result.SyncFiles.Add(New UpdateSyncFile(syncAction, localFile, remoteFile))
+                    End If
+                End If
+
+                i += 1
+
+                RaiseEvent CheckingProgressUpdated(loopTo, i)
+            Loop
+        End If
+
+        'Find all old files to remove
+        If responseSuccessfull Then
+            Dim allLocalFiles = Directory.GetFiles(localPath, "*", SearchOption.AllDirectories)
+            For Each lf As String In allLocalFiles
+                Dim isKnown As Boolean = False
+
+                For Each checkedFile As String In checkedFiles
+                    If Not isKnown AndAlso If(checkedFile, "") = If(lf, "") Then
+                        isKnown = True
+                    End If
+                Next
+
+                If Not isKnown Then
+                    result.SyncFiles.Add(New UpdateSyncFile(If(ignoreRevmoedFiles, UpdateSyncAction.None, UpdateSyncAction.RemovedFile), lf, String.Empty))
+                End If
+            Next
+        End If
+
+        Return result
+    End Function
+
+    Public Async Function InstallUpdates(checkResult As UpdateCheckResult, Optional ignoreActions As UpdateSyncAction = UpdateSyncAction.None) As Task(Of Boolean?)
+        Dim client As WebDavClient = CreateClient()
+        Dim success As Boolean? = False
+        Dim processed As Integer = 0
+
+        'Init process update
+        RaiseEvent InstallProgessUpdated(checkResult, processed)
+
+        For Each syncFile In checkResult.SyncFiles
+            If syncFile.SyncAction <> (ignoreActions And syncFile.SyncAction) Then
+                If syncFile.SyncAction = UpdateSyncAction.UpdatedFile OrElse
+                   syncFile.SyncAction = UpdateSyncAction.RemovedFile Then
+
+                    If File.Exists(syncFile.LocalFile) Then
+                        File.Delete(syncFile.LocalFile)
+                    End If
+                End If
+
+                If syncFile.SyncAction = UpdateSyncAction.UpdatedFile OrElse
+                   syncFile.SyncAction = UpdateSyncAction.NewFile Then
+
+                    Dim response = Await client.GetProcessedFile(syncFile.RemoveFile)
+                    Dim dirParent As New DirectoryInfo(Path.GetDirectoryName(syncFile.LocalFile))
+
+                    If response.IsSuccessful Then
+                        If Not dirParent.Exists Then
+                            dirParent.Create()
+                        End If
+
+                        Dim fs As New FileStream(syncFile.LocalFile, FileMode.Create, FileAccess.Write)
+                        response.Stream.CopyTo(fs)
+                        fs.Close()
+
+                        success = True
+                    End If
+                End If
+            End If
+
+            processed += 1
+            RaiseEvent InstallProgessUpdated(checkResult, processed)
+        Next
+
+        Return success
+    End Function
+
+End Class