Umbraco-CMS/tests/Umbraco.TestData/SegmentTestController.cs

using System;
using System.Linq;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using Umbraco.Cms.Core.Cache;
using Umbraco.Cms.Core.Logging;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Routing;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Core.Web;
using Umbraco.Cms.Infrastructure.Persistence;
using Umbraco.Cms.Web.Website.Controllers;
using Umbraco.Extensions;
using Umbraco.TestData.Configuration;

namespace Umbraco.TestData
{
    public class SegmentTestController : SurfaceController
    {
        private IOptions<TestDataSettings> _testDataSettings;

        public SegmentTestController(
            IUmbracoContextAccessor umbracoContextAccessor,
            IUmbracoDatabaseFactory databaseFactory,
            ServiceContext services,
            AppCaches appCaches,
            IProfilingLogger profilingLogger,
            IPublishedUrlProvider publishedUrlProvider,
            IOptions<TestDataSettings> testDataSettings)
            : base(umbracoContextAccessor, databaseFactory, services, appCaches, profilingLogger, publishedUrlProvider)
        {
            _testDataSettings = testDataSettings;
        }

        public IActionResult EnableDocTypeSegments(string alias, string propertyTypeAlias)
        {
            if(_testDataSettings.Value.Enabled != true)
            {
                return HttpNotFound();
            }

            IContentType ct = Services.ContentTypeService.Get(alias);
            if (ct == null)
            {
                return Content($"No document type found by alias {alias}");
            }

            IPropertyType propType = ct.PropertyTypes.FirstOrDefault(x => x.Alias == propertyTypeAlias);
            if (propType == null)
            {
                return Content($"The document type {alias} does not have a property type {propertyTypeAlias ?? "null"}");
            }

            if (ct.Variations.VariesBySegment())
            {
                return Content($"The document type {alias} already allows segments, nothing has been changed");
            }

            ct.SetVariesBy(ContentVariation.Segment);
            propType.SetVariesBy(ContentVariation.Segment);

            Services.ContentTypeService.Save(ct);
            return Content($"The document type {alias} and property type {propertyTypeAlias} now allows segments");
        }

        private IActionResult HttpNotFound() => throw new NotImplementedException();

        public IActionResult DisableDocTypeSegments(string alias)
        {
            if (_testDataSettings.Value.Enabled != true)
            {
                return HttpNotFound();
            }

            IContentType ct = Services.ContentTypeService.Get(alias);
            if (ct == null)
            {
                return Content($"No document type found by alias {alias}");
            }

            if (!ct.VariesBySegment())
            {
                return Content($"The document type {alias} does not allow segments, nothing has been changed");
            }

            ct.SetVariesBy(ContentVariation.Segment, false);

            Services.ContentTypeService.Save(ct);
            return Content($"The document type {alias} no longer allows segments");
        }

        public ActionResult AddSegmentData(int contentId, string propertyAlias, string value, string segment, string culture = null)
        {
            IContent content = Services.ContentService.GetById(contentId);
            if (content == null)
            {
                return Content($"No content found by id {contentId}");
            }

            if (propertyAlias.IsNullOrWhiteSpace() || !content.HasProperty(propertyAlias))
            {
                return Content($"The content by id {contentId} does not contain a property with alias {propertyAlias ?? "null"}");
            }

            if (content.ContentType.VariesByCulture() && culture.IsNullOrWhiteSpace())
            {
                return Content($"The content by id {contentId} varies by culture but no culture was specified");
            }

            if (value.IsNullOrWhiteSpace())
            {
                return Content("'value' cannot be null");
            }

            if (segment.IsNullOrWhiteSpace())
            {
                return Content("'segment' cannot be null");
            }

            content.SetValue(propertyAlias, value, culture, segment);
            Services.ContentService.Save(content);

            return Content($"Segment value has been set on content {contentId} for property {propertyAlias}");
        }
    }
}