Imports System
Imports System.Collections.Generic
Imports System.Text
Imports System.Diagnostics
Imports Microsoft.EnterpriseManagement.Administration
Imports Microsoft.EnterpriseManagement
Imports Microsoft.EnterpriseManagement.ConnectorFramework
Imports Microsoft.EnterpriseManagement.Monitoring
Imports Microsoft.EnterpriseManagement.Configuration
Imports Microsoft.EnterpriseManagement.Configuration.IO
Imports System.Collections.ObjectModel
Imports System.IO


Namespace DiscoveryDataInsertion

	Class Program

		Public Overloads Shared Function Main(ByVal args() As String) As Integer
			Dim insertionSample As New DiscoveryDataInsertionSample("localhost")

			insertionSample.InsertDiscoveryData()
		End Function 'Main 
	End Class 'Program

	Class DiscoveryDataInsertionSample
		Private m_managementGroup As ManagementGroup
		Private m_monitoringConnector As MonitoringConnector
		Private m_routerMonitoringObject As MonitoringObject
		Private m_routerPortMonitoringObject As MonitoringObject
		Private m_routerFanMonitoringObject As MonitoringObject


		Public Sub New(ByVal managementServerName As String)
			Debug.Assert((managementServerName.Length > 0))

			ConnectToManagementGroup(managementServerName)

			CreateConnector()

			ImportMPIfNeed()
		End Sub 'New


		Public Sub InsertDiscoveryData()
			Dim discoveryData As New SnapshotMonitoringDiscoveryData()
			Dim routerClass As MonitoringClass
			Dim routerPortClass As MonitoringClass
			Dim routerFanClass As MonitoringClass
			Dim entityClass As MonitoringClass
			Dim fanContainedInRouterRelationshipClass As MonitoringRelationshipClass
			Dim routerMonitoringObject As CustomMonitoringObject
			Dim routerPortObject As CustomMonitoringObject
			Dim routerFanMonitoringObject As CustomMonitoringObject
			Dim relationshipObject As CustomMonitoringRelationshipObject
			Dim routerIdProp As MonitoringClassProperty
			Dim portIdProp As MonitoringClassProperty
			Dim fanIdProp As MonitoringClassProperty
			Dim displayNameProp As MonitoringClassProperty


			routerClass = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.Router")
			routerPortClass = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.RouterPort")
			routerFanClass = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.RouterFan")
			entityClass = GetMonitoringClass("System.Entity")
			fanContainedInRouterRelationshipClass = GetMonitoringRelationshipClass("SdkDemo.DiscoveryDataInsertion.RouterContainsFan")

			routerIdProp = routerClass.GetMonitoringProperty("RouterId")
			portIdProp = routerPortClass.GetMonitoringProperty("PortId")
			fanIdProp = routerFanClass.GetMonitoringProperty("FanId")
			displayNameProp = entityClass.GetMonitoringProperty("DisplayName")

			routerMonitoringObject = New CustomMonitoringObject(routerClass)
			routerPortObject = New CustomMonitoringObject(routerPortClass)
			routerFanMonitoringObject = New CustomMonitoringObject(routerFanClass)
			relationshipObject = New CustomMonitoringRelationshipObject(fanContainedInRouterRelationshipClass)

			routerMonitoringObject.SetMonitoringPropertyValue(routerIdProp, "1")
			routerMonitoringObject.SetMonitoringPropertyValue(displayNameProp, "Router 123")

			'There is a hosting relationship between the router and the router port.
			'For Operations Manager to know what router it needs to associate with the port object, 
			'we have to set the key prop of the router (RouterId) on the hosted object(RouterPort).
			'There is no need to explicitly create a relationship object for relationships of type hosting.
			'An instance of the relationship is created automatically.
			routerPortObject.SetMonitoringPropertyValue(routerIdProp, "1")
			routerPortObject.SetMonitoringPropertyValue(portIdProp, "1")
			routerPortObject.SetMonitoringPropertyValue(displayNameProp, "Port #1")

			routerFanMonitoringObject.SetMonitoringPropertyValue(fanIdProp, "1")
			routerFanMonitoringObject.SetMonitoringPropertyValue(displayNameProp, "Fan #1")

			relationshipObject.SetSource(routerMonitoringObject)
			relationshipObject.SetTarget(routerFanMonitoringObject)

			discoveryData.Include(routerPortObject)
			discoveryData.Include(routerMonitoringObject)
			discoveryData.Include(routerFanMonitoringObject)
			discoveryData.Include(relationshipObject)

			discoveryData.Commit(m_monitoringConnector)

			' Obtain references to the newly created objects.
			' They are used to insert events and performance data.
			m_routerMonitoringObject = m_managementGroup.GetMonitoringObject(CType(routerMonitoringObject.Id, Guid))
			m_routerPortMonitoringObject = m_managementGroup.GetMonitoringObject(CType(routerPortObject.Id, Guid))
			m_routerFanMonitoringObject = m_managementGroup.GetMonitoringObject(CType(routerFanMonitoringObject.Id, Guid))
		End Sub 'InsertDiscoveryData


		Private Sub ConnectToManagementGroup(ByVal managementServerName As String)
			m_managementGroup = New ManagementGroup(managementServerName)
		End Sub 'ConnectToManagementGroup

		'---------------------------------------------------------------------
		Private Sub CreateConnector()
			Dim mcfAdmin As ConnectorFrameworkAdministration
			Dim connectorGuid As New Guid("DE1FC259-FE16-44ce-B90E-708CADE55909")

			mcfAdmin = m_managementGroup.GetConnectorFrameworkAdministration()

			Try
				m_monitoringConnector = mcfAdmin.GetMonitoringConnector(connectorGuid)
			Catch e As Microsoft.EnterpriseManagement.Common.ObjectNotFoundException
				'The connector does not exist, so create it.
				Dim connectorInfo As New ConnectorInfo()

				connectorInfo.Description = "This is a sample connector to demonstrate discovery data insertion"
				connectorInfo.DisplayName = "Sample connector"
				connectorInfo.Name = "Sample connector"

				m_monitoringConnector = mcfAdmin.Setup(connectorInfo, connectorGuid)
			End Try

			If Not m_monitoringConnector.Initialized Then
				m_monitoringConnector.Initialize()
			End If
		End Sub 'CreateConnector


		Private Function GetMonitoringClass(ByVal className As String) As MonitoringClass
			Dim monitoringClasses As ReadOnlyCollection(Of MonitoringClass)

			monitoringClasses = m_managementGroup.GetMonitoringClasses(className)

			If monitoringClasses.Count = 0 Then
				Throw New ApplicationException("Failed to find monitoring class " + className)
			End If

			Return monitoringClasses(0)
		End Function 'GetMonitoringClass


		Private Function GetMonitoringRelationshipClass(ByVal relationshipName As String) As MonitoringRelationshipClass
			Dim relationshipClasses As ReadOnlyCollection(Of MonitoringRelationshipClass)

			relationshipClasses = m_managementGroup.GetMonitoringRelationshipClasses(relationshipName)

			If relationshipClasses.Count = 0 Then
				Throw New ApplicationException("Failed to find monitoring relationship " + relationshipName)
			End If

			Return relationshipClasses(0)
		End Function 'GetMonitoringRelationshipClass


		Private Sub ImportMPIfNeed()
			Dim criteria As New ManagementPackCriteria("Name = 'SdkDemo.DiscoveryDataInsertion'")

			If m_managementGroup.GetManagementPacks(criteria).Count = 0 Then

				Dim mgmtPackPath As String = "C:\Program Files\System Center Operations Manager 2007\SdkDemo.DiscoveryDataInsertion.xml"
				Dim mp As ManagementPack = New ManagementPack(mgmtPackPath)

				m_managementGroup.ImportManagementPack(mp)

			End If
		End Sub 'ImportMPIfNeed
	End Class 'DiscoveryDataInsertionSample
