path: root/SendGrid/SendGridMail/SendGrid.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Net.Mail;
using System.Net.Mime;
using System.Runtime.InteropServices.ComTypes;
using System.Text;

namespace SendGridMail
{
    public class SendGrid : ISendGrid
    {
        private IHeader header;

        private Dictionary<String, String> _filters;

        //apps list and settings
        private const String ReText = @"<\%\s*\%>";
        private const String ReHtml = @"<\%\s*[^\s]+\s*\%>";

        public SendGrid(IHeader header)
        {
            this.header = header;
            
            //initialize the filters, for use within the library
            this.InitializeFilters();
        }

        public void InitializeFilters()
        {
            this._filters = 
            new Dictionary<string, string>
            {
                {"Gravatar", "gravatar"},
                {"OpenTracking", "opentrack"},
                {"ClickTracking", "clicktrack"},
                {"SpamCheck", "spamcheck"},
                {"Unsubscribe", "subscriptiontrack"},
                {"Footer", "footer"},
                {"GoogleAnalytics", "ganalytics"},
                {"Template", "template"},
                {"Bcc", "bcc"},
                {"BypassListManagement", "bypass_list_management"}
            };
        }

        private MailMessage message;

        // TODO find appropriate types for these
        const string encoding = "quoted-printable";
        const string charset = "utf-8";

    /*
            if (Html != null )
            {
                AlternateView htmlView = AlternateView.CreateAlternateViewFromString(html, null, "text/html");
                message.AlternateViews.Add(htmlView);
            }

            if (Text != null )
            {
                AlternateView plainView = AlternateView.CreateAlternateViewFromString(Text, null, "text/plain");
                message.AlternateViews.Add(plainView);
            }

            message.BodyEncoding = Encoding.GetEncoding(charset);                
     */

        public SendGrid(MailAddress from, MailAddress[] to, MailAddress[] cc, MailAddress[] bcc, 
            String subject, String html, String text, TransportType transport, IHeader header = null )
        {
            message = new MailMessage();
            Header = header;

            From = from;
            To = to;

            _subs = new Dictionary<string, string>();

            message.Subject = subject;
            message.SubjectEncoding = Encoding.GetEncoding(charset);

            Text = text;
            Html = html;
        }

        #region Properties
        public MailAddress From
        {
            get
            {
                return message.From;
            }
            set
            {
                if (value != null) message.From = value;
            }
        }

        public MailAddress[] To
        {
            get
            {
                return message.To.ToArray();
            }
            set
            {
                message.To.Clear();
                foreach (var mailAddress in value)
                {
                    message.To.Add(mailAddress);
                }
            }
        }

        public MailAddress[] Cc
        {
            get
            {
                return message.CC.ToArray();
            }
            set
            {
                message.CC.Clear();
                foreach (var mailAddress in value)
                {
                    message.CC.Add(mailAddress);
                }
            }
        }

        public MailAddress[] Bcc
        {
            get
            {
                return message.Bcc.ToArray();
            }
            set
            {
                message.Bcc.Clear();
                foreach (var mailAddress in value)
                {
                    message.Bcc.Add(mailAddress);
                }
            }
        }

        public String Subject
        {
            get
            {
                return message.Subject;
            }
            set
            {
                if (value != null) message.Subject = value;
            }
        }

        public IHeader Header
        {
            get { throw new NotImplementedException(); }
            set { throw new NotImplementedException(); }
        }

        public String Html { get; set; }
        public String Text { get; set; }

        public TransportType Transport { get; set; }
        #endregion

        #region Methods for setting data
        public void AddTo(String address)
        {
            var mailAddress = new MailAddress(address);
            message.To.Add(mailAddress);
        }

        public void AddTo(IEnumerable<String> addresses)
        {
            if (addresses != null)
            {
                foreach (var address in addresses)
                {
                    if (address != null) AddTo(address);
                }
            }
        }

        public void AddTo(IDictionary<String, IDictionary<String, String>> addresssInfo)
        {
            foreach (var address in addresssInfo.Keys)
            {
                var table = addresssInfo[address];
                //DisplayName is the only option that this implementation of MailAddress implements.
                var mailAddress = new MailAddress(address, table.ContainsKey("DisplayName") ? table["DisplayName"] : null);
                message.To.Add(mailAddress);
            }
        }

        public void AddCc(String address)
        {
            var mailAddress = new MailAddress(address);
            message.CC.Add(mailAddress);
        }

        public void AddCc(IEnumerable<String> addresses)
        {
            if (addresses != null)
            {
                foreach (var address in addresses)
                {
                    if (address != null) AddCc(address);
                }
            }
        }

        public void AddCc(IDictionary<String, IDictionary<String, String>> addresssInfo)
        {
            foreach (var address in addresssInfo.Keys)
            {
                var table = addresssInfo[address];
                //DisplayName is the only option that this implementation of MailAddress implements.
                var mailAddress = new MailAddress(address, table.ContainsKey("DisplayName") ? table["DisplayName"] : null);
                message.CC.Add(mailAddress);
            }
        }

        public void AddBcc(String address)
        {
            var mailAddress = new MailAddress(address);
            message.Bcc.Add(mailAddress);
        }

        public void AddBcc(IEnumerable<String> addresses)
        {
            if (addresses != null)
            {
                foreach (var address in addresses)
                {
                    if (address != null) AddBcc(address);
                }
            }
        }

