func startTransfer(filename string, numChunks, fsize int64, ws *websocket.Conn) (tc *transfer.TransferConnection) {
key := transfer.GetKeyForFilename(filename)
websocket.Message.Send(ws, "key|"+key)
tc = transfer.TheBookKeeper.GetTransferForKey(key)
tc.ReadySend(numChunks, fsize, filename)
websocket.Message.Send(ws, "ready|ready")
return
}
func (c App) SendChunk(ws *websocket.Conn) revelpkg.Result {
var data []byte
var filename string
numChunks := int64(-1)
var tc *transfer.TransferConnection
for {
err := websocket.Message.Receive(ws, &data)
fmt.Printf("received msg")
if err != nil {
fmt.Printf(err.Error())
return nil
}
if len(data) < 100 {
msg := string(data)
arr := strings.FieldsFunc(msg, func(c rune) bool {
return c == '|'
})
if arr != nil && len(arr) == 2 {
if arr[0] == "filename" {
filename = arr[1]
if numChunks > 0 {
key := transfer.GetKeyForFilename(filename)
websocket.Message.Send(ws, "key|"+key)
tc = transfer.TheBookKeeper.GetTransferForKey(key)
tc.ReadySend(numChunks, filename)
websocket.Message.Send(ws, "ready|ready")
}
} else if arr[0] == "numChunks" {
numChunks, err = strconv.ParseInt(arr[1], 10, 32)
if filename != "" {
key := transfer.GetKeyForFilename(filename)
websocket.Message.Send(ws, "key|"+key)
tc = transfer.TheBookKeeper.GetTransferForKey(key)
tc.ReadySend(numChunks, filename)
websocket.Message.Send(ws, "ready|ready")
}
} else {
panic("shit")
}
} else {
//if it doesn't match the protocol, just assume its data
if tc == nil {
panic("f**k")
}
next_chunk := strconv.FormatInt(tc.SendChunk(data), 10)
if !tc.Finished() {
websocket.Message.Send(ws, "next|"+next_chunk)
} else {
websocket.Message.Send(ws, "end|end")
}
}
} else {
if tc == nil {
panic("holyballs")
}
next_chunk := strconv.FormatInt(tc.SendChunk(data), 10)
if !tc.Finished() {
websocket.Message.Send(ws, "next|"+next_chunk)
} else {
websocket.Message.Send(ws, "end|end")
}
}
}
return nil
}