PersonalWebsites_CSExtension.vb

Imports Microsoft.MetadirectoryServices

Imports System.IO

Imports System.Data.SqlClient

Imports OCG.Utility.Xml

Imports zbitz

 

Public Class MACallExport_PersonalWebsites

    Implements IMAExtensibleFileImport

    Implements IMAExtensibleCallExport

    Const DB_TABLE_NAME As String = “PersonalWebsites”

    Const DB_CONNECTION_STRING As String = “Database=MIISData;Data Source=localhost;Integrated Security=Yes;”

    Dim sqlExportConnection As New SqlConnection(DB_CONNECTION_STRING)

    Dim MAcol As New MAConfiguration(Utils.ExtensionsDirectory & “\MA_Params.xml”)

    Dim Params As MAParams = MAcol.Parameters(“PersonalWebsites-MA”)

    Dim MAName As String = Params(“ma-name”)

 

    Public Sub GenerateImportFile(ByVal filename As String, ByVal connectTo As String, ByVal user As String, ByVal password As String, ByVal configParameters As ConfigParameterCollection, ByVal fullImport As Boolean, ByVal types As TypeDescriptionCollection, ByRef customData As String) Implements IMAExtensibleFileImport.GenerateImportFile

        Dim WebRoot As String = Params(“personal-website-root”)

        Dim fw As StreamWriter

        Try

            ‘Open the output file specified in the run profile

            fw = New StreamWriter(filename, False)

        Catch ex As Exception

            Throw New UnauthorizedAccessException(“Unable to open file: ” & filename)

        End Try

        fw.WriteLine(“WebPath,staffid,uid,Type,Status”)

        ListFolders(WebPath, fw)

        fw.Close()

    End Sub

 

    Private Sub ListFolders(ByVal path As String, ByRef fw As StreamWriter)

        Dim s() As String

        Dim folders As System.Collections.IEnumerator

        Dim sqlQueryConnection As New SqlConnection(DB_CONNECTION_STRING)

        sqlQueryConnection.Open()

        s = System.IO.Directory.GetDirectories(path)

        folders = s.GetEnumerator

        While folders.MoveNext

            Dim sqlQueryCommand As New SqlCommand(“SELECT * FROM ” & DB_TABLE_NAME & ” where WebPath='” & folders.Current & “‘”, sqlQueryConnection)

            Dim cnReader As SqlDataReader = sqlQueryCommand.ExecuteReader

            Dim importLine As String

            If cnReader.Read Then

                ‘DB columns: WebPath, staffid, uid, Status

                importLine = cnReader.Item(1) & “,” & cnReader.Item(2) & “,” & cnReader.Item(3)

            Else

                importLine = “,,,”

            End If

            importLine = folders.Current & “,” & importLine

            fw.WriteLine(importLine)

            cnReader.Close()

        End While

        sqlQueryConnection.Close()

    End Sub

 

    Public Sub BeginExport(ByVal connectTo As String, ByVal user As String, ByVal password As String, ByVal configParameters As ConfigParameterCollection, ByVal types As TypeDescriptionCollection) Implements IMAExtensibleCallExport.BeginExport

        sqlExportConnection.Open()

    End Sub

 

    Public Sub ExportEntry(ByVal modificationType As ModificationType, ByVal changedAttributes As String(), ByVal csentry As CSEntry) Implements IMAExtensibleCallExport.ExportEntry

        Dim WebPath As String = csentry(“WebPath”).Value

        If csentry(“Status”).IsPresent AndAlso csentry(“Status”).Value = “archive pending” Then

            If ArchiveWebFolder(WebPath, csentry(“uid”).Value) = 0 Then

                updateRow(WebPath, “archived”)

            End If

        ElseIf modificationType = modificationType.Delete AndAlso csentry(“Status”).Value = “archived” Then

            updateRow(WebPath, “deleted”)

            DeleteWebFolder(csentry)

        ElseIf modificationType = modificationType.Add Then

            If Not rowExists(WebPath) AndAlso csentry(“uid”).IsPresent Then

                If CreateWebFolder(csentry) = 0 Then

                    addRow(WebPath, csentry(“staffid”).Value, csentry(“uid”).Value, “adding”)

                End If

            ElseIf rowExists(WebPath) And csentry(“Status”).Value = “adding” Then

                If CreateWebFolder(csentry) = 0 Then

                    updateRow(WebPath, “active”)

                End If

            End If

        Else

            If Not rowExists(WebPath) Then

                addRow(WebPath, csentry(“LBSNo”).Value, csentry(“uid”).Value, “active”)

            End If

        End If

    End Sub

 

    Private Function CreateWebFolder(ByVal csentry As CSEntry) As Integer

        Dim script As String = Params(“script-createsite”)

        RunProcess(script, csentry(“uid”).Value)

    End Function

 

    Private Function DeleteWebFolder(ByVal csentry As CSEntry) As Integer

        Dim script As String = Params(“script-deletesite”)

        RunProcess(script, csentry(“WebPath”).Value.Split(“\”)(4))

    End Function

 

    Private Function ArchiveWebFolder(ByVal path As String, ByVal uid As String) As Int16

        Dim status As Int16

        Dim archive As New Czzip

        Dim archivePath As String = Params(“archive-path”) & uid & “.zip”

        If System.IO.File.Exists(archivePath) Then

            Throw New UnexpectedDataException(“Archive file ” & archivePath & ” already exists”)

        End If

        Dim logtxt = Date.Now & ” Zipping ” & path & “: “

        status = archive.ZipFile(path, archivePath)

        If status = 0 AndAlso Not System.IO.File.Exists(archivePath) Then

            status = -1

        End If

        Return status

    End Function

 

    Public Sub EndExport() Implements IMAExtensibleCallExport.EndExport

    End Sub

 

    Private Function RunProcess(ByVal cmd As String, ByVal params As String) As Integer

        Dim exitCode As Integer = -1

        Dim proc As New System.Diagnostics.Process

        Dim waitTime As Integer = 0

        Const WAIT_TIME_IN_SECONDS = 5

        Const ONE_SECOND = 1000

        Const SLEEP_TIME_IN_SECONDS = 1

        proc.StartInfo.FileName = cmd

        proc.StartInfo.Arguments = params

        proc.StartInfo.WindowStyle = ProcessWindowStyle.Normal

        proc.Start()

        Try

            If Not proc.HasExited Then

                While True

                    System.Threading.Thread.Sleep(SLEEP_TIME_IN_SECONDS * ONE_SECOND)

                    waitTime += SLEEP_TIME_IN_SECONDS

                    If proc.HasExited Or waitTime >= WAIT_TIME_IN_SECONDS Then

                        Exit While

                    End If

                End While

            End If

            If proc.HasExited Then

                exitCode = proc.ExitCode

            Else

                Throw New UnexpectedDataException(“Timeout running process ” & cmd & ” ” & params)

            End If

        Catch ex As Exception

            Throw New UnexpectedDataException(“Error: ” & ex.Message)

        End Try

        Return exitCode

        proc.Kill()

    End Function

 

    Private Function deleteRows(ByVal WebPath As String) As Integer

        Dim delSQL As String = “DELETE FROM ” & DB_TABLE_NAME & ” WHERE WebPath = ‘” & WebPath & “‘”

        Dim sqlDelCommand As SqlCommand = New SqlCommand(delSQL, sqlExportConnection)

        Return sqlDelCommand.ExecuteNonQuery()

    End Function

 

    Private Function addRow(ByVal WebPath As String, ByVal LBSNo As String, ByVal uid As String, ByVal Type As String, ByVal Status As String) As Integer

        Dim addSQL As String = “INSERT INTO ” & DB_TABLE_NAME & ” (WebPath, staffid, uid, Status) VALUES” & ”  (‘” & WebPath & “‘, ‘” & staffid & “‘, ‘” & uid & “‘, ‘” & Status & “‘)”

        Dim sqlAddCommand As SqlCommand = New SqlCommand(addSQL, sqlExportConnection)

        Return sqlAddCommand.ExecuteNonQuery()

    End Function

 

    Private Function updateRow(ByVal WebPath As String, ByVal Status As String) As Integer

        Dim modSQL As String

        Dim sqlModCommand As SqlCommand

        modSQL = “UPDATE ” & DB_TABLE_NAME & ” SET Status ='” & Status & “‘ where WebPath='” & WebPath & “‘”

        sqlModCommand = New SqlCommand(modSQL, sqlExportConnection)

        Return sqlModCommand.ExecuteNonQuery()

    End Function

 

    Private Function rowExists(ByVal WebPath As String) As Boolean

        ‘Returns TRUE if the row already exists in the table

        ‘Pass % for wildcard

        Dim mySelectQuery As String = “SELECT * FROM ” & DB_TABLE_NAME & ” where WebPath = ‘” & WebPath & “‘”

        Dim myConnection As New SqlConnection(“Database=MIISSync;Data Source=localhost;Integrated Security=Yes;”)

        Dim myCommand As New SqlCommand(mySelectQuery, myConnection)

        myConnection.Open()

        Dim myReader As SqlDataReader = myCommand.ExecuteReader()

        rowExists = myReader.HasRows()

        myReader.Close()

        myConnection.Close()

    End Function

 

    Private Function escapedString(ByVal originalString As String, ByVal commandType As String) As String

        Select Case commandType

            Case “cmd”

                escapedString = originalString.Replace(“,”, “\,”)

                escapedString = escapedString.Replace(“+”, “\+”)

                escapedString = escapedString.Replace(“>”, “\>”)

                Return escapedString

            Case “sql”

                escapedString = originalString.Replace(“‘”, “””)

                Return escapedString

            Case Else

                Throw New ArgumentException(“Valid arguments: cmd, sql”)

        End Select

    End Function

End Class