aboutsummaryrefslogtreecommitdiff
path: root/bin/maybe
blob: d017966a9754dad2c6059afa0a348c85714b0a24 (plain) (blame)
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
68
69
#!/usr/bin/env bash
# Exit with success or failure with a given probability
self=maybe

# Print usage information
usage() {
    printf '%s: usage: %s [-hv] [-d DENOMINATOR]\n' \
        "$self" "$self"
}

# Flag for whether to print diagnostics to stdout or not
declare -i verbose
verbose=0

# Denominator of the probability fraction, e.g. 3 is a probability of 1/3;
# defaults to 2
declare -i denom
denom=2

# Process options
while getopts 'hvd:' opt ; do
    case $opt in

        # -h: Print help
        h)
            usage
            exit 0
            ;;

        # -v: Print diagnostics to stdout
        v)
            verbose=1
            ;;

        # -d: Set the denominator of the probability fraction
        d)
            denom=$OPTARG
            ;;

        # Unknown option
        \?)
            usage >&2
            exit 2
            ;;
    esac
done
shift "$((OPTIND-1))"

# If there are any non-option arguments or our denominator isn't a positive
# integer, we're being abused and we won't put up with it
if (($#)) || ! ((denom > 0)) ; then
    usage >&2
    exit 2
fi

# If verbose, report the probability of the test
((verbose)) && printf '%s: Testing with probability 1/%u ... \n' \
    "$self" "$denom"

# Perform the test and print/exit appropriately
if ((RANDOM < 32767/denom)) ; then
    ((verbose)) && printf '%s: %s\n' \
        "$self" 'Success!'
    exit 0
else
    ((verbose)) && printf '%s: %s\n' \
        "$self" 'Failure!'
    exit 1
fi