Update start-colab.sh to add plugins dynamically

Signed-off-by: Matheus Fernandes <[email protected]>

Dockerfile

@@ -6,7 +6,8 @@ ENV BUILD_PACKAGES python-dev python3 python3-dev linux-headers \
 
 RUN apk add --update --no-cache $BUILD_PACKAGES
 RUN mkdir -p /etc/colab /etc/colab/settings.d /etc/colab/plugins.d \
-    /etc/colab/widgets.d /var/labhacker/colab/public/media
+    /etc/colab/widgets.d /var/labhacker/colab/public/media \
+    /etc/cron.d/
 
 ADD . /var/labhacker/colab
 WORKDIR /var/labhacker/colab

docker-compose.yml

@@ -61,6 +61,8 @@ services:
       SITE_NAME: Câmara dos Deputados
       SITE_LOGO: ''
       SITE_BACKGROUND_IMAGE: ''
+      ENABLE_AUDIENCIAS: 'true'
+      ENABLE_WIKILEGIS: 'true'
       AUDIENCIAS_API_KEY: secret_key
       WIKILEGIS_API_KEY: api_key
     volumes:

misc/etc/cron.d/colab-audiencias

@@ -0,0 +1 @@
+*   *   *   *   *  colab-admin fetch_data_audiencias

misc/etc/cron.d/colab-wikilegis

@@ -0,0 +1 @@
+*/30   *   *   *   * colab-admin fetch_data_wikilegis

start-colab.sh

@@ -1,11 +1,35 @@
 #!/bin/bash
 
-cp ./misc/etc/colab/plugins.d/audiencias.py /etc/colab/plugins.d/
-cp ./misc/etc/colab/widgets.d/audiencias_widgets.py /etc/colab/widgets.d/
+PLUGINS_SETTINGS="/etc/colab/plugins.d/"
+if [ "$(ls -A $PLUGINS_SETTINGS)" ]; then
+    find $PLUGINS_SETTINGS ! -name 'edemocracia.py' -type f -exec rm -f {} +
+fi
 
-cp  ./misc/etc/colab/plugins.d/wikilegis.py /etc/colab/plugins.d/
-cp ./misc/etc/colab/widgets.d/wikilegis_widgets.py /etc/colab/widgets.d/
+PLUGINS_WIDGETS="/etc/colab/widgets.d/"
+if [ "$(ls -A $PLUGINS_WIDGETS)" ]; then
+    rm /etc/colab/widgets.d/*
+fi
 
+activate_plugin() {
+    PLUGIN_NAME=$1
+    cp ./misc/etc/colab/plugins.d/${PLUGIN_NAME}.py /etc/colab/plugins.d/
+    cp ./misc/etc/colab/widgets.d/${PLUGIN_NAME}_widgets.py /etc/colab/widgets.d/
+    cp ./misc/etc/cron.d/colab-${PLUGIN_NAME} /etc/cron.d/
+    crontab /etc/cron.d/colab-${PLUGIN_NAME}
+}
+
+[[ -z "${ENABLE_AUDIENCIAS}" ]] && ENABLE_AUDIENCIAS=2 || ENABLE_AUDIENCIAS="${ENABLE_AUDIENCIAS}"
+[[ -z "${ENABLE_WIKILEGIS}" ]] && ENABLE_WIKILEGIS=2 || ENABLE_WIKILEGIS="${ENABLE_WIKILEGIS}"
+
+if [[ "$ENABLE_WIKILEGIS" = true ]]; then
+    activate_plugin wikilegis
+fi
+
+if [[ "$ENABLE_AUDIENCIAS" = true ]]; then
+    activate_plugin audiencias
+fi
+
+crond
 colab-admin migrate
 colab-admin initdb
 colab-admin compress --force