End Namespace 'DiscoveryDataInsertion

using System;
using System.Collections.Generic;
using System.Text;
using System.Diagnostics;
using Microsoft.EnterpriseManagement.Administration;
using Microsoft.EnterpriseManagement;
using Microsoft.EnterpriseManagement.ConnectorFramework;
using Microsoft.EnterpriseManagement.Monitoring;
using Microsoft.EnterpriseManagement.Configuration;
using Microsoft.EnterpriseManagement.Configuration.IO;
using System.Collections.ObjectModel;
using System.IO;

namespace DiscoveryDataInsertion
{
	class Program
	{
		static void Main(string[] args)
		{
			DiscoveryDataInsertionSample insertionSample = new DiscoveryDataInsertionSample("localhost");

			insertionSample.InsertDiscoveryData();

	}
}

   class DiscoveryDataInsertionSample
	{
		ManagementGroup	 m_managementGroup;
		MonitoringConnector m_monitoringConnector;
		MonitoringObject	m_routerMonitoringObject;
		MonitoringObject	m_routerPortMonitoringObject;
		MonitoringObject	m_routerFanMonitoringObject;

		public DiscoveryDataInsertionSample(
			string managementServerName
			)
		{
			Debug.Assert(managementServerName.Length > 0);

			ConnectToManagementGroup(managementServerName);

			CreateConnector();

			ImportMPIfNeed();
	}

