package api

import (
  "net/http"
  "log"
  "github.com/mssola/user_agent"
  "github.com/dannyvankooten/ana/models"
  "github.com/dannyvankooten/ana/core"
)

func CollectHandler(w http.ResponseWriter, r *http.Request) {
  ua := user_agent.New(r.UserAgent())

  // abort if this is a bot.
  if ua.Bot() {
    return
  }

  // prepare statement for inserting data
  stmt, err := core.DB.Prepare(`INSERT INTO visits(
    ip_address,
    path,
    referrer_url,
    browser_language,
    browser_name,
    browser_version,
    device_os,
    screen_resolution
    ) VALUES( ?, ?, ?, ?, ?, ?, ?, ? )`)
  if err != nil {
      log.Fatal(err.Error())
  }
  defer stmt.Close()

  // TODO: Mask IP Address
  // TODO: Query DB to determine whether visitor is returning
  ipAddress := r.RemoteAddr
  headerForwardedFor := r.Header.Get("X-Forwarded-For")
  if( headerForwardedFor != "" ) {
    ipAddress = headerForwardedFor
  }

  q := r.URL.Query()
  visit := models.Visit{
    Path: q.Get("p"),
    IpAddress: ipAddress,
    ReferrerUrl: q.Get("r"),
    BrowserLanguage: q.Get("l"),
    ScreenResolution: q.Get("sr"),
  }

  // add browser details
  visit.BrowserName, visit.BrowserVersion = ua.Browser()

  // add device details
  visit.DeviceOS = ua.OS()

  _, err = stmt.Exec(
    visit.IpAddress,
    visit.Path,
    visit.ReferrerUrl,
    visit.BrowserLanguage,
    visit.BrowserName,
    visit.BrowserVersion,
    visit.DeviceOS,
    visit.ScreenResolution,
  )
  if err != nil {
    log.Fatal(err)
  }

  // don't you cache this
  w.Header().Set("Content-Type", "image/gif")
  w.Header().Set("Expires", "Mon, 01 Jan 1990 00:00:00 GMT")
  w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
  w.Header().Set("Pragma", "no-cache")
  w.Header().Set("Status", "200")
}