Formatting.vb

' '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
'  Visual Studio format all files macro.
'  Copyright (C) 2009 Tim Abell <tim@timwise.co.uk>
'
'  This program is free software: you can redistribute it and/or modify
'  it under the terms of the GNU General Public License as published by
'  the Free Software Foundation, either version 3 of the License, or
'  (at your option) any later version.
'
'  This program is distributed in the hope that it will be useful,
'  but WITHOUT ANY WARRANTY; without even the implied warranty of
'  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
'  GNU General Public License for more details.
'
'  You should have received a copy of the GNU General Public License
'  along with this program. If not, see <http://www.gnu.org/licenses/>.
'
' See the file COPYING for the full license.
'
' http://timwise.blogspot.com/2009/01/format-all-document-in-visual-studio.html
' http://github.com/timabell/vi-formatter-macro
' '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''

Imports System
Imports EnvDTE
Imports EnvDTE80
Imports EnvDTE90
Imports System.Collections.Generic
Imports System.Diagnostics
Imports System.Text


Public Module Formatting

    Dim allowed As List(Of String)
    Dim processed As Integer
    Dim ignored As Integer
    Dim errors As StringBuilder
    Dim skippedExtensions As List(Of String)

    Public Sub FormatProject()
        allowed = New List(Of String)
        allowed.Add(".master")
        allowed.Add(".aspx")
        allowed.Add(".ascx")
        allowed.Add(".asmx")
        allowed.Add(".cs")
        allowed.Add(".vb")
        allowed.Add(".config")
        allowed.Add(".css")
        allowed.Add(".htm")
        allowed.Add(".html")
        allowed.Add(".js")

        processed = 0
        ignored = 0
        errors = New StringBuilder
        skippedExtensions = New List(Of String)

        Try
            recurseSolution(AddressOf processItem)
        Catch ex As Exception
            Debug.Print("error in main loop: " + ex.ToString())
        End Try
        Debug.Print("processed items: " + processed.ToString())
        Debug.Print("ignored items: " + ignored.ToString())
        Debug.Print("ignored extensions: " + String.Join(" ", skippedExtensions.ToArray()))
        Debug.Print(errors.ToString())
    End Sub

    Private Sub processItem(ByVal Item As ProjectItem)
        If Not Item.Name.Contains(".") Then
            'Debug.Print("no file extension. ignoring.")
            ignored += 1
            Return
        End If
        Dim ext As String
        ext = Item.Name.Substring(Item.Name.LastIndexOf("."))   'get file extension
        If allowed.Contains(ext) Then
            formatItem(Item)
            processed += 1
        Else
            'Debug.Print("ignoring file with extension: " + ext)
            If Not skippedExtensions.Contains(ext) Then
                skippedExtensions.Add(ext)
            End If
            ignored += 1
        End If
    End Sub

    Private Sub formatItem(ByVal Item As ProjectItem)
        Debug.Print("processing file " + Item.Name)
        Dim window As EnvDTE.Window
        Try
            Try
                window = Item.Open(ViewKind:=Constants.vsViewKindCode)
            Catch
                Try
                    window = Item.Open(ViewKind:=Constants.vsViewKindTextView)
                Catch
                    window = Item.Open(ViewKind:=Constants.vsViewKindAny)
                End Try
            End Try
            window.Activate()
            DTE.ExecuteCommand("Edit.FormatDocument", "")
            DTE.ExecuteCommand("EditorContextMenus.CodeWindow.OrganizeUsings.RemoveAndSort", "")
        Catch ex As Exception
            Debug.Print("error processing file: " + ex.Message)
            errors.AppendLine("error processing file " + Item.Name + ": " + ex.Message)
        End Try
        Try
            window.Document.Save()
            window.Close()
        Catch
        End Try
    End Sub

    Private Delegate Sub task(ByVal Item As ProjectItem)

    Private Sub recurseSolution(ByVal taskRoutine As task)
        For Each Proj As Project In DTE.Solution.Projects
            Debug.Print("project " + Proj.Name)
            For Each Item As ProjectItem In Proj.ProjectItems
                recurseItems(Item, 0, taskRoutine)
            Next
        Next
    End Sub

    Private Sub recurseItems(ByVal Item As ProjectItem, ByVal depth As Integer, ByVal taskRoutine As task)
        Dim indent As String = New String("-", depth)
        Debug.Print(indent + " " + Item.Name)
        taskRoutine(Item)
        If Not Item.ProjectItems Is Nothing Then
            For Each Child As ProjectItem In Item.ProjectItems
                taskRoutine(Child)
                recurseItems(Child, depth + 1, taskRoutine)
            Next
        End If
    End Sub

End Module