nixops/modules/websites/commons/stats.nix

   1 { lib, pkgs, config, mylibs, ... }:
   2 let
   3     cfg = config.services.myWebsites.commons.stats;
   4 in {
   5   options = {
   6     services.myWebsites.commons.stats = {
   7       enable = lib.mkEnableOption "enable statistics";
   8       sites = lib.mkOption {
   9         type = lib.types.listOf (lib.types.submodule {
  10           options = {
  11             conf = lib.mkOption {
  12               type = lib.types.nullOr lib.types.path;
  13               default = null;
  14             };
  15             name = lib.mkOption { type = lib.types.string; };
  16           };
  17         });
  18         default = [];
  19         description = "Sites to generate stats";
  20       };
  21     };
  22   };
  23
  24   config = lib.mkIf cfg.enable {
  25     users.users.root.packages = [
  26       pkgs.goaccess
  27     ];
  28
  29     services.cron = {
  30       enable = true;
  31       systemCronJobs = let
  32         stats = domain: conf: let
  33           config = if builtins.isNull conf
  34             then pkgs.runCommand "goaccess.conf" {
  35                 dbPath = "/var/lib/goaccess/${domain}";
  36               } "substituteAll ${./goaccess.conf} $out"
  37             else conf;
  38           d = pkgs.writeScriptBin "stats-${domain}" ''
  39             #!${pkgs.stdenv.shell}
  40             set -e
  41             shopt -s nullglob
  42             date_regex=$(LC_ALL=C date -d yesterday +'%d\/%b\/%Y')
  43             TMPFILE=$(mktemp)
  44             trap "rm -f $TMPFILE" EXIT
  45
  46             cat /var/log/httpd/access_log-${domain} | sed -n "/\\[$date_regex/ p" > $TMPFILE
  47             for i in /var/log/httpd/access_log-${domain}*.gz; do
  48               zcat "$i" | sed -n "/\\[$date_regex/ p" >> $TMPFILE
  49             done
  50             ${pkgs.goaccess}/bin/goaccess $TMPFILE --no-progress -o /var/lib/goaccess/${domain}/index.html -p ${config}
  51             '';
  52           in "${d}/bin/stats-${domain}";
  53         allStats = sites: pkgs.writeScript "stats" ''
  54           #!${pkgs.stdenv.shell}
  55
  56           ${builtins.concatStringsSep "\n" (map (v: stats v.name v.conf) sites)}
  57           '';
  58       in
  59         [
  60           "5 0 * * * root ${allStats cfg.sites}"
  61         ];
  62     };
  63
  64     system.activationScripts.goaccess = ''
  65       mkdir -p /var/lib/goaccess
  66     '' +
  67     builtins.concatStringsSep "\n" (map (v: "mkdir -p /var/lib/goaccess/${v.name}") cfg.sites);
  68   };
  69 }