CLI 命令列實用程式開發基礎-有解無憂

服務計算Homework03

專案地址
使用說明：根據博客內容配置好環境后，在對應goPath/src中創建selpg/selpg.go,完成代碼后在goPath/src/selpg中執行go build生成selpg檔案，即可在當前路徑下使用selpg xxx對應命令

目的

使用 golang 開發開發 Linux 命令列實用程式中的 selpg

要求

請按檔案使用 selpg 章節要求測驗你的程式
請使用 pflag 替代 goflag 以滿足 Unix 命令列規范，參考：Golang之使用Flag和Pflag：
golang 檔案讀寫、讀環境變數，請自己查 os 包
“-dXXX” 實作，請自己查 os/exec 庫，例如案例Command，管理子行程的標準輸入和輸出通常使用 io.Pipe，具體案例見Pipe
請自帶測驗程式，確保函式等功能正確

環境配置

Linux配置go語言環境
使用命令列apt install golang安裝的go語言沒有pflag庫，手動安裝命令為go get github.com/spf13/pflag，安裝好后可以直接import

代碼實作

引數結構體selpg_args

將page_type改為bool型別，因為pflag是要識別選項后面跟著的引數的，但-f后面不跟引數，如果用整數型別會報錯，而布爾型別的一個特點是不能直接在選項后面跟引數，如果要跟引數，必須用一個等號連接，如selpg -s1 -e2 -f=true，如果出現-f也相當于-f=true，所以如果沒有出現-f選項，則page_type為默認值false（對應定長72行數），如果出現-f選項，則page_type設為true（對應不定長行數），其他與c語言版本相同

type selpg_args struct {
        start_page int 
        end_page int
        page_len int
        page_type bool
        in_filename string
        print_dest string
}

獲取引數

由課件可得os、flag包的引數簡單處理為：

package main

import (
    "fmt"
    "os"
)

func main() {
    for i, a := range os.Args[1:] {
        fmt.Printf("Argument %d is %s\n", i+1, a)
    }

}

命令列引數決議FLAG

package main

import (
    "flag" 
    "fmt"
)

func main() {
    var port int
    flag.IntVar(&port, "p", 8000, "specify port to use.  defaults to 8000.")
    flag.Parse()

    fmt.Printf("port = %d\n", port)
    fmt.Printf("other args: %+v\n", flag.Args())
}

仿寫上述代碼獲取引數如下

func get_args(sa selpg_args) {
	progname = os.Args[0]
	flag.IntVarP(&sa.start_page, "start_page", "s", -1, "start page")
    flag.IntVarP(&sa.end_page, "end_page", "e", -1, "end page")
    flag.IntVarP(&sa.page_len, "page_len", "l", 15, "page len")
    flag.BoolVarP(&sa.page_type, "page_type", "f", false, "page_type")
    flag.StringVarP(&sa.print_dest, "print_dest", "d", "", "print destination")

    flag.Parse()
    if flag.NArg() > 0 {
        sa.in_filename = flag.Arg(0)
    }
}

檢查引數

對下列進行逐項檢查
引數個數
開始頁數、結束頁數、每頁行數越界
開始頁數>結束頁數
輸入檔案

func check_args(sa selpg_args) {
    if len(os.Args) < 3 {
        fmt.Fprintf(os.Stderr, "%s: not enough arguments\n", progname)
        usage()
        os.Exit(1)
    }
    if sa.start_page < 1 || sa.start_page > INT_MAX {
        fmt.Fprintf(os.Stderr, "%s: invalid start page %d\n", progname, sa.start_page)
        usage()
        os.Exit(2)
    }
    if sa.end_page < 1 || sa.end_page > INT_MAX {
        fmt.Fprintf(os.Stderr, "%s: invalid end page %d\n", progname, sa.end_page)
        usage()
        os.Exit(3)
	}
	if page_type == true && (sa.page_len < 1 || sa.page_len > INT_MAX) {
        fmt.Fprintf(os.Stderr, "%s: invalid page length %d\n", progname, sa.page_len)
        usage()
        os.Exit(4)
    }
    if sa.end_page < sa.start_page {
        fmt.Fprintf(os.Stderr, "%s: end page should not be less than start pag\n", progname)
        usage()
        os.Exit(3)
    }
    if sa.in_filename != "" {
        if _, err := os.Stat(sa.in_filename); os.IsNotExist(err) {
            fmt.Fprintf(os.Stderr, "%s: input file \"%s\" does not exist\n", progname, sa.in_filename)
			os.Exit(5);
        }
    }
}

讀取輸入

用緩沖區bufio.Reader進行輸入，如果輸入命令中不包含輸入檔案，則輸入默認來自標準輸入，否則嘗試打開檔案，將檔案流作為輸入

var reader *bufio.Reader
if sa.in_filename == "" {
    reader = bufio.NewReader(os.Stdin)
} else {
    fin, err := os.Open(sa.in_filename)
    if err != nil {
        fmt.Fprintf(os.Stderr, "%s: could not open input file \"%s\"\n", progname, sa.in_filename)
        os.Exit(6)
    }
    reader = bufio.NewReader(fin)
    defer fin.Close()
}

輸出

采用管道輸出，首先通過exec.Command創建了一個子行程，執行列印，列印的目的地為輸入命令中的目的地，如果輸入命令中沒有目的地選項，則輸出默認為標準輸出，然后將程式的輸出流writer設為列印行程的輸入管道，實作將讀取的內容通過管道寫到目的檔案中

var writer io.WriteCloser 
if sa.print_dest == "" {
    writer = os.Stdout
} else {
    cmd := exec.Command("lp","-d"+ sa.print_dest)
    var err error
    if writer, err = cmd.StdinPipe(); err != nil {
        fmt.Fprintf(os.Stderr, "%s: could not open pipe to \"%s\"\n", progname, sa.print_dest)
        fmt.Println(err)
        os.Exit(7)
    }
    cmd.Stdout = os.Stdout;
    cmd.Stderr = os.Stderr;
    if err = cmd.Start(); err != nil {
        fmt.Fprintf(os.Stderr, "%s: cmd start error\n", progname)
        fmt.Println(err)
        os.Exit(8)
    }
}

檔案讀寫

基本套用c語言版本的思路，不同的是將兩種不同頁型別（定長和不定長）的讀取統一在一個回圈內，因為不定長頁是以\f為結束符的，所以reader.ReadString每讀一個\f，不定長頁就增加一頁

line_ctr, page_ctr, page_len := 1, 1, sa.page_len
ptFlag := '\n'
if sa.page_type {
    ptFlag = '\f'
    page_len = 1
}

for {
    line, crc := reader.ReadString(byte(ptFlag));
    if crc != nil && len(line) == 0 {
        break
    }
    if line_ctr > page_len {
        page_ctr++
        line_ctr = 1
    }
    if page_ctr >= sa.start_page && page_ctr <= sa.end_page {
        _, err := writer.Write([]byte(line))
        if err != nil {
            fmt.Println(err)
            os.Exit(9)
        }
    }
    line_ctr++
}

// 檢查讀取是否出錯
if page_ctr < sa.start_page {
    fmt.Fprintf(os.Stderr, "\n%s: start_page (%d) greater than total pages (%d)," + " no output written\n", progname, sa.start_page, page_ctr)
} else if page_ctr < sa.end_page {
    fmt.Fprintf(os.Stderr, "\n%s: end_page (%d) greater than total pages (%d)," + " less output than expected\n", progname, sa.end_page, page_ctr)
}