rscli/rscli/cli.py

#!/usr/bin/env python3
import os
import os.path
import sys
from pprint import pprint
import requests
import argparse

from rscli import rsfiles
from rscli.httputils import req

try:
    from colors import color

    has_colors = True
except ImportError:
    has_colors = False

if not has_colors or not sys.stdout.isatty():

    def color(text, *args, **kwargs):
        return text


def err(msg):
    print(color(msg, fg="red", style="bold"))


def is_group_subscribed(mSubscribeFlags):
    return bool(mSubscribeFlags & 4)


def main_forum_list(args):
    r = req(args, "/rsGxsForums/getForumsSummaries")
    forums = r.json()["forums"]
    for item in forums:
        if is_group_subscribed(item["mSubscribeFlags"]):
            if "notsubscribed" in args.select:
                continue
            style = "bold"
        else:
            if "subscribed" in args.select:
                continue
            style = None
        print(color(item["mGroupName"], style=style, fg="green"))
        print("   " + color(item["mGroupId"], style="underline"))


def main_forum_read(args):
    r = req(args, "/rsGxsForums/getForumMsgMetaData", {"forumId": args.forum_id})
    items = r.json()["msgMetas"]
    items.sort(key=lambda p: p["mPublishTs"], reverse=True)
    if args.long:
        msgs = [item["mMsgId"] for item in items[: args.num_posts]]
        items_r = req(
            args,
            "/rsGxsForums/getForumContent",
            {"forumId": args.forum_id, "msgsIds": msgs},
        )
        items = items_r.json()["msgs"]
        items.sort(key=lambda p: p["mMeta"]["mPublishTs"], reverse=True)
        for item in items:
            print(color(item["mMeta"]["mMsgName"], style="bold", fg="green"))
            print()
            print(item["mMsg"])  # TODO: html2txt
            print()
    else:
        for item in items[: args.num_posts]:
            print(color(item["mMsgName"], style="bold", fg="green"))
            print("   " + color(item["mMsgId"], style="underline"))


def main_channel_list(args):
    r = req(args, "/rsGxsChannels/getChannelsSummaries")
    channels = r.json()["channels"]
    for item in channels:
        if is_group_subscribed(item["mSubscribeFlags"]):
            if "notsubscribed" in args.select:
                continue
            style = "bold"
        else:
            if "subscribed" in args.select:
                continue
            style = None
        print(color(item["mGroupName"], style=style, fg="green"))
        print("   " + color(item["mGroupId"], style="underline"))


def main_channel_read(args):
    r = req(args, "/rsGxsChannels/getContentSummaries", {"channelId": args.channel_id})
    posts = r.json()["summaries"]
    posts.sort(key=lambda p: p["mPublishTs"], reverse=True)
    if args.long:
        msgs = [post["mMsgId"] for post in posts[: args.num_posts]]
        posts_r = req(
            args,
            "/rsGxsChannels/getChannelContent",
            {"channelId": args.channel_id, "contentsIds": msgs},
        )
        posts = posts_r.json()["posts"]
        posts.sort(key=lambda p: p["mMeta"]["mPublishTs"], reverse=True)
        for post in posts:
            print(color(post["mMeta"]["mMsgName"], style="bold", fg="green"))
            print()
            print(post["mMsg"])
            print()
    else:
        for post in posts[: args.num_posts]:
            print(color(post["mMsgName"], style="bold", fg="green"))
            print("   " + color(post["mMsgId"], style="underline"))


def main_channel_show(args):
    r = req(args, "/rsGxsChannels/getChannelsInfo", {"chanIds": [args.channel_id]})
    data = r.json()
    channels = data["channelsInfo"]
    for chan in channels:
        print(color(chan["mMeta"]["mGroupName"], style="bold", fg="green"))
        print("   " + color(chan["mMeta"]["mGroupId"], style="underline"))
        print("   " + chan["mDescription"])
        print(color("Last Post:", style="bold") + " \t%s" % chan["mMeta"]["mLastPost"])


