//
// (C) Copyright 2003-2013 by Autodesk, Inc.
//
// Permission to use, copy, modify, and distribute this software in
// object code form for any purpose and without fee is hereby granted,
// provided that the above copyright notice appears in all copies and
// that both that copyright notice and the limited warranty and
// restricted rights notice below appear in all supporting
// documentation.
//
// AUTODESK PROVIDES THIS PROGRAM "AS IS" AND WITH ALL FAULTS.
// AUTODESK SPECIFICALLY DISCLAIMS ANY IMPLIED WARRANTY OF
// MERCHANTABILITY OR FITNESS FOR A PARTICULAR USE. AUTODESK, INC.
// DOES NOT WARRANT THAT THE OPERATION OF THE PROGRAM WILL BE
// UNINTERRUPTED OR ERROR FREE.
//
// Use, duplication, or disclosure by the U.S. Government is subject to
// restrictions set forth in FAR 52.227-19 (Commercial Computer
// Software - Restricted Rights) and DFAR 252.227-7013(c)(1)(ii)
// (Rights in Technical Data and Computer Software), as applicable.
//

using System;
using System.Collections;
using System.Diagnostics;
using System.IO;
using System.Reflection;
using System.Windows.Forms;
using System.Collections.Generic;
using System.Windows.Media.Imaging;

using Autodesk.Revit;
using Autodesk.Revit.UI;
using Autodesk.Revit.ApplicationServices;

namespace RvtSamples
{
  /// <summary>
  /// Main external application class.
  /// A generic menu generator application.
  /// Read a text file and add entries to the Revit menu.
  /// Any number and location of entries is supported.
  /// </summary>
  public class Application : IExternalApplication
  {
    /// <summary>
    /// Default pulldown menus for samples
    /// </summary>
    public enum DefaultPulldownMenus
    {
      /// <summary>
      /// Menu for Basics category
      /// </summary>
      Basics,
      /// <summary>
      /// Menu for Geometry category
      /// </summary>
      Geometry,
      /// <summary>
      /// Menu for Parameters category
      /// </summary>
      Parameters,
      /// <summary>
      /// Menu for Elements category
      /// </summary>
      Elements,
      /// <summary>
      /// Menu for Families category
      /// </summary>
      Families,
      /// <summary>
      /// Menu for Materials category
      /// </summary>
      Materials,
      /// <summary>
      /// Menu for Annotation category
      /// </summary>
      Annotation,
      /// <summary>
      /// Menu for Views category
      /// </summary>
      Views,
      /// <summary>
      /// Menu for Rooms/Spaces category
      /// </summary>
      RoomsAndSpaces,
      /// <summary>
      /// Menu for Data Exchange category
      /// </summary>
      DataExchange,
      /// <summary>
      /// Menu for MEP category
      /// </summary>
      MEP,
      /// <summary>
      /// Menu for Structure category
      /// </summary>
      Structure
    }

    #region Member Data
    /// <summary>
    /// Separator of category for samples have more than one category
    /// </summary>
    static char[] s_charSeparatorOfCategory = new char[] { ',' };
    /// <summary>
    /// chars which will be trimmed in the file to include extra sample list files
    /// </summary>
    static char[] s_trimChars = new char[] { ' ', '"', '\'', '<', '>' };
    /// <summary>
    /// The start symbol of lines to include extra sample list files
    /// </summary>
    static string s_includeSymbol = "#include";
    /// <summary>
    /// Assembly directory
    /// </summary>
    static string s_assemblyDirectory = Path.GetDirectoryName( System.Reflection.Assembly.GetExecutingAssembly().Location );
    /// <summary>
    /// Name of file which contains information required for menu items
    /// </summary>
    public const string m_fileNameStem = "RvtSamples.txt";
    /// <summary>
    /// The controlled application of Revit
    /// </summary>
    private UIControlledApplication m_application;
    /// <summary>
    /// List contains all pulldown button items contain sample items
    /// </summary>
    private SortedList<string, PulldownButton> m_pulldownButtons = new SortedList<string, PulldownButton>();
    /// <summary>
    /// List contains information of samples not belong to default pulldown menus
    /// </summary>
    private SortedList<string, List<SampleItem>> m_customizedMenus = new SortedList<string, List<SampleItem>>();
    /// <summary>
    /// List contains information of samples belong to default pulldown menus
    /// </summary>
    private SortedList<string, List<SampleItem>> m_defaultMenus = new SortedList<string, List<SampleItem>>();
    /// <summary>
    /// Panel for RvtSamples
    /// </summary>
    RibbonPanel m_panelRvtSamples;
    #endregion // Member Data

