#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
这个程序用来向系统中添加帐号。这些帐号会在 bitoj 中使用，用户提交的程序会以
不同的用户帐号执行。

"""

import os, sys


rundir = '/var/lib/bitoj'

def usage(message = None):
    if message:
        print(message)
    print("Usage: bitoj_adduser minnum maxnum")


def adduser(usernum):
    global rundir

    username = 'ojrun%02d' % usernum
    homedir = os.path.join(rundir, username)
    cmd = 'adduser --system --group --disabled-login --home %s %s' \
               % (homedir, username)
    if os.system(cmd) == 0:
        os.system('adduser oj %s' % username)
        os.system('adduser %s oj' % username)
        os.system('chmod g+w %s' % homedir)
        os.system('setquota -u %s 25600 25600 1000 1000 /' % username)

def main():
    if len(sys.argv) < 3:
        usage()
        sys.exit(1)

    minnum = int(sys.argv[1])
    maxnum = int(sys.argv[2])


    if minnum > maxnum:
        usage("minnum should be small than maxnum")
        sys.exit(1)

    if minnum < 1 or maxnum > 100:
        usage("minnum should between 1 and 100")
        sys.exit(1)

    if maxnum < 1 or maxnum > 100:
        usage("maxnum should between 1 and 100")
        sys.exit(1)

    for i in range(minnum, maxnum + 1):
        adduser(i)

main()