private CodeTypeDeclaration GetClassDeclaration(ModelClass cls, CodeNamespace nameSpace)
{
if (cls == null)
throw new ArgumentException("Class not supplied.", "cls");
if (String.IsNullOrEmpty(cls.Name))
throw new ArgumentException("Class name cannot be blank.", "cls");
string className = cls.Name;
if (cls.Model.GeneratesDoubleDerived)
className += cls.Model.DoubleDerivedNameSuffix;
CodeTypeDeclaration classDeclaration = CreateClass(className);
if (cls.Model.AutomaticAssociations)
classDeclaration.CreateEmptyPublicConstructor();
if (cls.Model.GeneratesDoubleDerived)
classDeclaration.TypeAttributes |= TypeAttributes.Abstract;
if (cls.ClassParent != null)
{
classDeclaration.BaseTypes.Add(new CodeTypeReference(cls.ClassParent.Name));
}
else if (cls.Model.UseBaseClass)
{
bool withValidator = cls.Properties.FindAll(delegate(ModelProperty property)
{
return property.IsValidatorSet();
}).Count > 0;
CodeTypeReference type;
// base class for every modelclass. If left empty then baseclass from model if left empty ...etc
if (!string.IsNullOrEmpty(cls.BaseClassName))
type = new CodeTypeReference(cls.BaseClassName);
else if (!string.IsNullOrEmpty(cls.Model.BaseClassName))
type = new CodeTypeReference(cls.Model.BaseClassName);
else if (withValidator)
type = new CodeTypeReference(Common.DefaultValidationBaseClass);
else
type = new CodeTypeReference(Common.DefaultBaseClass);
if (cls.IsGeneric())
type.TypeArguments.Add(cls.Name);
classDeclaration.BaseTypes.Add(type);
}
if (cls.DoesImplementINotifyPropertyChanged() && cls.ClassParent == null && !cls.Model.PropertyChangedDefinedInBaseClass)
{
classDeclaration.BaseTypes.Add(new CodeTypeReference(Common.INotifyPropertyChangedType));
AddINotifyPropertyChangedRegion(classDeclaration, cls.Lazy | Context.Model.UseVirtualProperties);
}
if (cls.DoesImplementINotifyPropertyChanging() && cls.ClassParent == null && !cls.Model.PropertyChangingDefinedInBaseClass)
{
classDeclaration.BaseTypes.Add(new CodeTypeReference(Common.INotifyPropertyChangingType));
AddINotifyPropertyChangingRegion(classDeclaration, cls.Lazy | Context.Model.UseVirtualProperties);
}
if (!String.IsNullOrEmpty(cls.Description))
classDeclaration.Comments.AddRange(GetSummaryComment(cls.Description));
if (!cls.Model.GeneratesDoubleDerived)
cls.AddActiveRecordAttributes(classDeclaration);
if (cls.Model.UseGeneratedCodeAttribute)
classDeclaration.CustomAttributes.Add(AttributeHelper.GetGeneratedCodeAttribute());
nameSpace.Types.Add(classDeclaration);
return classDeclaration;
}
private void GenerateDerivedClass(ModelClass cls, CodeNamespace nameSpace)
{
if (cls.Model.GeneratesDoubleDerived)
{
CodeTypeDeclaration derivedClass = new CodeTypeDeclaration(cls.Name);
derivedClass.IsPartial = true;
derivedClass.TypeAttributes = TypeAttributes.Public;
derivedClass.BaseTypes.Add(new CodeTypeReference(cls.Name + cls.Model.DoubleDerivedNameSuffix));
cls.AddActiveRecordAttributes(derivedClass);
nameSpace.Types.Add(derivedClass);
}
}
