public class SiteStructureService : ISiteStructureService
{
    /// <summary>
    /// Finds and updates all references associated with the specified page.
    /// </summary>
    public void UpdateReferences(IPage page)
    {
        var sectionType = page.IsSystem ? SectionType.SystemPages : page.IsInMenu ? SectionType.Menu : SectionType.NotMenu;
        var section = this.GetSection(sectionType);

        section.FindByPageEntityId(page.Id).Do(si => si.LinkPageEntity(page));
    }

    /// <summary>
    /// Finds and updates all references associated with the specified template.
    /// </summary>
    public void UpdateReferences(IPageTemplate template)
    {
        Action<ISiteStructureItem> updateTemplateName =
            si => si.LinkedPage.If(p => p.TemplateId == template.Id).With(p => p as PageReference).Do(p => p.TemplateName = template.Name);

        ISiteStructureSection menuSection = this.GetSection(SectionType.Menu);
        ISiteStructureSection notMenuSection = this.GetSection(SectionType.NotMenu);
        ISiteStructureSection systemSection = this.GetSection(SectionType.SystemPages);
        ISiteStructureSection templateSection = this.GetSection(SectionType.Templates);

        templateSection.FindByPageEntityId(template.Id).Do(si => si.LinkPageEntity(template));

        menuSection.ForEach(updateTemplateName);
        notMenuSection.ForEach(updateTemplateName);
        systemSection.ForEach(updateTemplateName);
    }
}