198 lines
6.8 KiB
Python
198 lines
6.8 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import and_
|
|
from sqlalchemy.orm import joinedload
|
|
|
|
from auth.authenticate import login_required
|
|
from auth.credentials import AuthCredentials
|
|
from base.orm import local_session
|
|
from base.resolvers import mutation
|
|
from orm.rbac import Resource
|
|
from orm.shout import Shout, ShoutAuthor, ShoutTopic
|
|
from orm.topic import TopicFollower, Topic
|
|
from orm.user import User
|
|
from resolvers.zine.reactions import reactions_follow, reactions_unfollow
|
|
from services.zine.gittask import GitTask
|
|
# from resolvers.inbox.chats import create_chat
|
|
# from services.inbox.storage import MessagesStorage
|
|
|
|
|
|
@mutation.field("createShout")
|
|
@login_required
|
|
async def create_shout(_, info, inp):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
with local_session() as session:
|
|
topics = session.query(Topic).filter(Topic.slug.in_(inp.get('topics', []))).all()
|
|
|
|
new_shout = Shout.create(**{
|
|
"title": inp.get("title"),
|
|
"subtitle": inp.get('subtitle'),
|
|
"body": inp.get("body", ''),
|
|
"authors": inp.get("authors", []),
|
|
"slug": inp.get("slug"),
|
|
"mainTopic": inp.get("mainTopic"),
|
|
"visibility": "owner",
|
|
"createdBy": auth.user_id
|
|
})
|
|
|
|
for topic in topics:
|
|
t = ShoutTopic.create(topic=topic.id, shout=new_shout.id)
|
|
session.add(t)
|
|
|
|
# if auth.user_id in authors:
|
|
# authors.remove(auth.user_id)
|
|
# Chat room code, uncomment it
|
|
# if authors:
|
|
# chat = create_chat(None, info, new_shout.title, members=authors)
|
|
# # create a cooperative chatroom
|
|
# await MessagesStorage.register_chat(chat)
|
|
# # now we should create a collab
|
|
# new_collab = Collab.create({
|
|
# "shout": new_shout.id,
|
|
# "authors": [auth.user_id, ],
|
|
# "invites": authors
|
|
# })
|
|
# session.add(new_collab)
|
|
|
|
# NOTE: shout made by one first author
|
|
sa = ShoutAuthor.create(shout=new_shout.id, user=auth.user_id)
|
|
session.add(sa)
|
|
|
|
# if "mainTopic" in inp:
|
|
# new_shout.topics.append(inp["mainTopic"])
|
|
|
|
session.add(new_shout)
|
|
|
|
reactions_follow(auth.user_id, new_shout.id, True)
|
|
|
|
# for slug in new_shout.topics:
|
|
# topic = session.query(Topic).where(Topic.slug == slug).one()
|
|
#
|
|
# st = ShoutTopic.create(shout=new_shout.id, topic=topic.id)
|
|
# session.add(st)
|
|
#
|
|
# tf = session.query(TopicFollower).where(
|
|
# and_(TopicFollower.follower == auth.user_id, TopicFollower.topic == topic.id)
|
|
# )
|
|
#
|
|
# if not tf:
|
|
# tf = TopicFollower.create(follower=auth.user_id, topic=topic.id, auto=True)
|
|
# session.add(tf)
|
|
|
|
session.commit()
|
|
|
|
# TODO
|
|
# GitTask(inp, user.username, user.email, "new shout %s" % new_shout.slug)
|
|
|
|
if new_shout.slug is None:
|
|
new_shout.slug = f"draft-{new_shout.id}"
|
|
session.commit()
|
|
|
|
return {"shout": new_shout}
|
|
|
|
|
|
@mutation.field("updateShout")
|
|
@login_required
|
|
async def update_shout(_, info, shout_id, shout_input=None, publish=False):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
with local_session() as session:
|
|
shout = session.query(Shout).options(
|
|
joinedload(Shout.authors),
|
|
joinedload(Shout.topics),
|
|
).filter(Shout.id == shout_id).first()
|
|
|
|
if not shout:
|
|
return {"error": "shout not found"}
|
|
|
|
if shout.createdBy != auth.user_id:
|
|
return {"error": "access denied"}
|
|
|
|
updated = False
|
|
|
|
if shout_input is not None:
|
|
topics_input = shout_input["topics"]
|
|
del shout_input["topics"]
|
|
|
|
new_topics_to_link = []
|
|
new_topics = [topic_input for topic_input in topics_input if topic_input["id"] < 0]
|
|
|
|
for new_topic in new_topics:
|
|
del new_topic["id"]
|
|
created_new_topic = Topic.create(**new_topic)
|
|
session.add(created_new_topic)
|
|
new_topics_to_link.append(created_new_topic)
|
|
|
|
if len(new_topics) > 0:
|
|
session.commit()
|
|
|
|
for new_topic_to_link in new_topics_to_link:
|
|
created_unlinked_topic = ShoutTopic.create(shout=shout.id, topic=new_topic_to_link.id)
|
|
session.add(created_unlinked_topic)
|
|
|
|
existing_topics_input = [topic_input for topic_input in topics_input if topic_input.get("id", 0) > 0]
|
|
existing_topic_to_link_ids = [existing_topic_input["id"] for existing_topic_input in existing_topics_input
|
|
if existing_topic_input["id"] not in [topic.id for topic in shout.topics]]
|
|
|
|
for existing_topic_to_link_id in existing_topic_to_link_ids:
|
|
created_unlinked_topic = ShoutTopic.create(shout=shout.id, topic=existing_topic_to_link_id)
|
|
session.add(created_unlinked_topic)
|
|
|
|
topic_to_unlink_ids = [topic.id for topic in shout.topics
|
|
if topic.id not in [topic_input["id"] for topic_input in existing_topics_input]]
|
|
|
|
shout_topics_to_remove = session.query(ShoutTopic).filter(
|
|
and_(
|
|
ShoutTopic.shout == shout.id,
|
|
ShoutTopic.topic.in_(topic_to_unlink_ids)
|
|
)
|
|
)
|
|
|
|
for shout_topic_to_remove in shout_topics_to_remove:
|
|
session.delete(shout_topic_to_remove)
|
|
|
|
shout_input["mainTopic"] = shout_input["mainTopic"]["slug"]
|
|
|
|
if shout_input["mainTopic"] == '':
|
|
del shout_input["mainTopic"]
|
|
|
|
shout.update(shout_input)
|
|
updated = True
|
|
|
|
if publish and shout.visibility == 'owner':
|
|
shout.visibility = "community"
|
|
shout.publishedAt = datetime.now(tz=timezone.utc)
|
|
updated = True
|
|
|
|
if updated:
|
|
shout.updatedAt = datetime.now(tz=timezone.utc)
|
|
|
|
session.commit()
|
|
# GitTask(inp, user.username, user.email, "update shout %s" % slug)
|
|
|
|
return {"shout": shout}
|
|
|
|
|
|
@mutation.field("deleteShout")
|
|
@login_required
|
|
async def delete_shout(_, info, shout_id):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
with local_session() as session:
|
|
shout = session.query(Shout).filter(Shout.id == shout_id).first()
|
|
|
|
if not shout:
|
|
return {"error": "invalid shout id"}
|
|
|
|
if auth.user_id != shout.createdBy:
|
|
return {"error": "access denied"}
|
|
|
|
for author_id in shout.authors:
|
|
reactions_unfollow(author_id, shout_id)
|
|
|
|
shout.deletedAt = datetime.now(tz=timezone.utc)
|
|
session.commit()
|
|
|
|
return {}
|