RelationshipCatalog.cs

/*
 * NanoXLSX is a small .NET library to generate and read XLSX (Microsoft Excel 2007 or newer) files in an easy and native way
 * Copyright Raphael Stoeckli © 2026
 * This library is licensed under the MIT License.
 * You find a copy of the license in project folder or on: http://opensource.org/licenses/MIT
 */
 
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
 
namespace NanoXLSX.Internal
{
    internal sealed class RelationshipCatalog
    {
        private readonly List<RelationshipInfo> relationships = new List<RelationshipInfo>();
        private readonly List<RelationshipDiscoveryIssue> issues = new List<RelationshipDiscoveryIssue>();
        private readonly Dictionary<string, Dictionary<string, RelationshipInfo>> relationshipsBySource
            = new Dictionary<string, Dictionary<string, RelationshipInfo>>(StringComparer.Ordinal);
        private readonly ReadOnlyCollection<RelationshipInfo> readOnlyRelationships;
        private readonly ReadOnlyCollection<RelationshipDiscoveryIssue> readOnlyIssues;

        public IReadOnlyList<RelationshipInfo> Relationships { get { return readOnlyRelationships; } }

        public IReadOnlyList<RelationshipDiscoveryIssue> Issues { get { return readOnlyIssues; } }

        public bool IsComplete { get { return issues.Count == 0; } }

        public RelationshipCatalog()
        {
            readOnlyRelationships = relationships.AsReadOnly();
            readOnlyIssues = issues.AsReadOnly();
        }

        public bool TryAdd(RelationshipInfo relationship)
        {
            string sourcePartPath = relationship.SourcePartPath ?? string.Empty;
            if (!relationshipsBySource.TryGetValue(sourcePartPath, out Dictionary<string, RelationshipInfo> sourceRelationships))
            {
                sourceRelationships = new Dictionary<string, RelationshipInfo>(StringComparer.Ordinal);
                relationshipsBySource.Add(sourcePartPath, sourceRelationships);
            }
            if (sourceRelationships.ContainsKey(relationship.Id))
            {
                return false;
            }
            sourceRelationships.Add(relationship.Id, relationship);
            relationships.Add(relationship);
            return true;
        }

        public void AddIssue(RelationshipDiscoveryIssue issue)
        {
            issues.Add(issue);
        }

        public RelationshipInfo GetBySourceAndId(string sourcePartPath, string relationshipId)
        {
            string normalizedSourcePartPath = sourcePartPath ?? string.Empty;
            if (relationshipsBySource.TryGetValue(normalizedSourcePartPath, out Dictionary<string, RelationshipInfo> sourceRelationships)
                && relationshipId != null
                && sourceRelationships.TryGetValue(relationshipId, out RelationshipInfo relationship))
            {
                return relationship;
            }
            return null;
        }

        public IReadOnlyList<RelationshipInfo> GetByType(string documentType)
        {
            List<RelationshipInfo> matches = new List<RelationshipInfo>();
            foreach (RelationshipInfo relationship in relationships)
            {
                if (string.Equals(relationship.Type, documentType, StringComparison.Ordinal))
                {
                    matches.Add(relationship);
                }
            }
            return matches.AsReadOnly();
        }
    }
}