Controllers/SiteController.cs

@@ -15,11 +15,11 @@ namespace rockfishCore.Controllers
     [Authorize]
     public class SiteController : Controller
     {
-        private readonly rockfishContext _context;
+        private readonly rockfishContext ct;
 
         public SiteController(rockfishContext context)
         {
-            _context = context;
+            ct = context;
         }
 
 
@@ -27,7 +27,7 @@ namespace rockfishCore.Controllers
         [HttpGet("{id}/purchases")]
         public IEnumerable<Purchase> GetPurchases([FromRoute] long id)
         {
-            var l = _context.Purchase
+            var l = ct.Purchase
                  .Where(b => b.SiteId.Equals(id))
                  .OrderByDescending(b => b.PurchaseDate)
                  .ToList();
@@ -46,16 +46,16 @@ namespace rockfishCore.Controllers
         //     return l;
         // }
 
-         //Get api/site/77/activepurchases
+        //Get api/site/77/activepurchases
         [HttpGet("{id}/activepurchases")]
         public IEnumerable<dtoNameIdItem> GetActivePurchases([FromRoute] long id)
         {
-            var res = from c in _context.Purchase
+            var res = from c in ct.Purchase
             .Where(c => c.SiteId.Equals(id))
-            .Where(c => c.CancelDate==null)
+            .Where(c => c.CancelDate == null)
             .OrderByDescending(c => c.PurchaseDate)
                       select new dtoNameIdItem
-                      {                         
+                      {
                           id = c.Id,
                           name = c.Name
                       };
@@ -94,7 +94,7 @@ namespace rockfishCore.Controllers
                 return BadRequest(ModelState);
             }
 
-            var ret = await _context.Site
+            var ret = await ct.Site
             .Select(r => new { r.Id, r.Name, r.CustomerId })
                            .Where(r => r.Id == id)
                            .FirstAsync();
@@ -107,7 +107,7 @@ namespace rockfishCore.Controllers
         [HttpGet]
         public IEnumerable<Site> GetSite()
         {
-            return _context.Site;
+            return ct.Site;
         }
 
         // GET: api/Site/5
@@ -119,7 +119,7 @@ namespace rockfishCore.Controllers
                 return BadRequest(ModelState);
             }
 
-            var site = await _context.Site.AsNoTracking().SingleOrDefaultAsync(m => m.Id == id);
+            var site = await ct.Site.AsNoTracking().SingleOrDefaultAsync(m => m.Id == id);
 
             if (site == null)
             {
@@ -143,11 +143,11 @@ namespace rockfishCore.Controllers
                 return BadRequest();
             }
 
-            _context.Entry(site).State = EntityState.Modified;
+            ct.Entry(site).State = EntityState.Modified;
 
             try
             {
-                await _context.SaveChangesAsync();
+                await ct.SaveChangesAsync();
             }
             catch (DbUpdateConcurrencyException)
             {
@@ -173,8 +173,8 @@ namespace rockfishCore.Controllers
                 return BadRequest(ModelState);
             }
 
-            _context.Site.Add(site);
-            await _context.SaveChangesAsync();
+            ct.Site.Add(site);
+            await ct.SaveChangesAsync();
 
             return CreatedAtAction("GetSite", new { id = site.Id }, site);
         }
@@ -190,7 +190,7 @@ namespace rockfishCore.Controllers
             }
 
 
-            var site = await _context.Site.SingleOrDefaultAsync(m => m.Id == id);
+            var site = await ct.Site.SingleOrDefaultAsync(m => m.Id == id);
 
             if (site == null)
             {
@@ -198,15 +198,21 @@ namespace rockfishCore.Controllers
             }
 
 
-            _context.Site.Remove(site);
-            await _context.SaveChangesAsync();
+            if (await ct.License.Where(z => z.SiteId == id).AnyAsync())
+            {
+                return BadRequest("This site has licenses and can not be deleted");
+            }
+
+
+            ct.Site.Remove(site);
+            await ct.SaveChangesAsync();
 
             return Ok(site);
         }
 
         private bool SiteExists(long id)
         {
-            return _context.Site.Any(e => e.Id == id);
+            return ct.Site.Any(e => e.Id == id);
         }
     }
 }