server/app/lib/tokens.py

#!/usr/bin/env python3
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF 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.
"""SelfServe Platform for the Apache Software Foundation"""

"""Personal Access Token handler for selfserve"""

import asfquart
import os

roleaccounts = {}
ROLEACCOUNT_FILE = os.path.normpath(os.path.join("..", "roleaccounts.txt"))

if os.path.exists(ROLEACCOUNT_FILE):
    print("Parsing role accounts list")
    for line in open(ROLEACCOUNT_FILE).readlines():
        if not line.startswith("#"):
            k, v = line.split(":", 1)
            k = k.strip().lower()
            print(f"Found role account: {k}")
            roleaccounts[v.strip()] = k


async def token_handler(token):
    if token in roleaccounts:
        return {
            "uid": roleaccounts[token],
            "fullname": f"{roleaccounts[token]} role Account",
            "roleaccount": True,
        }

asfquart.APP.token_handler = token_handler