    #region IExternalApplication Members
    /// <summary>
    /// Implement this method to implement the external application which should be called when
    /// Revit starts before a file or default template is actually loaded.
    /// </summary>
    /// <param name="application">An object that is passed to the external application
    /// which contains the controlled application.</param>
    /// <returns>Return the status of the external application.
    /// A result of Succeeded means that the external application successfully started.
    /// Cancelled can be used to signify that the user cancelled the external operation at
    /// some point.
    /// If false is returned then Revit should inform the user that the external application
    /// failed to load and the release the internal reference.</returns>
    public Autodesk.Revit.UI.Result OnStartup( UIControlledApplication application )
    {
      m_application = application;
      Autodesk.Revit.UI.Result rc = Autodesk.Revit.UI.Result.Failed;
      string[] lines = null;
      int n = 0;
      int k = 0;

      try
      {
        // Check whether the file contains samples' list exists
        // If not, return failure
        string filename = m_fileNameStem;
        if( !GetFilepath( ref filename ) )
        {
          ErrorMsg( m_fileNameStem + " not found." );
          return rc;
        }

        // Read all lines from the file
        lines = ReadAllLinesWithInclude( filename );
        // Remove comments
        lines = RemoveComments( lines );

        // Add default pulldown menus of samples to Revit
        m_panelRvtSamples = application.CreateRibbonPanel( "RvtSamples" );
        int i = 0;
        List<PulldownButtonData> pdData = new List<PulldownButtonData>( 3 );
        foreach( string category in Enum.GetNames( typeof( DefaultPulldownMenus ) ) )
        {
          if( ( i + 1 ) % 3 == 1 )
          {
            pdData.Clear();
          }

          //
          // Prepare PulldownButtonData for add stacked buttons operation
          //
          string displayName = GetDisplayNameByEnumName( category );

          List<SampleItem> sampleItems = new List<SampleItem>();
          m_defaultMenus.Add( category, sampleItems );

          PulldownButtonData data = new PulldownButtonData( displayName, displayName );
          pdData.Add( data );

          //
          // Add stacked buttons to RvtSamples panel and set their display names and images
          //
          if( ( i + 1 ) % 3 == 0 )
          {
            IList<RibbonItem> addedButtons = m_panelRvtSamples.AddStackedItems( pdData[0], pdData[1], pdData[2] );
            foreach( RibbonItem item in addedButtons )
            {
              String name = item.ItemText;
              string enumName = GetEnumNameByDisplayName( name );
              PulldownButton button = item as PulldownButton;
              button.Image = new BitmapImage(
                  new Uri( Path.Combine( s_assemblyDirectory, "Icons\\" + enumName + ".ico" ), UriKind.Absolute ) );
              button.ToolTip = Properties.Resource.ResourceManager.GetString( enumName );
              m_pulldownButtons.Add( name, button );
            }
          }

          i++;
        }

        //
        // Add sample items to the pulldown buttons
        //
        n = lines.GetLength( 0 );
        k = 0;
        while( k < n )
        {
          AddSample( lines, n, ref k );
        }

        AddSamplesToDefaultPulldownMenus();
        AddCustomizedPulldownMenus();

        rc = Autodesk.Revit.UI.Result.Succeeded;
      }
      catch( Exception e )
      {
        string s = string.Format( "{0}: n = {1}, k = {2}, lines[k] = {3}",
          e.Message, n, k, ( k < n ? lines[k] : "eof" ) );

        ErrorMsg( s );
      }
      return rc;
    }

    /// <summary>
    /// Get a button's enum name by its display name
    /// </summary>
    /// <param name="name">display name</param>
    /// <returns>enum name</returns>
    private string GetEnumNameByDisplayName( string name )
    {
      string enumName = null;
      if( name.Equals( "Rooms/Spaces" ) )
      {
        enumName = DefaultPulldownMenus.RoomsAndSpaces.ToString();
      }
      else if( name.Equals( "Data Exchange" ) )
      {
        enumName = DefaultPulldownMenus.DataExchange.ToString();
      }
      else
      {
        enumName = name;
      }

      return enumName;
    }

