{ config, lib, pkgs, ... }:

with lib;
with import ./initrd-order.nix;

let
  cfg = config.mobile.boot.stage-1;
  device_name = device_config.name;
  device_config = config.mobile.device;
  system_type = config.mobile.system.type;
in
{
  # FIXME Generic USB gadget support to come.
  options.mobile.boot.stage-1.usb = {
    enable = mkOption {
      type = types.bool;
      default = true;
      description = ''
        Enables USB features.
        For now, only Android-based devices are supported.
      '';
    };
    features = mkOption {
      type = types.listOf types.str;
      default = [];
      description = ''
        `android_usb` features to enable.
      '';
    };
  };

  config.mobile.boot.stage-1 = lib.mkIf cfg.usb.enable {
    usb.features = []
      ++ optional cfg.networking.enable "rndis"
    ;

    # TODO: Only run, when we have the android usb driver
    init = lib.mkOrder AFTER_DEVICE_INIT ''
      # Setting up Android-specific USB.
      (
      SYS=/sys/class/android_usb/android0
      if [ -e "$SYS" ]; then
        printf "%s" "0"    > "$SYS/enable"
        printf "%s" "18D1" > "$SYS/idVendor"
        printf "%s" "D001" > "$SYS/idProduct"
        printf "%s" "0"    > "$SYS/bDeviceClass"
        printf "%s" "${concatStringsSep "," cfg.usb.features}" > "$SYS/functions"
        printf "%s" "mobile-nixos" > "$SYS/iManufacturer"
        printf "%s" "${device_name}" > "$SYS/iProduct"
        printf "%s" "0123456789" > "$SYS/iSerial"

        sleep 0.1
        printf "%s" "1" > "$SYS/enable"
      fi
      )
    '';
  };
}