defmodule Plausible.Ingestion.RequestTest do
  use Plausible.DataCase, async: true

  import Phoenix.ConnTest
  import Plug.Conn

  alias Plausible.Ingestion.Request

  test "request cannot be built from conn without payload" do
    conn = build_conn(:post, "/api/events", %{})
    assert {:error, changeset} = Request.build(conn)

    errors = Keyword.keys(changeset.errors)
    assert :event_name in errors
    assert :domain in errors
    assert :url in errors
  end

  test "request cannot be built from non-json payload" do
    conn = build_conn(:post, "/api/events", "defnotjson")
    assert {:error, changeset} = Request.build(conn)
    assert changeset.errors[:request]
  end

  test "request can be built from URL alone" do
    payload = %{
      name: "pageview",
      url: "http://dummy.site"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, %Plug.Conn{}} = Request.build(conn)

    assert request.domains == ["dummy.site"]
    assert request.event_name == "pageview"
    assert request.pathname == "/"
    assert request.remote_ip == "127.0.0.1"
    assert %NaiveDateTime{} = request.timestamp
    assert request.user_agent == nil
    assert request.hostname == "dummy.site"
    assert request.uri.host == "dummy.site"
    assert request.uri.scheme == "http"
    assert request.props == %{}
  end

  @tag :slow
  test "requests include moving timestamp" do
    payload = %{
      name: "pageview",
      url: "http://dummy.site"
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request1, _conn} = Request.build(conn)
    :timer.sleep(1500)
    assert {:ok, request2, _conn} = Request.build(conn)

    ts1 = request1.timestamp
    ts2 = request2.timestamp

    assert NaiveDateTime.compare(ts1, ts2) == :lt
  end

  test "request can be built with domain" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.domains == ["dummy.site"]
    assert request.uri.host == "dummy.site"
  end

  test "request can be built with domain using shorthands" do
    payload = %{
      n: "pageview",
      d: "dummy.site",
      u: "http://dummy.site/index"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.domains == ["dummy.site"]
    assert request.uri.host == "dummy.site"
  end

  test "request can be built for multiple domains" do
    payload = %{
      n: "pageview",
      d: "dummy.site,crash.site",
      u: "http://dummy.site/index"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.domains == ["dummy.site", "crash.site"]
    assert request.uri.host == "dummy.site"
  end

  test "request can be built with numeric event name" do
    payload = %{
      n: 404,
      d: "dummy.site",
      u: "http://dummy.site/index"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.event_name == "404"
  end

  test "hostname is (none) if host-less uri provided" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "about:config"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.hostname == "(none)"
  end

  test "hostname is set" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.hostname == "dummy.site"
  end

  test "user agent is set as-is" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html"
    }

    conn = build_conn(:post, "/api/events", payload) |> put_req_header("user-agent", "Mozilla")
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.user_agent == "Mozilla"
  end

  test "request params are set" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      referrer: "https://example.com",
      hashMode: 1,
      props: %{
        "custom1" => "property1",
        "custom2" => "property2"
      },
      v: 137
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert request.referrer == "https://example.com"
    assert request.hash_mode == 1
    assert request.props["custom1"] == "property1"
    assert request.props["custom2"] == "property2"
    assert request.tracker_script_version == 137
  end

  @tag :ee_only
  test "parses revenue source field from a json string" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      revenue: "{\"amount\":20.2,\"currency\":\"EUR\"}"
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert %Money{amount: amount, currency: :EUR} = request.revenue_source
    assert Decimal.new("20.2") == amount
  end

  @tag :ee_only
  test "sets revenue source with integer amount" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      revenue: %{
        "amount" => 20,
        "currency" => "USD"
      }
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert %Money{amount: amount, currency: :USD} = request.revenue_source
    assert Decimal.equal?(amount, Decimal.new("20.0"))
  end

  @tag :ee_only
  test "sets revenue source with float amount" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      revenue: %{
        "amount" => 20.1,
        "currency" => "USD"
      }
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert %Money{amount: amount, currency: :USD} = request.revenue_source
    assert Decimal.equal?(amount, Decimal.new("20.1"))
  end

  @tag :ee_only
  test "parses string amounts into money structs" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      revenue: %{
        "amount" => "12.3",
        "currency" => "USD"
      }
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert %Money{amount: amount, currency: :USD} = request.revenue_source
    assert Decimal.equal?(amount, Decimal.new("12.3"))
  end

  @tag :ee_only
  test "ignores revenue data when currency is invalid" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      revenue: %{
        "amount" => 1233.2,
        "currency" => "EEEE"
      }
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert is_nil(request.revenue_source)
  end

  test "pathname is set" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/pictures/index.html#foo"
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)

    assert request.pathname == "/pictures/index.html"
  end

  test "pathname is set with hashMode" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/pictures/index.html#foo",
      hashMode: 1
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)

    assert request.pathname == "/pictures/index.html#foo"
  end

  for event_name <- Plausible.Event.SystemEvents.events_with_path_prop() do
    test "event.props.path is synced from event.pathname for special path-based event '#{event_name}'" do
      payload = %{
        name: unquote(event_name),
        domain: "dummy.site",
        url: "http://dummy.site/pictures/index.html#foo"
      }

      conn = build_conn(:post, "/api/events", payload)

      assert {:ok, request, _conn} = Request.build(conn)

      assert request.pathname == "/pictures/index.html"
      assert request.props == %{"path" => "/pictures/index.html"}
    end

    test "event.props.path is synced from event.pathname for special path-based event '#{event_name}' with hashMode" do
      payload = %{
        name: unquote(event_name),
        domain: "dummy.site",
        url: "http://dummy.site/pictures/index.html#foo",
        hashMode: 1
      }

      conn = build_conn(:post, "/api/events", payload)

      assert {:ok, request, _conn} = Request.build(conn)

      assert request.pathname == "/pictures/index.html#foo"
      assert request.props == %{"path" => "/pictures/index.html#foo"}
    end

    test "event.props.path is not synced from event.pathname for special path-based event '#{event_name}' if it's set explicitly (legacy support)" do
      payload = %{
        name: unquote(event_name),
        domain: "dummy.site",
        url: "http://dummy.site/pictures/index.html#foo",
        props: %{"path" => "/album"}
      }

      conn = build_conn(:post, "/api/events", payload)

      assert {:ok, request, _conn} = Request.build(conn)

      assert request.pathname == "/pictures/index.html"
      assert request.props == %{"path" => "/album"}
    end
  end

  test "query params are set" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/pictures/index.html?foo=bar&baz=bam"
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert request.query_params["foo"] == "bar"
    assert request.query_params["baz"] == "bam"
  end

  test "returns validation error when using data uris" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "data:text/html,%3Cscript%3Ealert%28%27hi%27%29%3B%3C%2Fscript%3E"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)
    assert {"scheme is not allowed", _} = changeset.errors[:url]
  end

  test "returns validation error when url is too long" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "https://dummy.site/#{String.duplicate("a", 5000)}"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)
    assert {"must be a valid url", _} = changeset.errors[:url]
  end

  test "returns validation error when event name is too long" do
    payload = %{
      name: String.duplicate("a", 500),
      domain: "dummy.site",
      url: "https://dummy.site/"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)
    assert {"should be at most %{count} character(s)", _} = changeset.errors[:event_name]
  end

  test "returns validation error when event name is blank" do
    payload = %{
      name: nil,
      domain: "dummy.site",
      url: "https://dummy.site/"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)
    assert {"can't be blank", _} = changeset.errors[:event_name]
  end

  test "returns validation error when event name cannot be cast to string" do
    payload = %{
      name: ["list", "of", "things"],
      domain: "dummy.site",
      url: "https://dummy.site/"
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)
    assert {"is invalid", _} = changeset.errors[:event_name]
  end

  test "truncates referrer when too long" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "https://dummy.site/",
      referrer: String.duplicate("a", 2500)
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:ok, request, _conn} = Request.build(conn)
    assert request.referrer == String.duplicate("a", 2000)
  end

  test "returns validation error when props keys are too long" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "https://dummy.site/",
      props: %{String.duplicate("a", 500) => "abc"}
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)

    assert {"keys should have at most 300 bytes and values 2000 bytes", _} =
             changeset.errors[:props]
  end

  test "returns validation error when props values are too long" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "https://dummy.site/",
      props: %{"abc" => String.duplicate("a", 2500)}
    }

    conn = build_conn(:post, "/api/events", payload)
    assert {:error, changeset} = Request.build(conn)

    assert {"keys should have at most 300 bytes and values 2000 bytes", _} =
             changeset.errors[:props]
  end

  test "trims prop list to 30 items when sending too many items" do
    payload = %{
      name: "pageview",
      domain: "dummy.site",
      url: "http://dummy.site/index.html",
      referrer: "https://example.com",
      hashMode: 1,
      props: for(i <- 1..50, do: {"#{i}", "foo"}, into: %{})
    }

    conn = build_conn(:post, "/api/events", payload)

    assert {:ok, request, _conn} = Request.build(conn)
    assert map_size(request.props) == 30
  end

  test "malicious input, technically valid json" do
    conn = build_conn(:post, "/api/events", "\"<script>\"")
    assert {:error, changeset} = Request.build(conn)
    assert changeset.errors[:request]
  end

  test "long body length" do
    payload = """
    {
      "name": "pageview",
      "domain": "dummy.site",
      "url": "#{:binary.copy("a", 1_000)}"
    }
    """

    within_read_limit =
      :post
      |> build_conn("/api/events", payload)
      |> Request.build()

    assert {:ok, _, _} = within_read_limit

    exceeding_read_limit =
      :post
      |> build_conn("/api/events", payload)
      |> Plug.Conn.assign(:read_body_limit, 800)
      |> Request.build()

    assert {:error, _} = exceeding_read_limit
  end

  for event_name <-
        Plausible.Event.SystemEvents.events_with_interactive_always_true() --
          Plausible.Event.SystemEvents.events_with_engagement_props() do
    test "ignores interactive: false parameter for #{event_name}" do
      params = %{
        name: unquote(event_name),
        domain: "dummy.site",
        url: "http://dummy.site/index.html",
        interactive: false
      }

      assert {:ok, request, _conn} =
               build_conn(:post, "/api/events", params)
               |> put_req_header("user-agent", "Mozilla")
               |> Request.build()

      assert request.interactive?
    end
  end

  for event_name <- Plausible.Event.SystemEvents.events_with_engagement_props() do
    test "ignores interactive: false parameter for #{event_name}" do
      params = %{
        name: unquote(event_name),
        domain: "dummy.site",
        url: "http://dummy.site/index.html",
        sd: 10,
        e: 100,
        interactive: false
      }

      assert {:ok, request, _conn} =
               build_conn(:post, "/api/events", params)
               |> put_req_header("user-agent", "Mozilla")
               |> Request.build()

      assert request.interactive?
    end
  end

  for param_key <- [:interactive, :i] do
    test "allows marking user-defined custom events as non-interactive with payload #{Atom.to_string(param_key)}: false" do
      params =
        %{
          name: "ping",
          domain: "dummy.site",
          url: "http://dummy.site/index.html"
        }
        |> Map.put(unquote(param_key), false)

      assert {:ok, request, _conn} =
               build_conn(:post, "/api/events", params)
               |> put_req_header("user-agent", "Mozilla")
               |> Request.build()

      refute request.interactive?
    end
  end

  @tag :ee_only
  test "encodable" do
    params = %{
      name: "pageview",
      domain: "dummy.site",
      url: "https://dummy.site/pictures/index.html?foo=bar&baz=bam",
      referrer: "https://example.com",
      props: %{"abc" => "qwerty", "hello" => "world"},
      hashMode: 1,
      revenue: %{
        "amount" => "12.3",
        "currency" => "USD"
      }
    }

    assert {:ok, request, _conn} =
             build_conn(:post, "/api/events", params)
             |> put_req_header("user-agent", "Mozilla")
             |> Request.build()

    request = request |> Jason.encode!() |> Jason.decode!()

    assert Map.drop(request, ["timestamp"]) == %{
             "domains" => ["dummy.site"],
             "event_name" => "pageview",
             "hash_mode" => 1,
             "hostname" => "dummy.site",
             "pathname" => "/pictures/index.html",
             "props" => %{"abc" => "qwerty", "hello" => "world"},
             "query_params" => %{"baz" => "bam", "foo" => "bar"},
             "referrer" => "https://example.com",
             "remote_ip" => "127.0.0.1",
             "revenue_source" => %{"amount" => "12.3", "currency" => "USD"},
             "uri" => "https://dummy.site/pictures/index.html?foo=bar&baz=bam",
             "user_agent" => "Mozilla",
             "ip_classification" => nil,
             "scroll_depth" => nil,
             "engagement_time" => nil,
             "tracker_script_version" => 0,
             "interactive?" => true
           }

    assert %NaiveDateTime{} = NaiveDateTime.from_iso8601!(request["timestamp"])
  end

  describe "engagement event" do
    test "fails validation if no engagement metrics are present or are invalid" do
      [%{}, %{sd: -1}, %{e: "abc"}, %{sd: "null", e: "null"}]
      |> Enum.each(fn invalid_param_set ->
        assert {:error, changeset} = build_engagement_request(invalid_param_set)
        assert {msg, _} = changeset.errors[:event_name]
        assert msg == Request.blank_engagement_error_message()
      end)
    end

    test "sets valid scroll_depth" do
      assert {:ok, %Request{scroll_depth: 0}, _conn} = build_engagement_request(%{sd: 0})
      assert {:ok, %Request{scroll_depth: 25}, _conn} = build_engagement_request(%{sd: 25})
      assert {:ok, %Request{scroll_depth: 25}, _conn} = build_engagement_request(%{sd: "25"})
    end

    test "scroll_depth defaults to 100 when given integer is greater than 100" do
      assert {:ok, %Request{scroll_depth: 100}, _conn} = build_engagement_request(%{sd: 101})
      assert {:ok, %Request{scroll_depth: 100}, _conn} = build_engagement_request(%{sd: "101"})
    end

    test "scroll_depth defaults to 255 when given value is invalid or missing" do
      assert {:ok, %Request{scroll_depth: 255}, _conn} = build_engagement_request(%{sd: -1, e: 1})

      assert {:ok, %Request{scroll_depth: 255}, _conn} =
               build_engagement_request(%{sd: "abc", e: 1})

      assert {:ok, %Request{scroll_depth: 255}, _conn} = build_engagement_request(%{e: 1})
    end

    test "sets valid engagement_time" do
      assert {:ok, %Request{engagement_time: 123}, _conn} = build_engagement_request(%{e: 123})
      assert {:ok, %Request{engagement_time: 123}, _conn} = build_engagement_request(%{e: "123"})
    end

    test "engagement_time defaults to 0 when given value is invalid or missing" do
      assert {:ok, %Request{engagement_time: 0}, _conn} =
               build_engagement_request(%{sd: 1, e: -1})

      assert {:ok, %Request{engagement_time: 0}, _conn} =
               build_engagement_request(%{sd: 1, e: "abc"})

      assert {:ok, %Request{engagement_time: 0}, _conn} =
               build_engagement_request(%{sd: 1, e: Request.too_large_engagement_time()})
    end

    test "sd and e fields are ignored if name is not engagement" do
      params = %{name: "pageview", domain: "site.com", url: "https://site.com", sd: 25, e: 1000}

      assert {:ok, %Request{engagement_time: nil, scroll_depth: nil}, _conn} =
               build_conn(:post, "/api/events", params)
               |> Request.build()

      assert {:ok, %Request{engagement_time: nil, scroll_depth: nil}, _conn} =
               build_conn(:post, "/api/events", Map.put(params, :name, "Custom Event"))
               |> Request.build()
    end

    test "ingests valid scroll_depth and engagement_time for the same request" do
      assert {:ok, %Request{scroll_depth: 23, engagement_time: 100}, _conn} =
               build_engagement_request(%{sd: 23, e: 100})

      assert {:ok, %Request{scroll_depth: 23, engagement_time: 100}, _conn} =
               build_engagement_request(%{sd: "23", e: "100"})
    end
  end

  defp build_engagement_request(extra_params) do
    params =
      %{name: "engagement", domain: "dummy.site", url: "https://dummy.site/"}
      |> Map.merge(extra_params)

    build_conn(:post, "/api/events", params)
    |> Request.build()
  end
end