websitepanel/WebsitePanel/Sources/WebsitePanel.Providers.DNS.MsDNS2012/RecordConverter.cs

using System;
using System.Linq;
using System.Management.Automation;
using Microsoft.Management.Infrastructure;
using Microsoft.Management.Infrastructure.Generic;
using WebsitePanel.Server.Utils;

namespace WebsitePanel.Providers.DNS
{
	/// <summary>Copy fields from CimInstance#DnsServerResourceRecord into DnsRecord</summary>
	/// <remarks>It's also possible to access native CIM object, and use Mgmtclassgen.exe for that.</remarks>
	internal static class RecordConverter
	{
		private static string RemoveTrailingDot( string str )
		{
			if( !str.EndsWith( "." ) )
				return str;
			return str.Substring( 0, str.Length - 1 );
		}

		private static string CorrectHost( string zoneName, string host )
		{
			if( host.ToLower() == zoneName.ToLower() )
				return "";
			if( host.ToLower().EndsWith( "." + zoneName.ToLower() ) )
				return host.Substring( 0, ( host.Length - zoneName.Length - 1 ) );
			return host;
		}

		public static DnsRecord asDnsRecord( this PSObject obj, string zoneName )
		{
			// Here's what comes from Server 2012 in the TypeNames:
			// "Microsoft.Management.Infrastructure.CimInstance#root/Microsoft/Windows/DNS/DnsServerResourceRecord"
			// "Microsoft.Management.Infrastructure.CimInstance#ROOT/Microsoft/Windows/DNS/DnsDomain"
			// "Microsoft.Management.Infrastructure.CimInstance#DnsServerResourceRecord"
			// "Microsoft.Management.Infrastructure.CimInstance#DnsDomain"
			// "Microsoft.Management.Infrastructure.CimInstance"
			// "System.Object"	string

			if( !obj.TypeNames.Contains( "Microsoft.Management.Infrastructure.CimInstance#DnsServerResourceRecord" ) )
			{
				Log.WriteWarning( "asDnsRecord: wrong object type {0}", obj.TypeNames.FirstOrDefault() );
				return null;
			}

			string strRT = (string)obj.Properties[ "RecordType" ].Value;
			DnsRecordType tp;
			if( !RecordTypes.recordFromString.TryGetValue( strRT, out tp ) )
				return null;

			/*// Debug code below: 
			obj.dumpProperties();
			CimInstance rd = (CimInstance)obj.Properties[ "RecordData" ].Value;
			rd.dumpProperties(); //*/

			CimKeyedCollection<CimProperty> data = ( (CimInstance)obj.Properties[ "RecordData" ].Value ).CimInstanceProperties;
			string host = CorrectHost( zoneName, (string)obj.Properties[ "HostName" ].Value );

			switch( tp )
			{
				// The compiler should create a Dictionary<> from dis switch
				case DnsRecordType.A:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = data[ "IPv4Address" ].Value as string,
						};
					}
				case DnsRecordType.AAAA:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = data[ "IPv6Address" ].Value as string,
						};
					}
				case DnsRecordType.CNAME:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = RemoveTrailingDot( data[ "HostNameAlias" ].Value as string ),
						};
					}
				case DnsRecordType.MX:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = RemoveTrailingDot( data[ "MailExchange" ].Value as string ),
							MxPriority = (UInt16)data[ "Preference" ].Value,
						};
					}
				case DnsRecordType.NS:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = RemoveTrailingDot( data[ "NameServer" ].Value as string ),
						};
					}
				case DnsRecordType.TXT:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = data[ "DescriptiveText" ].Value as string,
						};
					}
				case DnsRecordType.SOA:
					{
						string PrimaryServer = data[ "PrimaryServer" ].Value as string;
						string ResponsiblePerson = data[ "ResponsiblePerson" ].Value as string;
						UInt32? sn = (UInt32?)data[ "SerialNumber" ].Value;
						return new DnsSOARecord()
						{
							RecordType = tp,
							RecordName = host,
							PrimaryNsServer = PrimaryServer,
							PrimaryPerson = ResponsiblePerson,
							SerialNumber = ( sn.HasValue ) ? sn.Value.ToString() : null,
						};
					}
				case DnsRecordType.SRV:
					{
						return new DnsRecord()
						{
							RecordType = tp,
							RecordName = host,
							RecordData = RemoveTrailingDot( data[ "DomainName" ].Value as string ),
							SrvPriority = (UInt16)data[ "Priority" ].Value,
							SrvWeight = (UInt16)data[ "Weight" ].Value,
							SrvPort = (UInt16)data[ "Port" ].Value,
						};
					}
			}
			return null;
		}
	}
}