		public void InsertDiscoveryData()
		{
			SnapshotMonitoringDiscoveryData discoveryData = new SnapshotMonitoringDiscoveryData();
			MonitoringClass					 routerClass;
			MonitoringClass					 routerPortClass;
			MonitoringClass					 routerFanClass;
			MonitoringClass					 entityClass;
			MonitoringRelationshipClass		 fanContainedInRouterRelationshipClass;
			CustomMonitoringObject			routerMonitoringObject;
			CustomMonitoringObject			routerPortObject;
			CustomMonitoringObject			routerFanMonitoringObject;
			CustomMonitoringRelationshipObject  relationshipObject;
			MonitoringClassProperty			 routerIdProp;
			MonitoringClassProperty			 portIdProp;
			MonitoringClassProperty			 fanIdProp;
			MonitoringClassProperty			 displayNameProp;
		

			routerClass	 = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.Router");
			routerPortClass = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.RouterPort");
			routerFanClass  = GetMonitoringClass("SdkDemo.DiscoveryDataInsertion.RouterFan");
			entityClass	 = GetMonitoringClass("System.Entity");
			fanContainedInRouterRelationshipClass = GetMonitoringRelationshipClass("SdkDemo.DiscoveryDataInsertion.RouterContainsFan");

			routerIdProp	= routerClass.GetMonitoringProperty("RouterId");
			portIdProp	= routerPortClass.GetMonitoringProperty("PortId");
			fanIdProp	 = routerFanClass.GetMonitoringProperty("FanId");
			displayNameProp = entityClass.GetMonitoringProperty("DisplayName");

			routerMonitoringObject	= new CustomMonitoringObject(routerClass);
			routerPortObject			= new CustomMonitoringObject(routerPortClass);
			routerFanMonitoringObject   = new CustomMonitoringObject(routerFanClass);
			relationshipObject		= new CustomMonitoringRelationshipObject(fanContainedInRouterRelationshipClass);

			routerMonitoringObject.SetMonitoringPropertyValue(routerIdProp, "1");
			routerMonitoringObject.SetMonitoringPropertyValue(displayNameProp, "Router 123");

			//There is a hosting relationship between the router and the router port.
			//For Operations Manager to know what router it needs to associate with the port object, 
			//we have to set the key prop of the router (RouterId) on the hosted object(RouterPort).
			//There is no need to explicitly create a relationship object for relationships of type hosting.
			//An instance of the relationship is created automatically.
		
			routerPortObject.SetMonitoringPropertyValue(routerIdProp, "1");
			routerPortObject.SetMonitoringPropertyValue(portIdProp, "1");
			routerPortObject.SetMonitoringPropertyValue(displayNameProp, "Port #1");

			routerFanMonitoringObject.SetMonitoringPropertyValue(fanIdProp, "1");
			routerFanMonitoringObject.SetMonitoringPropertyValue(displayNameProp, "Fan #1");

			relationshipObject.SetSource(routerMonitoringObject);
			relationshipObject.SetTarget(routerFanMonitoringObject);

			discoveryData.Include(routerPortObject);
			discoveryData.Include(routerMonitoringObject);
			discoveryData.Include(routerFanMonitoringObject);
			discoveryData.Include(relationshipObject);

			discoveryData.Commit(m_monitoringConnector);

			// Obtain references to the newly created objects.
			// They are used to insert events and performance data.
			m_routerMonitoringObject		= m_managementGroup.GetMonitoringObject((Guid)routerMonitoringObject.Id);
			m_routerPortMonitoringObject	= m_managementGroup.GetMonitoringObject((Guid)routerPortObject.Id);
			m_routerFanMonitoringObject	 = m_managementGroup.GetMonitoringObject((Guid)routerFanMonitoringObject.Id);
	}

