fileutils.go
No OneTemporary
Actions

Size

1 KB

Subscribers

None

fileutils.go
View Options

	package fileutils

	import (
	"bufio"
	"fmt"
	"os"
	"strings"
	)

	// Exists returns whether or not the given file exists
	func Exists(p string) bool {
	if _, err := os.Stat(p); err == nil {
	return true
	}
	return false
	}

	func WriteData(path string, data []byte) {
	f, err := os.OpenFile(path, os.O_CREATE\|os.O_WRONLY\|os.O_TRUNC, 0600)
	if err != nil {
	fmt.Println(err)
	}
	// TODO: check for Close() errors
	// https://github.com/ncw/swift/blob/master/swift.go#L170
	defer f.Close()

	_, err = f.Write(data)
	if err != nil {
	fmt.Println(err)
	}
	}

	func RemoveLine(p, startsWith string) {
	f, err := os.Open(p)
	if err != nil {
	return
	}
	defer f.Close()

	var outText string
	found := false

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {
	if strings.HasPrefix(scanner.Text(), startsWith) {
	found = true
	} else {
	outText += scanner.Text() + string('\n')
	}
	}

	if err := scanner.Err(); err != nil {
	return
	}

	if found {
	WriteData(p, []byte(outText))
	}
	}

	func FindLine(p, startsWith string) string {
	f, err := os.Open(p)
	if err != nil {
	return ""
	}
	defer f.Close()

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {
	if strings.HasPrefix(scanner.Text(), startsWith) {
	return scanner.Text()
	}
	}

	if err := scanner.Err(); err != nil {
	return ""
	}

	return ""
	}