summaryrefslogtreecommitdiff
path: root/doc/progs/cat_rot13.go
blob: 27d1e467fe3f8c30533e00d02e9cc88ea3e4ddff (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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"file";
	"flag";
	"os";
)

var rot13_flag = flag.Bool("rot13", false, "rot13 the input")

func rot13(b byte) byte {
	if 'a' <= b && b <= 'z' {
	   b = 'a' + ((b - 'a') + 13) % 26;
	}
	if 'A' <= b && b <= 'Z' {
	   b = 'A' + ((b - 'A') + 13) % 26
	}
	return b
}

type reader interface {
	Read(b []byte) (ret int, err *os.Error);
	String() string;
}

type rotate13 struct {
	source	reader;
}

func newRotate13(source reader) *rotate13 {
	return &rotate13{source}
}

func (r13 *rotate13) Read(b []byte) (ret int, err *os.Error) {
	r, e := r13.source.Read(b);
	for i := 0; i < r; i++ {
		b[i] = rot13(b[i])
	}
	return r, e
}

func (r13 *rotate13) String() string {
	return r13.source.String()
}
// end of rotate13 implementation

func cat(r reader) {
	const NBUF = 512;
	var buf [NBUF]byte;

	if *rot13_flag {
		r = newRotate13(r)
	}
	for {
		switch nr, er := r.Read(buf); {
		case nr < 0:
			print("error reading from ", r.String(), ": ", er.String(), "\n");
			sys.Exit(1);
		case nr == 0:  // EOF
			return;
		case nr > 0:
			nw, ew := file.Stdout.Write(buf[0:nr]);
			if nw != nr {
				print("error writing from ", r.String(), ": ", ew.String(), "\n");
			}
		}
	}
}

func main() {
	flag.Parse();   // Scans the arg list and sets up flags
	if flag.NArg() == 0 {
		cat(file.Stdin);
	}
	for i := 0; i < flag.NArg(); i++ {
		f, err := file.Open(flag.Arg(i), 0, 0);
		if f == nil {
			print("can't open ", flag.Arg(i), ": error ", err, "\n");
			sys.Exit(1);
		}
		cat(f);
		f.Close();
	}
}