        public void AddBcc(IDictionary<String, IDictionary<String, String>> addresssInfo)
        {
            foreach (var address in addresssInfo.Keys)
            {
                var table = addresssInfo[address];
                //DisplayName is the only option that this implementation of MailAddress implements.
                var mailAddress = new MailAddress(address, table.ContainsKey("DisplayName") ? table["DisplayName"] : null);
                message.Bcc.Add(mailAddress);
            }
        }

        private Dictionary<string, string> _subs;

        public void AddSubVal(string tag, string value)
        {
            //let the system complain if they do something bad, since the function returns null
            _subs[tag] = value;
        }

        public void AddAttachment(string filePath)
        {
            var data = new Attachment(filePath, MediaTypeNames.Application.Octet);
            message.Attachments.Add(data);
        }

        public void AddAttachment(Attachment attachment)
        {
            message.Attachments.Add(attachment);
        }

        public void AddAttachment(Stream attachment, ContentType type)
        {
            var data = new Attachment(attachment, type);
        }

        public IEnumerable<string> GetRecipients()
        {
            List<MailAddress> tos = message.To.ToList();
            List<MailAddress> ccs = message.CC.ToList();
            List<MailAddress> bccs = message.Bcc.ToList();

            var rcpts = tos.Union(ccs.Union(bccs)).Select(address => address.Address);
            return rcpts;
        }

        private string Get(string field)
        {
            throw new NotImplementedException();
        }

        private void Set(string field, string value)
        {
            throw new NotImplementedException();
        }
        #endregion

        #region SMTP API Functions
        public void DisableGravatar()
        {
            this.header.Disable(this._filters["Gravatar"]);
        }

        public void DisableOpenTracking()
        {
            this.header.Disable(this._filters["OpenTracking"]);
        }

        public void DisableClickTracking()
        {
            this.header.Disable(this._filters["ClickTracking"]);
        }

        public void DisableSpamCheck()
        {
            this.header.Disable(this._filters["SpamCheck"]);
        }

        public void DisableUnsubscribe()
        {
            this.header.Disable(this._filters["Unsubscribe"]);
        }

        public void DisableFooter()
        {
            this.header.Disable(this._filters["Footer"]);
        }

        public void DisableGoogleAnalytics()
        {
            this.header.Disable(this._filters["GoogleAnalytics"]);
        }

        public void DisableTemplate()
        {
            this.header.Disable(this._filters["Template"]);
        }

        public void DisableBcc()
        {
            this.header.Disable(this._filters["Bcc"]);
        }

        public void DisableBypassListManagement()
        {
            this.header.Disable(this._filters["BypassListManagement"]);
        }

        public void EnableGravatar()
        {
            this.header.Enable(this._filters["Gravatar"]);
        }

        public void EnableOpenTracking()
        {
            this.header.Enable(this._filters["OpenTracking"]);
        }

        public void EnableClickTracking(string text = null)
        {
            var filter = this._filters["ClickTracking"];
                
            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "text" }, text);
        }

        public void EnableSpamCheck(int score = 5, string url = null)
        {
            var filter = this._filters["SpamCheck"];

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "score" }, score.ToString(CultureInfo.InvariantCulture));
            this.header.AddFilterSetting(filter, new List<string>(){ "url" }, url);
        }

        public void EnableUnsubscribe(string text, string html, string replace, string url, string landing)
        {
            var filter = this._filters["Unsubscribe"];

            if(!System.Text.RegularExpressions.Regex.IsMatch(text, SendGrid.ReText))
            {
                throw new Exception("Missing substitution tag in text");
            }

            if(!System.Text.RegularExpressions.Regex.IsMatch(html, SendGrid.ReHtml))
            {
                throw new Exception("Missing substitution tag in html");
            }

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "text" }, text);
            this.header.AddFilterSetting(filter, new List<string>(){ "html" }, html);
            this.header.AddFilterSetting(filter, new List<string>(){ "replace"}, replace);
            this.header.AddFilterSetting(filter, new List<string>(){ "landing" }, landing);
        }

        public void EnableFooter(string text = null, string html = null)
        {
            var filter = this._filters["Footer"];

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "text" }, text);
            this.header.AddFilterSetting(filter, new List<string>(){ "html" }, html);
        }

        public void EnableGoogleAnalytics(string source, string medium, string term, string content = null, string campaign = null)
        {
            var filter = this._filters["GoogleAnalytics"];

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "source " }, source);
            this.header.AddFilterSetting(filter, new List<string>(){ "medium" }, medium);
            this.header.AddFilterSetting(filter, new List<string>(){ "term" }, term);
            this.header.AddFilterSetting(filter, new List<string>(){ "content" }, content);
            this.header.AddFilterSetting(filter, new List<string>(){ "compaign" }, campaign);
        }

        public void EnableTemplate(string html)
        {
            var filter = this._filters["Template"];

            if (!System.Text.RegularExpressions.Regex.IsMatch(html, SendGrid.ReHtml))
            {
                throw new Exception("Missing substitution tag in html");
            }

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "html" }, html);
        }

        public void EnableBcc(string email)
        {
            var filter = this._filters["Bcc"];

            this.header.Enable(filter);
            this.header.AddFilterSetting(filter, new List<string>(){ "email" }, email);
        }

        public void EnableBypassListManagement()
        {
            this.header.Enable(this._filters["BypassListManagement"]);
        }
        #endregion

        public MailMessage CreateMimeMessage()
        {
            String smtpapi = Header.AsJson();

            if (!String.IsNullOrEmpty(smtpapi))
                message.Headers.Add("X-SmtpApi", "{" + smtpapi + "}");

            return message;
        }

        public void Mail()
        {
            throw new NotImplementedException();
        }
    }
}