feat: add baron users registrator (#25)

README.md

@@ -5,6 +5,7 @@
 
 | Tool | Description | Ejudge | Polygon | Status |
 | --- | --- | :---: | :---: | :---: |
+| [baron](#baron) | register users to contest | 🦍 | | ✅ |
 | [blanka](#blanka) | create contest | 🦍 | | ✅ |
 | [boban](#boban) | filter runs | 🦍 | | ✅ |
 | [casper](#casper) | change visibility | 🦍 | | ✅ |
@@ -61,6 +62,32 @@ Put your config file in `~/.config/algolymp/config.json`.
 }
 ```
 
+## baron
+*Register users to Ejudge contest (Pending status).*
+
+### About
+
+Read user logins from `stdin` and register them to Ejudge contest.
+
+Don't forget to set `OK` status manually!
+
+### Flags
+- `-i` - contest id (required)
+
+### Config
+- `ejudge.url`
+- `ejudge.login`
+- `ejudge.password`
+
+### Examples
+```bash
+baron --help
+baron -i 48501 # read from stdin
+cat users.csv | baron -i 48600 # read from file
+```
+
+![baron logo](https://algolymp.ru/static/img/baron.png)
+
 ## blanka
 *Create Ejuge contest from template.*
 

cmd/baron/main.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"os"
+
+	"github.com/Gornak40/algolymp/config"
+	"github.com/Gornak40/algolymp/ejudge"
+	"github.com/akamensky/argparse"
+	"github.com/sirupsen/logrus"
+)
+
+func main() {
+	parser := argparse.NewParser("baron", "Register users to Ejudge contest (Pending status).")
+	cID := parser.Int("i", "cid", &argparse.Options{
+		Required: true,
+		Help:     "Ejudge contest ID",
+	})
+	if err := parser.Parse(os.Args); err != nil {
+		logrus.WithError(err).Fatal("bad arguments")
+	}
+
+	cfg := config.NewConfig()
+	ejClient := ejudge.NewEjudge(&cfg.Ejudge)
+
+	sid, err := ejClient.Login()
+	if err != nil {
+		logrus.WithError(err).Fatal("login failed")
+	}
+
+	csid, err := ejClient.MasterLogin(sid, *cID)
+	if err != nil {
+		logrus.WithError(err).Fatal("master login failed")
+	}
+
+	logrus.Info("waiting for logins input...")
+	for {
+		var login string
+		_, err := fmt.Scan(&login)
+		if errors.Is(err, io.EOF) {
+			break
+		}
+		if err != nil {
+			logrus.WithError(err).Fatal("invalid login")
+		}
+		if err := ejClient.RegisterUser(csid, login); err != nil {
+			logrus.WithError(err).Error("register user failed")
+		}
+	}
+
+	if err := ejClient.Logout(sid); err != nil {
+		logrus.WithError(err).Fatal("logout failed")
+	}
+}

ejudge/ejudge.go

@@ -289,3 +289,14 @@ func (ej *Ejudge) MakeVisible(sid string, cid int) error {
 
 	return err
 }
+
+func (ej *Ejudge) RegisterUser(csid, login string) error {
+	logrus.WithFields(logrus.Fields{"CSID": csid, "login": login}).Info("register user")
+	_, _, err := ej.postRequest("new-master", url.Values{
+		"SID":       {csid},
+		"action":    {"20"},
+		"add_login": {login},
+	})
+
+	return err
+}