Also allow mounting a /preseed_services.txt to preseed specified services

Author: manisandro

Diff for: README.md

@@ -34,13 +34,20 @@ Add the `qwc-qgs-cache-preseed` container configuration to your QWC `docker-comp
       QGS_EXT: ".qgs"
       FCGI_INSTANCES: 10
     volumes:
-      - ./volumes/qgs-resources:/data:ro
+      - ./volumes/preseed_services.txt:/preseed_services.txt:ro
+      # OR
+      # - ./volumes/qgs-resources:/data:ro
 ```
 
 Configuration
 -------------
 
-Make sure to mount the `qgs-resources` dir (or whichever directory is mounted to `/data` for `qwc-qgis-server`) to `/data`.
+To control which QGS projects will be processed, you can:
+
+- Mount a file to `/preseed_services.txt` which contains the services names, one per line. For example:
+  -  `subdir/projectname` for a QGS file located in `qgs-resources/subdir/projectname.qgs`
+  - `pg/schema/projectname` for a QGS project located in a DB in schema `schema` and named `projectname`
+- Mount the `qgs-resources` dir (or whichever directory is mounted to `/data` for `qwc-qgis-server`) to `/data`, which will be then searches for projects (ending which `$QGS_EXT`).
 
 The following environment variables can be set:
 

Diff for: qgs_cache_preseed.sh

@@ -3,22 +3,32 @@
 projects=$(find /data -name '*'${QGS_EXT})
 
 preseed() {
-  service_name=$1
+  url=$1
   nr=$2
   echo "- Sending request $nr"
-  code=$(curl -o /dev/null -s -w "%{http_code}" "http://qwc-qgis-server/ows/${service_name}?SERVICE=WMS&REQUEST=GetCapabilities&VERSION=1.3.0")
+  code=$(curl -o /dev/null -s -w "%{http_code}" "${url}")
   echo "- Request $nr completed with status code $code"
 }
 
+if [ -f /preseed_services.txt ]; then
+  echo "Reading services names from preseed_services.txt..."
+  service_names=$(cat /preseed_services.txt | sed 's|^/ows/||');
+elif [ -d /data ]; then
+  echo "Scanning /data for projects..."
+  service_names=$(find /data -name '*'${QGS_EXT} | sed "s|^/data/||; s|${QGS_EXT}$||")
+else
+  echo "No service names found. Mount a file to /preseed_services.txt or mount your projects dir to /data."
+  exit 0
+fi
 
-find /data -name '*'${QGS_EXT} -print0 | while IFS= read -r -d $'\0' file; do
-  echo "Processing project $file"
-  service_name=$(echo $file | sed 's|^\/data/||' | sed "s|$QGS_EXT||")
-  echo "- Service name is ${service_name}"
+echo "$service_names" | while IFS= read -r service_name; do
+  echo "Processing service ${service_name}"
+  url="http://qwc-qgis-server/ows/${service_name}?SERVICE=WMS&REQUEST=GetCapabilities&VERSION=1.3.0"
+  echo "- Request URL: ${url}"
 
   pids=""
   for i in $(seq 1 $FCGI_INSTANCES); do
-    preseed "${service_name}" $i &
+    preseed "${url}" $i &
     pid=$!
     pids="$pids $pid"
   done