]> Sergey Matveev's repositories - tofuproxy.git/blobdiff - fifos/warcs.go
WARC
[tofuproxy.git] / fifos / warcs.go
diff --git a/fifos/warcs.go b/fifos/warcs.go
new file mode 100644 (file)
index 0000000..6d38700
--- /dev/null
@@ -0,0 +1,78 @@
+/*
+tofuproxy -- flexible HTTP/WARC proxy with TLS certificates management
+Copyright (C) 2021 Sergey Matveev <stargrave@stargrave.org>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, version 3 of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package fifos
+
+import (
+       "bufio"
+       "fmt"
+       "log"
+       "os"
+
+       "go.stargrave.org/tofuproxy/warc"
+)
+
+func listWARCs(p string) {
+       for {
+               fd, err := os.OpenFile(p, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
+               if err != nil {
+                       log.Fatalln(err)
+               }
+               warc.WARCsM.RLock()
+               for warcPath, uris := range warc.WARCs {
+                       fmt.Fprintf(fd, "%s\t%d\n", warcPath, len(uris))
+               }
+               warc.WARCsM.RUnlock()
+               fd.Close()
+       }
+}
+
+func addWARC(p string) {
+       for {
+               fd, err := os.OpenFile(p, os.O_RDONLY, os.FileMode(0666))
+               if err != nil {
+                       log.Fatalln(err)
+               }
+               var warcPaths []string
+               scanner := bufio.NewScanner(fd)
+               for scanner.Scan() {
+                       t := scanner.Text()
+                       if len(t) > 0 {
+                               warcPaths = append(warcPaths, t)
+                       }
+               }
+               fd.Close()
+               for _, warcPath := range warcPaths {
+                       if warcPath == "SAVE" {
+                               if err = warc.SaveIndexes(); err != nil {
+                                       log.Printf("%s: can not save index %s: %+v\n", p, warcPath, err)
+                               }
+                               continue
+                       }
+                       if _, exists := warc.WARCs[warcPath]; exists {
+                               continue
+                       }
+                       log.Printf("%s: adding WARC %s\n", p, warcPath)
+                       err = warc.Add(warcPath)
+                       if err != nil {
+                               log.Printf("%s: can not open %s: %+v\n", p, warcPath, err)
+                               break
+                       }
+                       log.Printf("%s: %s: added %d URIs\n", p, warcPath, len(warc.WARCs[warcPath]))
+               }
+       }
+}