mirrors_public/schollz_croc
0
0
Fork 0
mirror of https://github.com/schollz/croc.git synced 2025-10-11 13:21:00 +02:00
Code Activity

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

16
connect.go
View file

@ -207,9 +207,16 @@ func (c *Connection) runClient() {
// have the main thread ask for the okay // have the main thread ask for the okay
if id == 0 { if id == 0 {
fmt.Printf("Receiving file (%d bytes) into: %s\n", c.File.Size, c.File.Name) 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): ") getOK := getInput("ok? (y/n): ")
if getOK == "y" { if getOK == "y" {
gotOK = true gotOK = true
sentFileNames = append(sentFileNames, c.File.Name)
} }
gotResponse = true 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) { func (c *Connection) catFile(fname string) {
// cat the file // cat the file
os.Remove(fname) os.Remove(fname)