def main_channel_post_v1(args):
    r = req(
        args,
        "/rsGxsChannels/createPost",
        {
            "post": {
                "mMeta": {"mGroupId": args.channel_id, "mMsgName": args.post_title},
                "mMsg": args.post_body,
            }
        },
    )
    if r.status_code != 200:
        print(color("ERROR: could not create post", fg="red", style="bold"))
        print("Error %d" % r.status_code)
        sys.exit(1)
    ret = r.json()
    if ret.get("retval", True) is False:
        print(color("ERROR: could not create post", fg="red", style="bold"))
        pprint(ret)
        sys.exit(1)
    pprint(ret)


def main_channel_post(args):
    attachments = [
        {
            "mHash": digest,
            "mName": os.path.basename(fpath),
            "mSize": os.stat(fpath).st_size,
        }
        for digest, fpath in zip(
            rsfiles.file_publish(args, args.attach_fpaths), args.attach_fpaths
        )
    ]
    try:
        r = req(
            args,
            "/rsGxsChannels/createPostV2",
            {
                "channelId": args.channel_id,
                "title": args.post_title,
                "mBody": args.post_body,
                "files": attachments,
            },
        )
    except requests.exceptions.HTTPError as exc:
        if exc.response.status_code != 404:
            raise
    else:
        ret = r.json()
        if ret.get("retval", True) is False:
            print(color("ERROR: could not create post", fg="red", style="bold"))
            print(ret["errorMessage"])
            pprint(ret)
            sys.exit(1)
        print(color(ret["postId"], fg="green", style="bold"))
        return
    return main_channel_post_v1(args)


def main_file_publish(args):
    ret = rsfiles.file_publish(args, args.fnames)
    for filehash, fname in zip(ret, args.fnames):
        print(color(filehash, fg="green") + " \t%s" % fname)
        print("    " + rsfiles.get_file_link(args, filehash, fname=fname))


def get_parser():
    p = argparse.ArgumentParser()
    p.add_argument("--endpoint", default="http://127.0.0.1:9092")
    p.add_argument("-u", "--auth", dest="auth")

    p_sub = p.add_subparsers()

    ch = p_sub.add_parser("channel")
    ch.add_argument("--channel-id")
    ch_sub = ch.add_subparsers()
    ch_list = ch_sub.add_parser("list")
    ch_list.add_argument(
        "--select",
        nargs="+",
        choices=["all", "subscribed", "notsubscribed"],
        default=["all"],
    )
    ch_list.set_defaults(mainfunc=main_channel_list)

    ch_show = ch_sub.add_parser("show")
    ch_show.set_defaults(mainfunc=main_channel_show)

    ch_read = ch_sub.add_parser("read")
    ch_read.add_argument("--long", action="store_true", default=False)
    ch_read.add_argument("--num-posts", type=int, default=10)
    ch_read.set_defaults(mainfunc=main_channel_read)

    ch_post = ch_sub.add_parser("post")
    ch_post.set_defaults(mainfunc=main_channel_post)
    ch_post.add_argument("--post-title")
    ch_post.add_argument("--post-body")
    ch_post.add_argument("--attach-name", nargs="*", dest="attach_fpaths")

    forum = p_sub.add_parser("forum")
    forum.add_argument("--forum-id")
    forum_sub = forum.add_subparsers()
    forum_list = forum_sub.add_parser("list")
    forum_list.add_argument(
        "--select",
        nargs="+",
        choices=["all", "subscribed", "notsubscribed"],
        default=["all"],
    )
    forum_list.set_defaults(mainfunc=main_forum_list)

    forum_read = forum_sub.add_parser("read")
    forum_read.add_argument("--long", action="store_true", default=False)
    forum_read.add_argument("--num-posts", type=int, default=10)
    forum_read.set_defaults(mainfunc=main_forum_read)

    files = p_sub.add_parser("file")
    files_sub = files.add_subparsers()
    files_list = files_sub.add_parser("publish")
    files_list.add_argument("fnames", nargs="+", metavar="fname")
    files_list.set_defaults(mainfunc=main_file_publish)

    # TODO: channel rss -> read and convert to rss

    return p


def main():
    p = get_parser()
    args = p.parse_args()
    if getattr(args, "mainfunc", None) is None:
        print("Not a complete command")
        sys.exit(2)
    args.mainfunc(args)


if __name__ == "__main__":
    main()