命令行工具 selpg

selpg 允许用户指定从输入文本抽取的页的范围，这些输入文本可以来自文件或另一个进程。详情可见开发 Linux 命令行实用程序。

实现过程

为了解析命令行参数，需要安装 pflag 包：

go get github.com/spf13/pflag

首先使用 pflag 包定义命令行参数：

// flags
startPage := pflag.IntP("start page", "s", 0, "Start page to print")
endPage := pflag.IntP("end page", "e", 0, "End page to print")
pageLen := pflag.IntP("page length", "l", 72, "The number of lines in each page")
pageType := pflag.BoolP("page type", "f", false, "Whether '\\f' is page breaker or not")
printDest := pflag.StringP("print dest", "d", "", "The destination to print")

解析命令行参数并检验它们的合法性：

// parse flags
pflag.Parse()

// check flags
if *startPage < 0 || *endPage < 0 {
	return errors.New("The page number should be positive.")
}
if *startPage > *endPage {
	return errors.New("The start page number should be smaller than the end page number.")
}
if *pageLen != 72 && *pageType == true {
	return errors.New("The \"-l\" flag and the \"-f\" flag shouldn't be used together.")
}
if *pageLen < 0 {
	return errors.New("The page size should be positive.")
}
if pflag.NArg() > 1 {
	return errors.New("Too many arguments.")
}

使用 bufio 包创建读取器：如果命令行中提供了文件名，则在文件中读取；否则在标准输入中读取。

// create reader
var reader *bufio.Reader
if pflag.NArg() == 1 {
	// read from file
	file, err := os.Open(pflag.Arg(0))
	if err != nil {
		return err
	}
	defer file.Close()
	reader = bufio.NewReader(file)
} else {
	// read from stdin
	reader = bufio.NewReader(os.Stdin)
}

如果设定了 -f 标志，则以 -f 为间隔符逐页读取；否则逐行读取，并根据默认的或用户设置的行数来确定每个页的大小。最后根据用户设置的页数范围将所需的内容写入缓冲区。

// write target pages to buffer
var buf strings.Builder
if *pageType == true {
	// read one page each time
	pageCount := 0
	for {
		page, err := reader.ReadBytes('\f')
		if err == io.EOF {
			break
		} else if err != nil {
			return err
		}
		pageCount += 1
		if pageCount >= *startPage && pageCount <= *endPage {
			buf.Write(page)
		} else if pageCount > *endPage {
			break
		}
	}
} else {
	// read one line each time
	pageCount := 0
	lineCount := 0
	for {
		line, err := reader.ReadBytes('\n')
		if err == io.EOF {
			break
		} else if err != nil {
			return err
		}
		lineCount += 1
		if lineCount > *pageLen {
			pageCount += 1
			lineCount = 1
		}
		if pageCount >= *startPage && pageCount <= *endPage {
			buf.Write(line)
		} else if pageCount > *endPage {
			break
		}
	}
}

如果用户没有设置输出位置，则直接将结果输出至标准输出；而如果用户指定了打印机的位置，则调用 lp 命令来将结果输出至目标打印机中。

// write buffer to destination
if *printDest == "" {
	// write to stdout
	fmt.Printf("%s", buf.String())
} else {
	// write to printer
	cmd := exec.Command("lp", "-d" + *printDest)
	cmd.Stdin = strings.NewReader(buf.String())
	var stderr bytes.Buffer
	cmd.Stderr = &stderr
	err := cmd.Run()
	if err != nil {
		return fmt.Errorf("Write to printer failed with %s.\n Message: %s\n", err, stderr.Bytes())
	}
}

最后，使用 Go 工具构建并安装该程序:

go install github.com/whichxjy/Service-Computing-Homework/selpg

@whichxjy

命令行工具 selpg

实现过程

测试结果