aocf 0.1.21

A crate and CLI helper tool for Advent of Code
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

mod db_models;
mod db_schema;

use diesel::prelude::*;
use diesel::sqlite::SqliteConnection;
use failure::{Error, bail};
use std::path::Path;

use db_models::*;
use db_schema::*;

fn connect_sqlite(path: impl AsRef<Path>) -> Result<SqliteConnection, Error> {
    let path = match path.as_ref().to_str() {
        Some(p) => p,
        None => bail!("can't parse path to string"),
    };
    let connection = SqliteConnection::establish(path)?;
    Ok(connection)
}

pub fn get_session_cookie(path: impl AsRef<Path>) -> Result<String, Error> {
    let connection = connect_sqlite(path)?;

    let records = moz_cookies::table
        .filter(moz_cookies::host.eq(".adventofcode.com"))
        .filter(moz_cookies::name.eq("session"))
        .limit(1)
        .load::<FirefoxCookie>(&connection)?;

    if records.is_empty() {
        bail!("no cookie found in the cookie store");
    } else {
        Ok(records[0].value.to_owned())
    }
}