# Copyright (C) 2021 by David Bremner <david@tethera.net>
# Licensed under the LGPL v3 https://www.gnu.org/licenses/lgpl-3.0.html

import os
import cmd
import configparser
from pathlib import PurePath
from mailmanclient.client import Client

class ModShell(cmd.Cmd):
    intro = 'Type help or ? for a list of commands'

    def __init__(self, client, mlist):
        self.client = client;
        self.mlist = mlist;
        self.prompt = '({:s}) '.format(mlist.list_name)
        cmd.Cmd.__init__(self)

    def emptyline(self):
        pass

    def print_held(self):
        self.held_messages = self.mlist.get_held_page (50,1)
        print ()
        for msg in self.held_messages:
            print(msg.request_id, msg.subject, msg.sender)
        print ()

    def matching_request_ids(self,text):
        candidates = []
        prefix_len = len(text)

        for msg in self.held_messages:
            string =  str(msg.request_id)
            if text == string[0:prefix_len]:
                candidates.append(string)
        return candidates

    def complete_accept(self, text, line, begidx, endidx):
        return self.matching_request_ids(text)

    def do_accept(self, arg):
        '''
        accept <number>

        Forward held message(s) to mailing list.
        '''
        try:
            n = int(arg)
            self.mlist.get_held_message (n).accept()
        except ValueError:
            print('Illegal request_id: {:s}'.format(arg))

        return False

    def complete_discard(self, text, line, begidx, endidx):
        candidates = self.matching_request_ids(text)
        if text == "page"[0:prefix_len]:
            candidates.append('page')
        return candidates

    def do_discard(self, arg):
        '''
        discard <number> | page

        Discard held message(s) as spam.
        '''
        try:
            if arg == 'page':
                for msg in self.held_messages:
                    msg.discard()
            else:
                n = int(arg)
                self.mlist.get_held_message (n).discard()
        except ValueError:
            print('Illegal request_id: {:s}'.format(arg))

        return False

    def preloop(self):
        self.print_held()

    def postcmd(self, stop, line):
        if (not stop):
            self.print_held()
        return stop

    def do_EOF(self,arg):
        'exit'
        return True

    def do_quit(self,arg):
        '''
        exit moderation shell
        '''
        return True

def main ():

    home = os.environ.get('HOME')
    xdg_config_home = os.environ.get('XDG_CONFIG_HOME') or \
        os.path.join(home, '.config')

    config_path = PurePath (xdg_config_home, 'mm3mod', 'config.ini')
    config = configparser.ConfigParser ()

    with open(config_path) as f:
        config.read_file (f)

    baseurl = config.get('rest', 'baseurl', fallback="http://localhost:8001/3.1/")
    user = config.get('rest', 'user', fallback='restadmin')
    password = config.get('rest', 'password')
    client = Client (baseurl, name=user, password=password);
    list_name = config.get('list','name')

    mlist = client.get_list (list_name)
    ModShell(client, mlist).cmdloop()

if __name__ == "__main__":
    main()