diff --git a/TwilioSMSReceiver/Controllers/SMSReceiverController.cs b/TwilioSMSReceiver/Controllers/SMSReceiverController.cs index ffd8b81..26c2b6d 100644 --- a/TwilioSMSReceiver/Controllers/SMSReceiverController.cs +++ b/TwilioSMSReceiver/Controllers/SMSReceiverController.cs @@ -19,17 +19,25 @@ namespace TwilioSMSReceiver.Controllers { private readonly SMSDbCtx _context; private readonly IEnumerable _messageHandlers; - public SMSReceiverController(SMSDbCtx context, IEnumerable messageHandlers) + private readonly IWebHostEnvironment _environment; + + public SMSReceiverController(SMSDbCtx context, IEnumerable messageHandlers, + IWebHostEnvironment environment) { _context = context; _messageHandlers = messageHandlers; + _environment = environment; } // GET: api/SMSReceiver [HttpGet] public async Task>> GetSMSMessages() { - return await _context.SMSMessages.Include(b => b.MMSContent).ToListAsync(); + if (_environment.IsDevelopment()) + { + return await _context.SMSMessages.Include(b => b.MMSContent).ToListAsync(); + } + return NotFound(); } // GET: api/SMSReceiver/5