lemonbrain :: 개발자 수련장

  public void OnConnection(object application, ext_ConnectMode connectMode, object addInInst, ref Array custom)
  {
   _applicationObject = (DTE2)application;
   _addInInstance = (AddIn)addInInst;
            switch (connectMode)
            {
                   
                case ext_ConnectMode.ext_cm_Startup:
                    SetToolBar();
                    break;

                case ext_ConnectMode.ext_cm_AfterStartup: 
                    SetToolBar();
                    break;

                case ext_ConnectMode.ext_cm_CommandLine:
                    break;

                case ext_ConnectMode.ext_cm_External:
                    break;

                case ext_ConnectMode.ext_cm_Solution:
                    break;

                case ext_ConnectMode.ext_cm_UISetup:
                    break;

                default:
                    break;
            }
  }

        /// <summary>
        /// 표준Toolbar에 버튼을 등록
        /// </summary>
        private void SetToolBar()
        {
            CommandBar stdCmdBar;
            CommandBarButton cmdBarBtn;
            object[] contextGUIDS = new object[] { };

            //커맨드가 등록되어 있지않다면 새로 등록
            if (ToolbarCommand == null)
            {
                ToolbarCommand = _applicationObject.Commands.AddNamedCommand(_addInInstance,
                  "ANewComment", "코멘트추가", "코멘트를 추가해줍니다.", true, 59, ref  contextGUIDS,
                      (int)vsCommandStatus.vsCommandStatusSupported 
                     | (int)vsCommandStatus.vsCommandStatusEnabled);
            }

            //표준툴바의 객체를 취득
            stdCmdBar =
            (Microsoft.VisualStudio.CommandBars.CommandBar)((Microsoft.VisualStudio.CommandBars.CommandBars)_applicationObject.CommandBars)["Standard"];

            //표준툴바에 버튼을 추가
            stdCmdBarCtl = (Microsoft.VisualStudio.CommandBars.CommandBarControl)ToolbarCommand.AddControl(stdCmdBar, stdCmdBar.Controls.Count + 1);

            //버튼에 캡션설정
            stdCmdBarCtl.Caption = "이거 누르면 코멘트가 추가됩니다.";

            //버튼을 아이콘으로 설정
            cmdBarBtn = (CommandBarButton)stdCmdBarCtl;
            cmdBarBtn.Style = MsoButtonStyle.msoButtonIcon;
        }

  public void QueryStatus(string commandName, vsCommandStatusTextWanted neededText, ref vsCommandStatus status, ref object commandText)
  {
   if(neededText == vsCommandStatusTextWanted.vsCommandStatusTextWantedNone)
   {

                //커맨드명과 현재 활성화된 창의 문서가 Document일 경우에만 사용가능하다는 신호를 보냄
                if (commandName == "MyTestAddin1.Connect.ANewComment" && _applicationObject.ActiveWindow.Type == vsWindowType.vsWindowTypeDocument)
                 {
                    status = (vsCommandStatus) vsCommandStatus.vsCommandStatusEnabled | vsCommandStatus.vsCommandStatusSupported;
                    return;
                 }

   }
  }

  public void Exec(string commandName, vsCommandExecOption executeOption, ref object varIn, ref object varOut, ref bool handled)
  {

   handled = false;
   if(executeOption == vsCommandExecOption.vsCommandExecOptionDoDefault)
   {
                switch (commandName) //스위치를 사용하여 불려진 커맨드를 체크.(혹시 하나의 추가기능에 버튼이 여러개 있을수도 있으니까 스위치를 사용합니다.)
                {
                    case "MyTestAddin1.Connect.ANewComment":
                        //현재 열려진 창이 문서인지 재확인을 하고
                        if (_applicationObject.ActiveWindow.Type == vsWindowType.vsWindowTypeDocument)
                        {
                            //열려진 문서의 인스턴스를 취득해서
                            TextDocument textDoc = (TextDocument) _applicationObject.ActiveWindow.Document.Object("TextDocument");
                            textDoc.StartPoint.CreateEditPoint(); //커서위치를 가져오고
  
                            //커맨드를 삽입하고
                            textDoc.Selection.Insert(string.Format("//{0}에 입력되었습니다.",DateTime.Now.ToString()), 0); 

                            //보기좋게 하기위해 개행문자를 넣어줌...
                            textDoc.Selection.EndOfLine(true);
                            textDoc.Selection.NewLine(1);

                            handled = true;
                        }
                        return;

                    default:
                        break;
                }
   }


}

  public void OnDisconnection(ext_DisconnectMode disconnectMode, ref Array custom)
  {

            //추가기능이 언로드될 등록한 버튼과 커맨드를 삭제
            try
            {
                if (disconnectMode == ext_DisconnectMode.ext_dm_UserClosed)
                {
                    if (stdCmdBarCtl != null)
                    {
                        stdCmdBarCtl.Delete(null);
                        stdCmdBarCtl = null;

                        ToolbarCommand.Delete();
                        ToolbarCommand = null;
                    }

                    if (MenuToolCommand != null)
                    {
                        MenuToolCommand.Delete();
                        MenuToolCommand = null;
                    }
                }
            }
            catch (Exception ex)
            {
                System.Windows.Forms.MessageBox.Show(ex.ToString());
            }
  }