Imports System
Imports Microsoft.VisualStudio.CommandBars
Imports Extensibility
Imports EnvDTE
Imports EnvDTE80

Public Class Connect
	
    Implements IDTExtensibility2

    Private _applicationObject As DTE2
    Private _addInInstance As AddIn
    Private WithEvents BuildEvents As EnvDTE.BuildEvents

    Public Sub OnConnection(ByVal application As Object, ByVal connectMode As ext_ConnectMode, ByVal addInInst As Object, ByRef custom As Array) Implements IDTExtensibility2.OnConnection
        _applicationObject = CType(application, DTE2)
        _addInInstance = CType(addInInst, AddIn)
        BuildEvents = _applicationObject.Events.BuildEvents
    End Sub

    Private Sub BuildEvents_OnBuildDone(Scope As vsBuildScope, Action As vsBuildAction) Handles BuildEvents.OnBuildDone
        Dim p As Project
        If Action = vsBuildAction.vsBuildActionRebuildAll Then
            If Scope = vsBuildScope.vsBuildScopeProject Then
                p = _applicationObject.ActiveSolutionProjects(0)
                IncValue(p.Properties.Item("AssemblyVersion"))
            Else
                For Each p In _applicationObject.Solution.Projects
                    IncValue(p.Properties.Item("AssemblyVersion"))
                Next
            End If
        End If
    End Sub

    Private Sub IncValue(ByVal ver As EnvDTE.[Property])
        Dim sVer As String = ver.Value.ToString, iNum As Integer
        Dim iDot As Integer = InStrRev(sVer, ".")
        If iDot = 0 Then
            iNum = Val(sVer)
        Else
            iNum = Val(Mid(sVer, iDot + 1))
        End If
        ver.Value = Mid(sVer, 1, iDot) & (iNum + 1)
    End Sub

    Public Sub OnDisconnection(ByVal disconnectMode As ext_DisconnectMode, ByRef custom As Array) Implements IDTExtensibility2.OnDisconnection
    End Sub

    Public Sub OnAddInsUpdate(ByRef custom As Array) Implements IDTExtensibility2.OnAddInsUpdate
    End Sub

    Public Sub OnStartupComplete(ByRef custom As Array) Implements IDTExtensibility2.OnStartupComplete
    End Sub

    Public Sub OnBeginShutdown(ByRef custom As Array) Implements IDTExtensibility2.OnBeginShutdown
    End Sub

End Class