    /// <summary>
    /// Get a button's display name by its enum name
    /// </summary>
    /// <param name="enumName">The button's enum name</param>
    /// <returns>The button's display name</returns>
    private string GetDisplayNameByEnumName( string enumName )
    {
      string displayName = null;
      if( enumName.Equals( DefaultPulldownMenus.RoomsAndSpaces.ToString() ) )
      {
        displayName = "Rooms/Spaces";
      }
      else if( enumName.Equals( DefaultPulldownMenus.DataExchange.ToString() ) )
      {
        displayName = "Data Exchange";
      }
      else
      {
        displayName = enumName;
      }

      return displayName;
    }

    /// <summary>
    /// Implement this method to implement the external application which should be called when
    /// Revit is about to exit,Any documents must have been closed before this method is called.
    /// </summary>
    /// <param name="application">An object that is passed to the external application
    /// which contains the controlled application.</param>
    /// <returns>Return the status of the external application.
    /// A result of Succeeded means that the external application successfully shutdown.
    /// Cancelled can be used to signify that the user cancelled the external operation at
    /// some point.
    /// If false is returned then the Revit user should be warned of the failure of the external
    /// application to shut down correctly.</returns>
    public Autodesk.Revit.UI.Result OnShutdown( UIControlledApplication application )
    {
      return Autodesk.Revit.UI.Result.Succeeded;
    }
    #endregion // IExternalApplication Members

    /// <summary>
    /// Display error message
    /// </summary>
    /// <param name="msg">Message to display</param>
    public void ErrorMsg( string msg )
    {
      Debug.WriteLine( "RvtSamples: " + msg );
      TaskDialog.Show( "RvtSamples", msg, TaskDialogCommonButtons.Ok );
    }

    #region Parser
    /// <summary>
    /// Read file contents, including contents of files included in the current file
    /// </summary>
    /// <param name="filename">Current file to be read</param>
    /// <returns>All lines of file contents</returns>
    private string[] ReadAllLinesWithInclude( string filename )
    {
      if( !File.Exists( filename ) )
      {
        ErrorMsg( filename + " not found." );
        return new string[] { };
      }
      string[] lines = File.ReadAllLines( filename );

      int n = lines.GetLength( 0 );
      ArrayList all_lines = new ArrayList( n );
      foreach( string line in lines )
      {
        string s = line.TrimStart();
        if( s.ToLower().StartsWith( s_includeSymbol ) )
        {
          string filename2 = s.Substring( s_includeSymbol.Length );
          filename2 = filename2.Trim( s_trimChars );
          all_lines.AddRange( ReadAllLinesWithInclude( filename2 ) );
        }
        else
        {
          all_lines.Add( line );
        }
      }
      return all_lines.ToArray( typeof( string ) ) as string[];
    }

    /// <summary>
    /// Get the input file path.
    /// Search and return the full path for the given file
    /// in the current exe directory or one or two directory levels higher.
    /// </summary>
    /// <param name="filename">Input filename stem, output full file path</param>
    /// <returns>True if found, false otherwise.</returns>
    bool GetFilepath( ref string filename )
    {
      string path = Path.Combine( s_assemblyDirectory, filename );
      bool rc = File.Exists( path );

      // Get full path of the file
      if( rc )
      {
        filename = Path.GetFullPath( path );
      }
      return rc;
    }

    /// <summary>
    /// Remove all comments and empty lines from a given array of lines.
    /// Comments are delimited by '#' to the end of the line.
    /// </summary>
    string[] RemoveComments( string[] lines )
    {
      int n = lines.GetLength( 0 );
      string[] a = new string[n];
      int i = 0;
      foreach( string line in lines )
      {
        string s = line;
        int j = s.IndexOf( '#' );
        if( 0 <= j )
        {
          s = s.Substring( 0, j );
        }
        s = s.Trim();
        if( 0 < s.Length )
        {
          a[i++] = s;
        }
      }
      string[] b = new string[i];
      n = i;
      for( i = 0; i < n; ++i )
      {
        b[i] = a[i];
      }
      return b;
    }
    #endregion // Parser

