はじめに

RustでAtCoderにログイン認証を通すコードを作成したので備忘録として残しておきます。

処理の流れ

https://atcoder.jp/login にアクセスしてCookieとcsrf_tokenを取得
csrf_token、ユーザ名、パスワードをクエリパラメータに設定してログイン

コード

reqwestのfeaturesに"blocking"、"cookies"を追加する必要があります。

use scraper::Selector;

use reqwest::{blocking::Client, cookie::Jar};
use std::{io, sync::Arc};

fn get_user_id() -> String {
    let mut input = String::new();
    print!("ユーザ名を入力してください: ");
    io::stdout().flush().unwrap();
    io::stdin()
        .read_line(&mut input)
        .expect("Failed to read line");
    input.trim().to_string()
}

fn login() -> Result<(), reqwest::Error> {
    let user = get_user_id();
    let pass = rpassword::prompt_password("パスワードを入力してください: ")
        .expect("Failed to read password");

    // Cookieストアを作成
    let cookie_store = Arc::new(Jar::default());

    // Cookieストアを持つClientを作成
    let client = Client::builder()
        .cookie_store(true) // クライアントでCookieストアを有効にする
        .cookie_provider(Arc::clone(&cookie_store)) // カスタムCookieストアを設定
        .build()?;

    // csrf_tokenを取得
    let selector = Selector::parse("input[name=\"csrf_token\"]").unwrap();
    let body = client.get("https://atcoder.jp/login").send()?.text()?;
    let document = scraper::Html::parse_document(&body);
    let csrf_token = document
        .select(&selector)
        .next()
        .unwrap()
        .value()
        .attr("value")
        .unwrap()
        .to_string();

    let params = [
        ("username", &user),
        ("password", &pass),
        ("csrf_token", &csrf_token),
    ];

    let response = client
        .post("https://atcoder.jp/login")
        .query(&params)
        .send()?;

    response.error_for_status()?;

    println!("ログイン成功");

    Ok(())
}

fn main() -> Result<(), Box<dyn std::error::Error>> {
    login()?;

    Ok(())
}

まとめ

cookieの保存のあたりで苦戦しましたがRustフォーラムの記事に助けられました。

参考文献