using System; using System.Collections.Generic; using System.Linq; using System.Reflection; using System.Xml.XPath; using Examine; using Examine.LuceneEngine.Providers; using Examine.LuceneEngine.SearchCriteria; using Examine.SearchCriteria; using Umbraco.Core; using Umbraco.Core.Models.PublishedContent; using Umbraco.Core.Services; using Umbraco.Core.Xml; using Umbraco.Web.PublishedCache; namespace Umbraco.Web { using Examine = global::Examine; /// /// A class used to query for published content, media items /// public class PublishedContentQuery : IPublishedContentQuery { private readonly IPublishedContentCache _contentCache; private readonly IPublishedMediaCache _mediaCache; /// /// Constructor used to return results from the caches /// /// /// public PublishedContentQuery(IPublishedContentCache contentCache, IPublishedMediaCache mediaCache) { _contentCache = contentCache ?? throw new ArgumentNullException(nameof(contentCache)); _mediaCache = mediaCache ?? throw new ArgumentNullException(nameof(mediaCache)); } #region Content public IPublishedContent Content(int id) { return ItemById(id, _contentCache); } public IPublishedContent Content(Guid id) { return ItemById(id, _contentCache); } public IPublishedContent Content(Udi id) { if (!(id is GuidUdi udi)) return null; return ItemById(udi.Guid, _contentCache); } public IPublishedContent ContentSingleAtXPath(string xpath, params XPathVariable[] vars) { return ItemByXPath(xpath, vars, _contentCache); } public IEnumerable Content(IEnumerable ids) { return ItemsByIds(_contentCache, ids); } public IEnumerable Content(IEnumerable ids) { return ItemsByIds(_contentCache, ids); } public IEnumerable ContentAtXPath(string xpath, params XPathVariable[] vars) { return ItemsByXPath(xpath, vars, _contentCache); } public IEnumerable ContentAtXPath(XPathExpression xpath, params XPathVariable[] vars) { return ItemsByXPath(xpath, vars, _contentCache); } public IEnumerable ContentAtRoot() { return ItemsAtRoot(_contentCache); } #endregion #region Media public IPublishedContent Media(int id) { return ItemById(id, _mediaCache); } public IPublishedContent Media(Guid id) { return ItemById(id, _mediaCache); } public IPublishedContent Media(Udi id) { if (!(id is GuidUdi udi)) return null; return ItemById(udi.Guid, _mediaCache); } public IEnumerable Media(IEnumerable ids) { return ItemsByIds(_mediaCache, ids); } public IEnumerable Media(IEnumerable ids) { return ItemsByIds(_mediaCache, ids); } public IEnumerable MediaAtRoot() { return ItemsAtRoot(_mediaCache); } #endregion #region Used by Content/Media private static IPublishedContent ItemById(int id, IPublishedCache cache) { var doc = cache.GetById(id); return doc; } private static IPublishedContent ItemById(Guid id, IPublishedCache cache) { var doc = cache.GetById(id); return doc; } private static IPublishedContent ItemByXPath(string xpath, XPathVariable[] vars, IPublishedCache cache) { var doc = cache.GetSingleByXPath(xpath, vars); return doc; } //NOTE: Not used? //private IPublishedContent ItemByXPath(XPathExpression xpath, XPathVariable[] vars, IPublishedCache cache) //{ // var doc = cache.GetSingleByXPath(xpath, vars); // return doc; //} private static IEnumerable ItemsByIds(IPublishedCache cache, IEnumerable ids) { return ids.Select(eachId => ItemById(eachId, cache)).WhereNotNull(); } private IEnumerable ItemsByIds(IPublishedCache cache, IEnumerable ids) { return ids.Select(eachId => ItemById(eachId, cache)).WhereNotNull(); } private static IEnumerable ItemsByXPath(string xpath, XPathVariable[] vars, IPublishedCache cache) { var doc = cache.GetByXPath(xpath, vars); return doc; } private static IEnumerable ItemsByXPath(XPathExpression xpath, XPathVariable[] vars, IPublishedCache cache) { var doc = cache.GetByXPath(xpath, vars); return doc; } private static IEnumerable ItemsAtRoot(IPublishedCache cache) { return cache.GetAtRoot(); } #endregion #region Search /// public IEnumerable Search(string term, bool useWildCards = true, string indexName = null) { return Search(0, 0, out _, term, useWildCards, indexName); } /// public IEnumerable Search(int skip, int take, out long totalRecords, string term, bool useWildCards = true, string indexName = null) { //fixme: inject IExamineManager indexName = string.IsNullOrEmpty(indexName) ? Constants.UmbracoIndexes.ExternalIndexName : indexName; if (!ExamineManager.Instance.TryGetIndex(indexName, out var index)) throw new InvalidOperationException($"No index found by name {indexName}"); var searcher = index.GetSearcher(); var results = skip == 0 && take == 0 ? searcher.Search(term) : searcher.Search(term, maxResults: skip + take); totalRecords = results.TotalItemCount; return results.ToPublishedSearchResults(_contentCache); } /// public IEnumerable Search(ISearchCriteria criteria, ISearcher searchProvider = null) { return Search(0, 0, out _, criteria, searchProvider); } /// public IEnumerable Search(int skip, int take, out long totalRecords, ISearchCriteria criteria, ISearcher searcher = null) { //fixme: inject IExamineManager if (searcher == null) { if (!ExamineManager.Instance.TryGetIndex(Constants.UmbracoIndexes.ExternalIndexName, out var index)) throw new InvalidOperationException($"No index found by name {Constants.UmbracoIndexes.ExternalIndexName}"); searcher = index.GetSearcher(); } var results = skip == 0 && take == 0 ? searcher.Search(criteria) : searcher.Search(criteria, maxResults: skip + take); totalRecords = results.TotalItemCount; return results.ToPublishedSearchResults(_contentCache); } #endregion } }