import csv
from io import StringIO
import requests

BASE_URL = "https://api.membershipworks.com/v1/"

class MembershipWorksRemoteError(Exception):
    def __init__(self, reason, r):
        super().__init__(
            f"Error when attempting {reason}: {r.status_code} {r.reason}\n{r.text}")

class MembershipWorks:
    def __init__(self):
        self.auth_token = None
        self.org_num = None

    def login(self, username, password):
        """Authenticate against the membershipworks api"""
        r = requests.post(BASE_URL + 'usr',
                          data={"_st": "all",
                                "eml": username,
                                "org": "10000",
                                "pwd": password})
        if r.status_code != 200 or 'SF' not in r.json():
            raise MembershipWorksRemoteError('login', r)
        self.auth_token = r.json()['SF']
        self.org_num = r.json()['org']

    def _inject_auth(self, kwargs):
        # TODO: should probably be a decorator or something
        if self.auth_token is None:
            raise RuntimeError('Not Logged in to MembershipWorks')
        # add auth token to params
        if 'params' not in kwargs:
            kwargs['params'] = {}
        kwargs['params']["SF"] = self.auth_token

    def _get(self, *args, **kwargs):
        self._inject_auth(kwargs)
        # TODO: should probably do some error handling in here
        return requests.get(*args, **kwargs)

    def _post(self, *args, **kwargs):
        self._inject_auth(kwargs)
        # TODO: should probably do some error handling in here
        return requests.post(*args, **kwargs)

    def get_member_ids(self, folders):
        # TODO: this is hardcoded for CMS
        folder_map = {
            'members': '5ae37979f033bfe8534f8799',
            'staff': '5771675edcdf126302a2f6b9',
            'misc': '5b69ee9bf033bf8e7346c434'
        }

        r = self._get(BASE_URL + "ylp", params={
            "lbl": ",".join([folder_map[f] for f in folders]),
            "org": self.org_num,
            "var": "_id,nam,ctc"})
        if r.status_code != 200 or 'usr' not in r.json():
            raise MembershipWorksRemoteError('user listing', r)

        # get list of member/staff IDs
        return [user['uid'] for user in r.json()['usr']]

    def get_members(self, folders, columns):
        """ Pull the members csv from the membershipworks api
            folders: a list of the names of the folders to get
                     (see folder_map in this function for mapping to ids)
            columns: which columns to get"""
        ids = self.get_member_ids(folders)

        # get members CSV
        # TODO: maybe can just use previous get instead? would return JSON
        r = self._post(BASE_URL + "csv",
                       data={"_rt": "946702800", # unknown
                             "mux": "", # unknown
                             "tid": ",".join(ids), # ids of members to get
                             "var": columns})
        if r.status_code != 200:
            raise MembershipWorksRemoteError('csv generation', r)
        return list(csv.DictReader(StringIO(r.text)))