package readers import ( "StoreBackEnd/pkg/protocol" "StoreBackEnd/pkg/protocol/repository" "StoreBackEnd/pkg/protocol/rules/writers" "StoreBackEnd/pkg/utils" "bufio" "fmt" "os" ) // RetrieveFileRulePrefix Identifiant de cette règle const RetrieveFileRulePrefix = "RETRIEVEFILE" // RetrieveFileRule Demande de suppression d'un fichier type RetrieveFileRule struct { // Cmd Nom de la règle Cmd string // matcher Permet de vérifier le matching matcher *protocol.RegexMatcher // protocolRepo Instance de ProtocolRepository protocolRepo *repository.ProtocolRepository // storagePath Chemin de stockage du fichier storagePath string } // CreateRetrieveFileRule Création d'une instance de RetrieveFileRule func CreateRetrieveFileRule(pattern string, protocolRepo *repository.ProtocolRepository, storagePath string) protocol.IProtocolReader { return &RetrieveFileRule{ Cmd: RetrieveFileRulePrefix, matcher: protocol.CreateRegexMatcher(pattern), protocolRepo: protocolRepo, storagePath: storagePath, } } func (rule RetrieveFileRule) GetCmd() string { return rule.Cmd } func (rule RetrieveFileRule) Execute(data string) (*protocol.ProtocolWriterResult, func(reader *bufio.Reader) *protocol.ProtocolWriterResult) { if rule.Match(data) { values := rule.matcher.Parse(data) hashFileName := values[1] path := fmt.Sprintf("%s/%s", rule.storagePath, hashFileName) if fileinfo, err := os.Stat(path); err == nil { println("OKOKOKOEKOEKE : ", hashFileName, fmt.Sprintf("%d", fileinfo.Size()), fmt.Sprintf("%x", utils.GenFingerPrint(path))) return rule.protocolRepo.ExecuteWriter(writers.RetrieveOkRulePrefix, hashFileName, fmt.Sprintf("%d", fileinfo.Size()), fmt.Sprintf("%x", utils.GenFingerPrint(path))), nil } } return rule.protocolRepo.ExecuteWriter(writers.RetrieveErrorRulePrefix), nil } func (rule RetrieveFileRule) Match(data string) bool { return rule.matcher.Match(data) }