# SPDX-FileCopyrightText: 2022 Richard Brežák <richard@brezak.sk>
#
# SPDX-License-Identifier: LGPL-3.0-or-later
{
  notnft,
  inputs',
  lib,
  config,
  ...
}: let
  inherit
    (lib)
    mkBefore
    flip
    genAttrs
    ;
in {
  networking.notnft = {
    enable = true;
    flush = false;
  };

  networking.notnft.preRules = [
    {
      add.table = {
        family = "bridge";
        name = "bridge-t";
      };
    }
    {
      flush.table = {
        family = "bridge";
        name = "bridge-t";
      };
    }
  ];

  networking.notnft.rules = let
    interfaces = ["mvm-test" "mvm0"];
    logRule = with notnft.dsl;
    with payload;
      prefix: [
        (log {
          prefix = "${prefix} dropped:  ";
          flags = f: [f.all];
        })
      ];

    dropRule = with notnft.dsl; with payload; [drop];
  in
    with notnft.dsl;
    with payload;
      ruleset {
        bridge-t = add table {family = f: f.bridge;} {
          input-body = add chain;

          input-mvm =
            add chain
            [
              (vmap ct.state {
                established = accept;
                related = accept;
                invalid = drop;
              })
            ]
            [(is.eq meta.protocol (f: f.arp)) accept]
            [(mangle meta.nftrace 1)]
            [(jump "input-body")]
            (logRule "Bridge input")
            dropRule;

          input =
            add chain
            {
              type = f: f.filter;
              hook = f: f.input;
              prio = 0;
              policy = f: f.accept;
            }
            [(vmap meta.iifname (genAttrs interfaces (_: (goto "input-mvm"))))]
            [(vmap meta.oifname (genAttrs interfaces (_: (goto "input-mvm"))))];

          output-body = add chain;

          output-mvm =
            add chain
            [(is.eq ether.type (f: f.arp)) accept]
            [(mangle meta.nftrace 1)]
            [(jump "output-body")]
            (logRule "Bridge output")
            dropRule;

          output =
            add chain
            {
              type = f: f.filter;
              hook = f: f.output;
              prio = 0;
              policy = f: f.accept;
            }
            [(vmap meta.iifname (genAttrs interfaces (_: (goto "output-mvm"))))]
            [(vmap meta.oifname (genAttrs interfaces (_: (goto "output-mvm"))))];

          forward-body = add chain;

          forward-mvm =
            add chain
            [(mangle meta.nftrace 1)]
            [(jump "forward-body")]
            (logRule "Bridge forward")
            dropRule;

          forward =
            add chain
            {
              type = f: f.filter;
              hook = f: f.forward;
              prio = 0;
              policy = f: f.accept;
            }
            [(vmap meta.iifname (genAttrs interfaces (_: (goto "input-mvm"))))]
            [(vmap meta.oifname (genAttrs interfaces (_: (goto "input-mvm"))))];

          # prerouting = add chain
          #   { type = f: f.filter; hook = f: f.prerouting; prio = -300; policy = f: f.accept; }
          # ;

          # postrouting = add chain
          #   { type = f: f.filter; hook = f: f.postrouting; prio = -300; policy = f: f.accept; }
          # ;
        };
      };

  systemd.services.notnftables = {
    requires = ["nftables.service"];
    after = ["nftables.service"];
  };

  networking.bridges.mvm0 = {
    interfaces = [];
  };

  networking.interfaces.mvm0 = {
    useDHCP = false;
    ipv4.addresses = [
      {
        address = "10.80.1.1";
        prefixLength = 24;
      }
    ];
  };

  microvm.services.tcpUdp.test-ssh = {
    hostName = "test";
    port = 22;
    protocol = ["tcp"];
  };

  microvm.services.tcpUdp.test-http = {
    hostName = "test";
    port = 80;
    protocol = ["tcp"];
  };

  microvm.services.icmp.test = {
    hostName = "test";
  };

  microvm.connections.tcpUdp = [
    {
      target = "test-ssh";
    }
    {
      target = "test-http";
    }
  ];
  microvm.connections.icmp = [
    {
      target = "test";
    }
  ];

  microvm.vms = {
    test.config = {
      imports = [inputs'.self.nixosModules.microvm-extras];

      microvm = {
        hostName = "test";
        hostsHostName = "omen";
        groupId = 1;
        taskId = 2;
      };

      microvm.hypervisor = "cloud-hypervisor";
      microvm.shares = [
        {
          source = "/nix/store";
          mountPoint = "/nix/.ro-store";
          tag = "ro-store";
          proto = "virtiofs";
        }
      ];
      microvm.storeOnDisk = false;

      networking.firewall.allowedTCPPorts = [80 22];

      services.nginx = {
        enable = true;
        virtualHosts."example.com" = {
          root = "/var/www/blog";
        };
      };

      users.users.root.password = "";
      services.getty.helpLine = ''
        Log in as "root" with an empty password.
      '';
      services.openssh = {
        enable = true;
        settings.PermitRootLogin = "yes";
      };
    };
  };
}