aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib.rs
blob: c0f537de402a437dd5a4af9ed680f5d457304123 (plain)
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#[macro_use]
extern crate glium; 

extern crate image;
extern crate getopts;

use std::env;
use std::u64;
use std::process::exit;
use getopts::{Options, Matches};
use glium::glutin::os::unix::WindowBuilderExt;
    
pub mod util;

fn print_usage(program: &str, opts: Options) {
    let brief = format!("Usage: {} [options]", program);
    print!("{}", opts.usage(&brief));
}

pub fn generic_main(mut opts: Options) -> (Matches, Option<u64>) {
    let args: Vec<String> = env::args().collect();
    let program = args[0].clone();

    // Turn, eg, "-root" into "--root"
    let args = util::convert_xscreensaver_args(args);

    // Common Args (all screensavers)
    opts.optflag("h", "help", "print this help menu");
    opts.optflag("", "window", "run in a window (IGNORED)");
    opts.optflag("", "root", "run in root window (IGNORED)");
    opts.optflag("", "fps", "show frames per second (IGNORED)");
    opts.optopt("", "window-id", "X window id number", "NUM");

    let matches = match opts.parse(&args[1..]) {
        Ok(m) => { m }
        Err(f) => {
            print_usage(&program, opts);
            println!("");
            println!("{}", f.to_string());
            exit(-1);
        }
    };

    if matches.opt_present("help") {
        print_usage(&program, opts);
        exit(0);
    }

    // if no "--window-id", try environment variable (arg has priority though)
    let window_id_string: Option<String> =
        matches.opt_str("window-id")
               .or(env::var("XSCREENSAVER_WINDOW").ok());

    let window_id = window_id_string.map(|id| match util::dechex2u64(&id) {
        Ok(y) => y,
        Err(e) => {
            println!("Couldn't parse numerical argument: {}", e);
            exit(-1); },
    });

    if window_id.is_some() {
        println!("Drawing on existing X window: 0x{:07X}", window_id.unwrap());
    }

    (matches, window_id)
}