4 files changed, 70 insertions, 4 deletions
diff --git a/pkgs/default.nix b/pkgs/default.nix
index a299b8b..de22536 100644
--- a/pkgs/default.nix
+++ b/pkgs/default.nix
@@ -27,7 +27,11 @@ in
     ];
   };
 
-  override =
+  override = {
+    home-assistant-custom-components = prev.home-assistant-custom-components // {
+      dreame_vacuum = final.home-assistant.python.pkgs.callPackage ./dreame-vacuum.nix { };
+    };
+  } // (
     let
       makePyOverrides = version:
         let
@@ -58,5 +62,6 @@ in
 
       pyVersionRange = start: pyVersionRange' start start;
     in
-    listToAttrs (map makePyOverrides (pyVersionRange 9));
+    listToAttrs (map makePyOverrides (pyVersionRange 9))
+  );
 }
diff --git a/pkgs/dreame-vacuum.nix b/pkgs/dreame-vacuum.nix
new file mode 100644
index 0000000..1039405
--- /dev/null
+++ b/pkgs/dreame-vacuum.nix
@@ -0,0 +1,40 @@
+{ buildHomeAssistantComponent
+, callPackage
+, fetchFromGitHub
+, numpy
+, paho-mqtt
+, pillow
+, pybase64
+, pycryptodome
+, python-miio
+, requests
+}:
+let
+  version = "2.0.0b15";
+in
+buildHomeAssistantComponent {
+  owner = "tasshack";
+  domain = "dreame_vacuum";
+  inherit version;
+
+  src = fetchFromGitHub {
+    repo = "dreame-vacuum";
+    owner = "Tasshack";
+
+    rev = "v${version}";
+    hash = "sha256-ydWirMQsO6DYU5UdygHRXCx8I3M7haYX9zPLela9H7E=";
+  };
+
+  dontBuild = true;
+
+  propagatedBuildInputs = [
+    numpy
+    paho-mqtt
+    pillow
+    pybase64
+    pycryptodome
+    python-miio
+    requests
+    (callPackage ./py-mini-racer.nix { })
+  ];
+}
diff --git a/pkgs/py-mini-racer.nix b/pkgs/py-mini-racer.nix
new file mode 100644
index 0000000..5f2bcbc
--- /dev/null
+++ b/pkgs/py-mini-racer.nix
@@ -0,0 +1,20 @@
+{ buildPythonPackage
+, fetchPypi
+, pytest
+, six
+, tox
+}:
+let
+  pname = "py_mini_racer";
+  version = "0.6.0";
+in
+buildPythonPackage {
+  inherit pname version;
+
+  src = fetchPypi {
+    inherit pname version;
+    hash = "sha256-9x42tkPZR7ppjFfNm9IjLIPKmXsIAvwvf3lYI3cEDBE=";
+  };
+
+  doCheck = false;
+}
diff --git a/sys/home-assistant/default.nix b/sys/home-assistant/default.nix
index eaa9cc9..8af383c 100644
--- a/sys/home-assistant/default.nix
+++ b/sys/home-assistant/default.nix
@@ -30,8 +30,9 @@ in
       };
 
       customComponents = with pkgs.home-assistant-custom-components; [
-	    xiaomi_miot
-	  ];
+        dreame_vacuum
+        xiaomi_miot
+      ];
     };
   };
 }