Use type's source URI

mganss · May 26, 2021 · 7ba7203 · 7ba7203
1 parent 0a0568e
commit 7ba7203
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/XmlSchemaClassGenerator/ModelBuilder.cs b/XmlSchemaClassGenerator/ModelBuilder.cs
@@ -247,11 +247,10 @@ private void CreateElements(IEnumerable<XmlSchemaElement> elements)
  {
  foreach (var rootElement in elements)
  {
- var rootSchema = rootElement.GetSchema();
- var source = CodeUtilities.CreateUri(rootSchema.SourceUri);
+ var typeSource = CodeUtilities.CreateUri(rootElement.ElementSchemaType.SourceUri);
  var qualifiedName = rootElement.ElementSchemaType.QualifiedName;
  if (qualifiedName.IsEmpty) { qualifiedName = rootElement.QualifiedName; }
- var type = CreateTypeModel(source, rootElement.ElementSchemaType, qualifiedName);
+ var type = CreateTypeModel(typeSource, rootElement.ElementSchemaType, qualifiedName);
  ClassModel derivedClassModel = null;
 
  if (type.RootElementName != null)
@@ -261,10 +260,12 @@ private void CreateElements(IEnumerable<XmlSchemaElement> elements)
  // There is already another global element with this type.
  // Need to create an empty derived class.
 
+ var elementSource = CodeUtilities.CreateUri(rootElement.SourceUri);
+
  derivedClassModel = new ClassModel(_configuration)
  {
  Name = _configuration.NamingProvider.RootClassNameFromQualifiedName(rootElement.QualifiedName),
- Namespace = CreateNamespaceModel(source, rootElement.QualifiedName)
+ Namespace = CreateNamespaceModel(elementSource, rootElement.QualifiedName)
  };
 
  derivedClassModel.Documentation.AddRange(GetDocumentation(rootElement));