# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
# license agreements.  See the NOTICE file distributed with this work for
# additional information regarding copyright ownership.  Crate licenses
# this file to you under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.  You may
# obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
# License for the specific language governing permissions and limitations
# under the License.
#
# However, if you have executed another commercial license agreement
# with Crate these terms will supersede the license and you may use the
# software solely pursuant to the terms of the relevant commercial agreement.

from argparse import Namespace

from croud.config import Configuration
from croud.printer import print_error, print_format
from croud.util import gql_mutation


def roles_add(args: Namespace) -> None:
    """
    Adds a new role to a user
    """

    mutation = """
mutation {
    addRoleToUser(input: {userId: "uid", roleFqn: "rfqn", resourceId: "resid"}) {
        user {
            uid,
            email,
            username,
            organizationId
        }
    }
}
"""

    mutation = mutation.replace("uid", args.user, 1)
    mutation = mutation.replace("rfqn", args.role)
    mutation = mutation.replace("resid", args.resource)

    data = gql_mutation(mutation, args, "addRoleToUser")
    if "errors" in data:
        print_error(data["errors"][0]["message"])
    else:
        fmt = args.output_fmt or Configuration.get_setting("output_fmt")
        print_format(data, fmt)
