args.go


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

package main

import (
	"errors"
	"fmt"
	"os"
	"strconv"
)

// The arguments to this program
type args struct {
	Radix uint
}

func parseArgs() (args, error) {
	if len(os.Args) == 2 {
		if radix, err := strconv.ParseUint(os.Args[1], 0, 0); err == nil {
			if radix > 1 {
				return args{uint(radix)}, nil
			} else {
				return args{}, errors.New("Radix must be an integer above 1.")
			}
		} else {
			return args{}, fmt.Errorf(
				"Argument must be an integer above 1 [%w].", err)
		}
	} else if len(os.Args) < 2 {
		return args{}, errors.New("Please provide an argument (radix to study).")
	} else {
		return args{}, errors.New("Too many arguments provided.")
	}
}