using System;
using System.Web.UI;
using System.Web.UI.WebControls;
using System.ComponentModel;
using System.Web;
using System.Configuration;
using System.IO;
using System.Web.Security;
using DownLoadTracker;
using Microsoft.ApplicationBlocks.Data;
using System.Data;
using System.Data.SqlClient;
using System.Text.RegularExpressions;
namespace UmbracoDocumentSecurity
{

      /// <summary>
      /// To intercept requests for file types and check for an authenticated user
      /// </summary>
      public class SecuringHandler : System.Web.IHttpHandler
      {

            #region IHttpHandler Members

            public void ProcessRequest(HttpContext context)
            {
                  if(context.Request.Cookies["umbracoMemberGuid"] != null)
                  {
                        //log into tracker that we are downloading
                        logDownload(context);
                        TransmitDocument(context);
                  }
                  else
                  {
                        RedirectToLogin(context);
                  }
            }

            /// <summary>
            /// There *has* to be a better way than this, possibly using the metabase API.
            /// </summary>
            /// <param name="extension"></param>
            /// <returns></returns>
            private string GetMime(string extension)
            {
                  switch(extension)
                  {
                        case "doc":
                        {
                              return "application/ms-word";
                        }
                        case "pdf":
                        {
                              return "application/pdf";
                        }
                        case "mpeg":
                        {
                              return "application/mpeg";
                        }
                        default:
                        {
                              return "application/unknown";
                        }
                  }
            }

            /// <summary>
            /// writes into the database the download and who is downloading
            /// </summary>
            private void logDownload(HttpContext context){
                  string memberId = "un known";
                  //umbracoMemberId
                  if(context.Request.Cookies["umbracoMemberId"] != null)
                  {
                        memberId = context.Request.Cookies["umbracoMemberId"].Value;
                  }
                  
                  //context.Request.Url.ToString() need to get item id
                  Tracker.AddTrackingEntry(memberId ,getMediaIdFromUrl(context.Request.Url.ToString(),context),DateTime.Now);
            }

            private int getMediaIdFromUrl(string url, HttpContext context){

                  Regex reg = new Regex("/media.*");
                  int id;
                  try
                  {
                        SqlParameter[] sqParams = {new SqlParameter("@url", reg.Match(url).Value)};
                        string sql = "select contentNodeId from cmsPropertyData where dataNvarchar = @url";                                    
                        id= (int)SqlHelper.ExecuteScalar(umbraco.GlobalSettings.DbDSN,CommandType.Text,sql,sqParams);
                        return id;
                  }
                  catch(Exception ex){
                         umbraco.BusinessLogic.Log.Add(umbraco.BusinessLogic.LogTypes.Error,new umbraco.BusinessLogic.User(0),0,"Error from download security handler ->" + ex.Message.ToString());
                  }
                  return 0;
                  
                  
            }

            private void TransmitDocument(HttpContext context)
            {
                  string path      = HttpUtility.UrlDecode(context.Request.Url.LocalPath);
                  string fileName  = System.IO.Path.GetFileName(path);
                  string extension = System.IO.Path.GetExtension(path);
                  
                  context.Response.Clear();
                  context.Response.AddHeader("content-disposition","attachment");
                  context.Response.ContentType = GetMime(extension);
                  context.Response.WriteFile(path);
            }

            private void RedirectToLogin(HttpContext context)
            {
                  context.Response.Redirect(ConfigurationSettings.AppSettings["LoginPage"]);
            }

            public bool IsReusable
            {
                  get
                  {
                        return true;
                  }
            }



            #endregion
      }


}