Extract auth code to own file, add some tests

toot/auth.py

@@ -0,0 +1,101 @@
+# -*- coding: utf-8 -*-
+
+import webbrowser
+
+from builtins import input
+from getpass import getpass
+
+from toot import api, config, DEFAULT_INSTANCE, User, App, ConsoleError
+from toot.output import print_out
+
+
+def register_app(instance):
+    print_out("Registering application with <green>{}</green>".format(instance))
+
+    try:
+        response = api.create_app(instance)
+    except Exception:
+        raise ConsoleError("Registration failed. Did you enter a valid instance?")
+
+    base_url = 'https://' + instance
+
+    app = App(instance, base_url, response['client_id'], response['client_secret'])
+    path = config.save_app(app)
+    print_out("Application tokens saved to: <green>{}</green>\n".format(path))
+
+    return app
+
+
+def create_app_interactive(instance=None):
+    if not instance:
+        print_out("Choose an instance [<green>{}</green>]: ".format(DEFAULT_INSTANCE), end="")
+        instance = input()
+        if not instance:
+            instance = DEFAULT_INSTANCE
+
+    return config.load_app(instance) or register_app(instance)
+
+
+def create_user(app, email, access_token):
+    user = User(app.instance, email, access_token)
+    path = config.save_user(user)
+
+    print_out("Access token saved to: <green>{}</green>".format(path))
+
+    return user
+
+
+def login_interactive(app, email=None):
+    print_out("Log in to <green>{}</green>".format(app.instance))
+
+    if email:
+        print_out("Email: <green>{}</green>".format(email))
+
+    while not email:
+        email = input('Email: ')
+
+    password = getpass('Password: ')
+
+    try:
+        print_out("Authenticating...")
+        response = api.login(app, email, password)
+    except api.ApiError:
+        raise ConsoleError("Login failed")
+
+    return create_user(app, email, response['access_token'])
+
+
+BROWSER_LOGIN_EXPLANATION = """
+This authentication method requires you to log into your Mastodon instance
+in your browser, where you will be asked to authorize <yellow>toot</yellow> to access
+your account. When you do, you will be given an <yellow>authorization code</yellow>
+which you need to paste here.
+"""
+
+
+def login_browser_interactive(app):
+    url = api.get_browser_login_url(app)
+
+    print_out(BROWSER_LOGIN_EXPLANATION)
+
+    print_out("This is the login URL:")
+    print_out(url)
+    print_out("")
+
+    yesno = input("Open link in default browser? [Y/n]")
+    if not yesno or yesno.lower() == 'y':
+        webbrowser.open(url)
+
+    authorization_code = ""
+    while not authorization_code:
+        authorization_code = input("Authorization code: ")
+
+    print_out("\nRequesting access token...")
+    response = api.request_access_token(app, authorization_code)
+
+    # TODO: user email is not available in this workflow, maybe change the User
+    # to store the username instead? Currently set to "unknown" since it's not
+    # used anywhere.
+    email = "unknown"
+
+    return create_user(app, email, response['access_token'])