		private void ConnectToManagementGroup(string managementServerName)
		{
			m_managementGroup = new ManagementGroup(managementServerName);
	}

		//---------------------------------------------------------------------
		private void CreateConnector()
		{
			ConnectorFrameworkAdministration mcfAdmin;
			Guid connectorGuid = new Guid("DE1FC259-FE16-44ce-B90E-708CADE55909");

			mcfAdmin = m_managementGroup.GetConnectorFrameworkAdministration();

			try
			{
				m_monitoringConnector = mcfAdmin.GetMonitoringConnector(connectorGuid);
		}
			catch (Microsoft.EnterpriseManagement.Common.ObjectNotFoundException error)
			{
				//The connector does not exist, so create it.

				ConnectorInfo connectorInfo = new ConnectorInfo();

				connectorInfo.Description = "This is a sample connector to demonstrate discovery data insertion";
				connectorInfo.DisplayName = "Sample connector";
				connectorInfo.Name = "Sample connector";

				m_monitoringConnector = mcfAdmin.Setup(connectorInfo, connectorGuid);
		}

			if (!m_monitoringConnector.Initialized)
			{
				m_monitoringConnector.Initialize();
		}
	}

		private MonitoringClass GetMonitoringClass(
			string className
			)
		{
			ReadOnlyCollection<MonitoringClass> monitoringClasses;

			monitoringClasses = m_managementGroup.GetMonitoringClasses(className);

			if (monitoringClasses.Count == 0)
			{
				throw new ApplicationException("Failed to find monitoring class " + className);
		}

			return (monitoringClasses[0]);
	}

		private MonitoringRelationshipClass GetMonitoringRelationshipClass(
			string relationshipName
			)
		{
			ReadOnlyCollection<MonitoringRelationshipClass> relationshipClasses;

			relationshipClasses = m_managementGroup.GetMonitoringRelationshipClasses(relationshipName);

			if (relationshipClasses.Count == 0)
			{
				throw new ApplicationException("Failed to find monitoring relationship " + relationshipName);
		}

			return (relationshipClasses[0]);
	}

		private void ImportMPIfNeed()
		{
			ManagementPackCriteria criteria = new ManagementPackCriteria("Name = 'SdkDemo.DiscoveryDataInsertion'");

			if (m_managementGroup.GetManagementPacks(criteria).Count == 0)
			{
			 
				string mgmtPackPath = @"C:\Program Files\System Center Operations Manager 2007\SdkDemo.DiscoveryDataInsertion.xml";
				ManagementPack mp = new ManagementPack(mgmtPackPath);
				 
				m_managementGroup.ImportManagementPack(mp);

		}					
	}
}
}