package readers import ( "StoreBackEnd/pkg/protocol" "fmt" "io" "net" "os" ) // SendFileRulePrefix Rule command prefix const SendFileRulePrefix = "ffe_sendfile" // SendFileRule Storage structure for the SendFileRule type SendFileRule struct { // cmd Rule name cmd string // matcher Allow to make a regex match matcher *protocol.RegexMatcher } // CreateSendFileRule Creating an instance of SendFileRule func CreateSendFileRule(pattern string) protocol.IProtocolReader { return &SendFileRule{ cmd: SendFileRulePrefix, matcher: protocol.CreateRegexMatcher(pattern), } } // GetCmd retrieve the command name. func (rule SendFileRule) GetCmd() string { return rule.cmd } // Execute the Rule with a string command. func (rule SendFileRule) Execute(data string) (string, bool, func(r net.Conn) (string, bool)) { if rule.Match(data) { // TODO : cloture this command. values := rule.matcher.Parse(data) // Values fileName := values[1] // fileSize := values[2] // fileContentHash := values[3] // print received data println(values[1], " File Name Hash") println(values[2], " File Size") println(values[3], " File Content Hash") // function callback callback := func(con net.Conn) (string, bool) { file, _ := os.Create(fmt.Sprintf("D:\\tmp\\%s", fileName)) _, err := io.Copy(file, con) println("HEY1") if err != nil { println("Can't copy file") return "SEND_ERROR\r", false } println("HEY2") return "SEND_OK\r", true } return "SEND_OK\r", true, callback } else { println("AHAHAHAHAHAHA") return "SEND_ERROR\r", false, nil } } // Match make a match with the current Rule and check if it matches. func (rule SendFileRule) Match(data string) bool { return rule.matcher.Match(data) }