Remotely/Server/API/BrandingController.cs

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Remotely.Server.Services;
using Remotely.Shared.Models;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;

namespace Remotely.Server.API
{
    [Route("api/[controller]")]
    [ApiController]
    public class BrandingController : ControllerBase
    {
        private readonly IDataService _dataService;

        public BrandingController(IDataService dataService)
        {
            _dataService = dataService;
        }


        [HttpGet("{organizationId}")]
        public async Task<BrandingInfo> Get(string organizationId)
        {
            var org = _dataService.GetOrganizationById(organizationId);
            // We're doing the honor system here.
            if (org.SponsorLevel < Shared.Enums.SponsorLevel.Branding)
            {
                return null;
            }

            return await _dataService.GetBrandingInfo(organizationId);
        }

        [HttpGet]
        public async Task<BrandingInfo> GetDefault()
        {
            var defaultOrg = await _dataService.GetDefaultOrganization();
            if (defaultOrg?.SponsorLevel >= Shared.Enums.SponsorLevel.Branding)
            {
                return await _dataService.GetBrandingInfo(defaultOrg.ID);
            }

            return null;
        }
    }
}