Changeset 8bf8a9b3ecff…
Parent 08f06e94592b…
by Benjamin Pollack
Changes to one file · Browse files at 8bf8a9b3ecff Showing diff from parent 08f06e94592b Diff from another changeset...
@@ -7,6 +7,8 @@ "io"
"math/rand"
"os"
+ "os/exec"
+ "path/filepath"
"time"
)
@@ -45,16 +47,28 @@ return
}
+func executablePath() string {
+ path, err := exec.LookPath(os.Args[0])
+ if err != nil {
+ panic(err)
+ }
+ path, err = filepath.Abs(path)
+ if err != nil {
+ panic(err)
+ }
+ return filepath.Dir(path)
+}
+
func main() {
offensive := flag.Bool("o", false, "be offensive")
flag.Parse()
- fortunes, err := readFortunes(NORMAL)
+ fortunes, err := readFortunes(filepath.Join(executablePath(), NORMAL))
if err != nil {
fmt.Fprintf(os.Stderr, "unable to load normal fortunes: %v\n", err)
os.Exit(1)
}
if *offensive {
- offensiveFortunes, err := readFortunes(OFFENSIVE)
+ offensiveFortunes, err := readFortunes(filepath.Join(executablePath(), OFFENSIVE))
if err != nil {
fmt.Fprintf(os.Stderr, "unable to load offensive fortunes: %v\n", err)
os.Exit(1)
|
Loading...