    #region Menu Helpers
    /// <summary>
    /// Add a new command to the corresponding pulldown button.
    /// </summary>
    /// <param name="lines">Array of lines defining sample's category, display name, description, large image, image, assembly and classname</param>
    /// <param name="n">Total number of lines in array</param>
    /// <param name="i">Current index in array</param>
    void AddSample( string[] lines, int n, ref int i )
    {
      if( n < i + 6 )
      {
        throw new Exception( string.Format( "Incomplete record at line {0} of {1}", i, m_fileNameStem ) );
      }

      string categories = lines[i++].Trim();
      string displayName = lines[i++].Trim();
      string description = lines[i++].Trim();
      string largeImage = lines[i++].Remove( 0, 11 ).Trim();
      string image = lines[i++].Remove( 0, 6 ).Trim();
      string assembly = lines[i++].Trim();
      string className = lines[i++].Trim();

      if( !File.Exists( assembly ) ) // jeremy
      {
        ErrorMsg( string.Format( "Assembly '{0}' specified in line {1} of {2} not found",
          assembly, i, m_fileNameStem ) );
      }

      bool testClassName = false; // jeremy

      if( testClassName )
      {
        Debug.Print( "RvtSamples: testing command {0} in assembly '{1}'.", className, assembly );

        try
        {
          // first load the revit api assembly, otherwise we cannot query the external app for its types:

          //Assembly revit = Assembly.LoadFrom( "C:/Program Files/Revit Architecture 2009/Program/RevitAPI.dll" );
          //string root = "C:/Program Files/Autodesk Revit Architecture 2010/Program/";
          //Assembly adWindows = Assembly.LoadFrom( root + "AdWindows.dll" );
          //Assembly uiFramework = Assembly.LoadFrom( root + "UIFramework.dll" );
          //Assembly revit = Assembly.LoadFrom( root + "RevitAPI.dll" );

          // load the assembly into the current application domain:

          Assembly a = Assembly.LoadFrom( assembly );

          if( null == a )
          {
            ErrorMsg( string.Format( "Unable to load assembly '{0}' specified in line {1} of {2}",
              assembly, i, m_fileNameStem ) );
          }
          else
          {
            // get the type to use:
            Type t = a.GetType( className );
            if( null == t )
            {
              ErrorMsg( string.Format( "External command class {0} in assembly '{1}' specified in line {2} of {3} not found",
                className, assembly, i, m_fileNameStem ) );
            }
            else
            {
              // get the method to call:
              MethodInfo m = t.GetMethod( "Execute" );
              if( null == m )
              {
                ErrorMsg( string.Format( "External command class {0} in assembly '{1}' specified in line {2} of {3} does not define an Execute method",
                  className, assembly, i, m_fileNameStem ) );
              }
            }
          }
        }
        catch( Exception ex )
        {
          ErrorMsg( string.Format( "Exception '{0}' \ntesting assembly '{1}' \nspecified in line {2} of {3}",
            ex.Message, assembly, i, m_fileNameStem ) );
        }
      }

      //
      // If sample belongs to default category, add the sample item to the sample list of the default category
      // If not, store the information for adding to RvtSamples panel later
      //
      string[] entries = categories.Split( s_charSeparatorOfCategory, StringSplitOptions.RemoveEmptyEntries );
      foreach( string value in entries )
      {
        string category = value.Trim();
        SampleItem item = new SampleItem( category, displayName, description, largeImage, image, assembly, className );
        if( m_pulldownButtons.ContainsKey( category ) )
        {
          m_defaultMenus.Values[m_defaultMenus.IndexOfKey( GetEnumNameByDisplayName( category ) )].Add( item );
        }
        else if( m_customizedMenus.ContainsKey( category ) )
        {
          List<SampleItem> sampleItems = m_customizedMenus.Values[m_customizedMenus.IndexOfKey( category )];
          sampleItems.Add( item );
        }
        else
        {
          List<SampleItem> sampleItems = new List<SampleItem>();
          sampleItems.Add( item );
          m_customizedMenus.Add( category, sampleItems );
        }
      }
    }

    /// <summary>
    /// Add sample item to pulldown menu
    /// </summary>
    /// <param name="pullDownButton">Pulldown menu</param>
    /// <param name="item">Sample item to be added</param>
    private void AddSampleToPulldownMenu( PulldownButton pullDownButton, SampleItem item )
    {
      PushButtonData pushButtonData = new PushButtonData( item.DisplayName, item.DisplayName, item.Assembly, item.ClassName );
      PushButton pushButton = pullDownButton.AddPushButton( pushButtonData );
      if( !string.IsNullOrEmpty( item.LargeImage ) )
      {
        BitmapImage largeImageSource = new BitmapImage( new Uri( item.LargeImage, UriKind.Absolute ) );
        pushButton.LargeImage = largeImageSource;
      }
      if( !string.IsNullOrEmpty( item.Image ) )
      {
        BitmapImage imageSource = new BitmapImage( new Uri( item.Image, UriKind.Absolute ) );
        pushButton.Image = imageSource;
      }

      pushButton.ToolTip = item.Description;
    }

