The Power of a POC

using System;
using System.Collections.Generic;
using System.Data.Objects;
using System.Linq;
using System.Runtime.Serialization;

using EntityModel = AdventureWorksEntityModel;


namespace AdventureWorksModel
{
    [DataContract(IsReference = true)]
    public partial class ProductDocument : PocoBase, IExtensibleDataObject
    {
   		// Class Name
		public const string PRODUCT_DOCUMENT = "ProductDocument";
		
		// Property Names
        public const string DOCUMENT_ID = "DocumentID";
        public const string PRODUCT_ID = "ProductID";
        public const string MODIFIED_DATE = "ModifiedDate";

		// Data Members - Keys
        [DataMember(EmitDefaultValue = false)] public Int32 DocumentID { get; private set; }
        [DataMember(EmitDefaultValue = false)] public Int32 ProductID { get; private set; }

		// Data Members
        [DataMember(EmitDefaultValue = false)] public DateTime ModifiedDate { get; set; }

		// Reference (FK) to other Table (PK)	
		[DataMember(EmitDefaultValue = false)] public Document Document { get; set; }
		[DataMember(EmitDefaultValue = false)] public PocoReference<Document> DocumentReference { get; set; }
		
		[DataMember(EmitDefaultValue = false)] public Product Product { get; set; }
		[DataMember(EmitDefaultValue = false)] public PocoReference<Product> ProductReference { get; set; }
		

		public ExtensionDataObject ExtensionData { get; set; }

		//
		// Public Class Methods
		//

   		public override IList<KeyValuePair<string, object>> GetKeys()
    	{
			return new List<KeyValuePair<string, object>> 
			{ 
				new KeyValuePair<string, object>(DOCUMENT_ID, DocumentID),
				new KeyValuePair<string, object>(PRODUCT_ID, ProductID)
			};
    	}

    	public PocoReference<ProductDocument> GetReference()
		{
			PocoReference<ProductDocument> reference = GetReference(DocumentID, ProductID);
			reference.PocoKey.ID = ObjectID;
			return reference; 
		}

		public static PocoReference<ProductDocument> GetReference(Int32 documentID, Int32 productID)
		{
			List<KeyValuePair<string, object>> pairs = new List<KeyValuePair<string, object>> 
			{ 
				new KeyValuePair<string, object>(DOCUMENT_ID, documentID),
				new KeyValuePair<string, object>(PRODUCT_ID, productID)
			};

			return GetReference(new PocoKey { Name = PRODUCT_DOCUMENT, KeyValues = pairs });
		}		
    	
    	private static PocoReference<ProductDocument> GetReference(PocoKey pocoKey)
		{
			return new PocoReference<ProductDocument>
			{
				PocoKey = pocoKey
			};
		}

		//
		// Repository Calls
		//
		
		public override string EntitySetName()
    	{
			return Repository.EntitySetName<ProductDocument, EntityModel.ProductDocument>(false);
    	}

		public override IQueryable<PocoBase> Find()
    	{
			return Repository.Find<ProductDocument, EntityModel.ProductDocument>(SearchOptions, MergeOption.NoTracking).Cast<PocoBase>();
    	}

		public override PocoKey Add()
    	{
			PocoKey key = Repository.Add<ProductDocument, EntityModel.ProductDocument>(this);
			PocoReference<ProductDocument> reference = GetReference(key);
			
			DocumentID = (Int32)key.KeyValues.FirstOrDefault(k=> k.Key.Equals(DOCUMENT_ID)).Value;
			ProductID = (Int32)key.KeyValues.FirstOrDefault(k=> k.Key.Equals(PRODUCT_ID)).Value;

			return key;
    	}
    	
		public override void Update()
    	{
			Repository.Update<ProductDocument, EntityModel.ProductDocument>(this);
    	}
    	
		public override void Delete()
		{
			Repository.Delete<ProductDocument, EntityModel.ProductDocument>(this);
		}
	}
}