From 373753e595de96a0492edfc6cc180c3e47696a55 Mon Sep 17 00:00:00 2001
From: Roger Braun <roger@rogerbraun.net>
Date: Tue, 9 May 2017 18:11:51 +0200
Subject: [PATCH] Add some basic changesets.

---
 lib/pleroma/object.ex |  9 ++++++++-
 lib/pleroma/user.ex   | 14 ++++++++++++++
 test/object_test.exs  | 13 ++++++++++++-
 test/user_test.exs    | 41 +++++++++++++++++++++++++++++++++++++++++
 4 files changed, 75 insertions(+), 2 deletions(-)

diff --git a/lib/pleroma/object.ex b/lib/pleroma/object.ex
index 949ccb0f6..715a35591 100644
--- a/lib/pleroma/object.ex
+++ b/lib/pleroma/object.ex
@@ -1,7 +1,7 @@
 defmodule Pleroma.Object do
   use Ecto.Schema
   alias Pleroma.{Repo, Object}
-  import Ecto.Query
+  import Ecto.{Query, Changeset}
 
   schema "objects" do
     field :data, :map
@@ -9,6 +9,13 @@ defmodule Pleroma.Object do
     timestamps()
   end
 
+  def change(struct, params \\ %{}) do
+    changeset = struct
+    |> cast(params, [:data])
+    |> validate_required([:data])
+    |> unique_constraint(:ap_id, name: :objects_unique_apid_index)
+  end
+
   def get_by_ap_id(ap_id) do
     Repo.one(from object in Object,
       where: fragment("? @> ?", object.data, ^%{id: ap_id}))
diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
index 4510be770..9b2433674 100644
--- a/lib/pleroma/user.ex
+++ b/lib/pleroma/user.ex
@@ -61,6 +61,17 @@ defmodule Pleroma.User do
     }
   end
 
+  @email_regex ~r/^[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$/
+  def remote_user_creation(params) do
+    changeset = %User{}
+    |> cast(params, [:bio, :name, :ap_id, :nickname, :info])
+    |> validate_required([:bio, :name, :ap_id, :nickname])
+    |> unique_constraint(:nickname)
+    |> validate_format(:nickname, @email_regex)
+    |> validate_length(:bio, max: 1000)
+    |> validate_length(:name, max: 100)
+  end
+
   def register_changeset(struct, params \\ %{}) do
     changeset = struct
     |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation])
@@ -69,6 +80,9 @@ defmodule Pleroma.User do
     |> unique_constraint(:email)
     |> unique_constraint(:nickname)
     |> validate_format(:nickname, ~r/^[a-zA-Z\d]+$/)
+    |> validate_format(:email, @email_regex)
+    |> validate_length(:bio, max: 1000)
+    |> validate_length(:name, max: 100)
 
     if changeset.valid? do
       hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
diff --git a/test/object_test.exs b/test/object_test.exs
index 3ffddbbc5..ae9515c26 100644
--- a/test/object_test.exs
+++ b/test/object_test.exs
@@ -1,11 +1,22 @@
 defmodule Pleroma.ObjectTest do
   use Pleroma.DataCase
   import Pleroma.Factory
+  alias Pleroma.{Repo, Object}
 
   test "returns an object by it's AP id" do
     object = insert(:note)
-    found_object = Pleroma.Object.get_by_ap_id(object.data["id"])
+    found_object = Object.get_by_ap_id(object.data["id"])
 
     assert object == found_object
   end
+
+  describe "generic changeset" do
+    test "it ensures uniqueness of the id" do
+      object = insert(:note)
+      cs = Object.change(%Object{}, %{data: %{id: object.data["id"]}})
+      assert cs.valid?
+
+      {:error, result} = Repo.insert(cs)
+    end
+  end
 end
diff --git a/test/user_test.exs b/test/user_test.exs
index e6de4a5cd..bfa79db28 100644
--- a/test/user_test.exs
+++ b/test/user_test.exs
@@ -139,5 +139,46 @@ defmodule Pleroma.UserTest do
     user = insert(:user)
     assert User.ap_followers(user) == Pleroma.Web.Router.Helpers.o_status_url(Pleroma.Web.Endpoint, :feed_redirect, user.nickname) <> "/followers"
   end
+
+  describe "remote user creation changeset" do
+    @valid_remote %{
+      bio: "hello",
+      name: "Someone",
+      nickname: "a@b.de",
+      ap_id: "http...",
+      info: %{ some: "info" }
+    }
+
+    test "it confirms validity" do
+      cs = User.remote_user_creation(@valid_remote)
+      assert cs.valid?
+    end
+
+    test "it enforces the fqn format for nicknames" do
+      cs = User.remote_user_creation(%{@valid_remote | nickname: "bla"})
+      refute cs.valid?
+    end
+
+    test "it has required fields" do
+      [:bio, :name, :nickname, :ap_id]
+      |> Enum.each(fn (field) ->
+        cs = User.remote_user_creation(Map.delete(@valid_remote, field))
+        refute cs.valid?
+      end)
+    end
+
+    test "it restricts some sizes" do
+      [bio: 1000, name: 100]
+      |> Enum.each(fn ({field, size}) ->
+        string = String.pad_leading(".", size)
+        cs = User.remote_user_creation(Map.put(@valid_remote, field, string))
+        assert cs.valid?
+
+        string = String.pad_leading(".", size + 1)
+        cs = User.remote_user_creation(Map.put(@valid_remote, field, string))
+        refute cs.valid?
+      end)
+    end
+  end
 end