Quits program if a file already exists (issue 22).

This commit is contained in:
Arman Bhalla 2017-10-19 22:29:37 +01:00
parent 91f31993ec
commit c5291cb6ad
1 changed files with 16 additions and 0 deletions

View File

@ -207,9 +207,16 @@ func (c *Connection) runClient() {
// have the main thread ask for the okay
if id == 0 {
fmt.Printf("Receiving file (%d bytes) into: %s\n", c.File.Size, c.File.Name)
var sentFileNames []string
if fileAlreadyExists(sentFileNames, c.File.Name) {
fmt.Printf("Will not overwrite file!")
os.Exit(1)
}
getOK := getInput("ok? (y/n): ")
if getOK == "y" {
gotOK = true
sentFileNames = append(sentFileNames, c.File.Name)
}
gotResponse = true
}
@ -273,6 +280,15 @@ func (c *Connection) runClient() {
}
}
func fileAlreadyExists(s []string, f string) bool {
for _, a := range s {
if a == f {
return true
}
}
return false
}
func (c *Connection) catFile(fname string) {
// cat the file
os.Remove(fname)