1 files changed, 91 insertions, 0 deletions
diff --git a/mediagoblin/tools/federation.py b/mediagoblin/tools/federation.py
new file mode 100644
index 00000000..6c2d27da
--- /dev/null
+++ b/mediagoblin/tools/federation.py
@@ -0,0 +1,91 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2014 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from mediagoblin.db.models import Activity, Generator, User
+
+def create_generator(request):
+    """
+    This creates a Generator object based on the Client associated with the
+    OAuth credentials used. If the request has invalid OAuth credentials or
+    no OAuth credentials None is returned.
+    """
+    if not hasattr(request, "access_token"):
+        return None
+
+    client = request.access_token.get_requesttoken.get_client
+    
+    # Check if there is a generator already
+    generator = Generator.query.filter_by(
+        name=client.application_name,
+        object_type="client"
+    ).first()
+    if generator is None:
+        generator = Generator(
+            name=client.application_name,
+            object_type="client"
+        )
+        generator.save()
+
+    return generator
+    
+     
+
+def create_activity(verb, obj, actor, target=None, generator=None):
+    """
+    This will create an Activity object which for the obj if possible
+    and save it. The verb should be one of the following:
+        add, author, create, delete, dislike, favorite, follow
+        like, post, share, unfollow, unfavorite, unlike, unshare,
+        update, tag.
+
+    If none of those fit you might not want/need to create an activity for
+    the object. The list is in mediagoblin.db.models.Activity.VALID_VERBS
+    """
+    # exception when we try and generate an activity with an unknow verb
+    # could change later to allow arbitrary verbs but at the moment we'll play
+    # it safe.
+
+    if verb not in Activity.VALID_VERBS:
+        raise ValueError("A invalid verb type has been supplied.")
+
+    if generator is None:
+        # This should exist as we're creating it by the migration for Generator
+        generator = Generator.query.filter_by(name="GNU MediaGoblin").first()
+        if generator is None:
+            generator = Generator(
+                name="GNU MediaGoblin",
+                object_type="service"
+            )
+            generator.save()
+
+    activity = Activity(verb=verb)
+    activity.set_object(obj)
+
+    if target is not None:
+        activity.set_target(target)
+
+   # If they've set it override the actor from the obj.
+    activity.actor = actor.id if isinstance(actor, User) else actor
+
+    activity.generator = generator.id
+    activity.save()
+
+    # Sigh want to do this prior to save but I can't figure a way to get
+    # around relationship() not looking up object when model isn't saved.
+    if activity.generate_content():
+        activity.save()
+
+    return activity