    /// <summary>
    /// Comparer to sort sample items by their display name
    /// </summary>
    /// <param name="s1">sample item 1</param>
    /// <param name="s2">sample item 2</param>
    /// <returns>compare result</returns>
    private static int SortByDisplayName( SampleItem item1, SampleItem item2 )
    {
      return string.Compare( item1.DisplayName, item2.DisplayName );
    }

    /// <summary>
    /// Sort samples in one category by the sample items' display name
    /// </summary>
    /// <param name="menus">samples to be sorted</param>
    private void SortSampleItemsInOneCategory( SortedList<string, List<SampleItem>> menus )
    {
      int iCount = menus.Count;

      for( int j = 0; j < iCount; j++ )
      {
        List<SampleItem> sampleItems = menus.Values[j];
        sampleItems.Sort( SortByDisplayName );
      }
    }

    /// <summary>
    /// Add samples of categories in default categories
    /// </summary>
    private void AddSamplesToDefaultPulldownMenus()
    {
      int iCount = m_defaultMenus.Count;

      // Sort sample items in every category by display name
      SortSampleItemsInOneCategory( m_defaultMenus );

      for( int i = 0; i < iCount; i++ )
      {
        string category = m_defaultMenus.Keys[i];
        List<SampleItem> sampleItems = m_defaultMenus.Values[i];
        PulldownButton menuButton = m_pulldownButtons.Values[m_pulldownButtons.IndexOfKey( GetDisplayNameByEnumName( category ) )];
        foreach( SampleItem item in sampleItems )
        {
          AddSampleToPulldownMenu( menuButton, item );
        }
      }
    }

    /// <summary>
    /// Add samples of categories not in default categories
    /// </summary>
    private void AddCustomizedPulldownMenus()
    {
      int iCount = m_customizedMenus.Count;

      // Sort sample items in every category by display name
      SortSampleItemsInOneCategory( m_customizedMenus );

      int i = 0;

      while( iCount >= 3 )
      {
        string name = m_customizedMenus.Keys[i++];
        PulldownButtonData data1 = new PulldownButtonData( name, name );
        name = m_customizedMenus.Keys[i++];
        PulldownButtonData data2 = new PulldownButtonData( name, name );
        name = m_customizedMenus.Keys[i++];
        PulldownButtonData data3 = new PulldownButtonData( name, name );
        IList<RibbonItem> buttons = m_panelRvtSamples.AddStackedItems( data1, data2, data3 );
        AddSamplesToStackedButtons( buttons );

        iCount -= 3;
      }

      if( iCount == 2 )
      {
        string name = m_customizedMenus.Keys[i++];
        PulldownButtonData data1 = new PulldownButtonData( name, name );
        name = m_customizedMenus.Keys[i++];
        PulldownButtonData data2 = new PulldownButtonData( name, name );
        IList<RibbonItem> buttons = m_panelRvtSamples.AddStackedItems( data1, data2 );
        AddSamplesToStackedButtons( buttons );
      }
      else if( iCount == 1 )
      {
        string name = m_customizedMenus.Keys[i];
        PulldownButtonData pulldownButtonData = new PulldownButtonData( name, name );
        PulldownButton button = m_panelRvtSamples.AddItem( pulldownButtonData ) as PulldownButton;
        List<SampleItem> sampleItems = m_customizedMenus.Values[m_customizedMenus.IndexOfKey( button.Name )];
        foreach( SampleItem item in sampleItems )
        {
          AddSampleToPulldownMenu( button, item );
        }
      }
    }

    /// <summary>
    /// Add samples to corresponding pulldown button
    /// </summary>
    /// <param name="buttons">pulldown buttons</param>
    private void AddSamplesToStackedButtons( IList<RibbonItem> buttons )
    {
      foreach( RibbonItem rItem in buttons )
      {
        PulldownButton button = rItem as PulldownButton;
        List<SampleItem> sampleItems = m_customizedMenus.Values[m_customizedMenus.IndexOfKey( button.Name )];
        foreach( SampleItem item in sampleItems )
        {
          AddSampleToPulldownMenu( button, item );
        }
      }
    }
    #endregion // Menu Helpers
  }
}