add support for ggsave function

NAMESPACE

@@ -39,6 +39,7 @@ S3method(ggheatmap_add,data.frame)
 S3method(ggheatmap_add,default)
 S3method(ggheatmap_add,list)
 S3method(ggplot_build,ggheatmap)
+S3method(grid.draw,ggheat)
 S3method(heatmap_add,CoordFlip)
 S3method(heatmap_add,default)
 S3method(heatmap_add,facetted_pos_scales)
@@ -86,6 +87,7 @@ importFrom(ggplot2,ggplot_build)
 importFrom(ggplot2,is.ggplot)
 importFrom(ggplot2,waiver)
 importFrom(grid,gpar)
+importFrom(grid,grid.draw)
 importFrom(grid,is.unit)
 importFrom(grid,unit)
 importFrom(grid,unit.c)

R/ggheat.R

@@ -195,13 +195,23 @@ methods::setClass(
 
 #' Subset a `ggheatmap` object
 #'
-#' Used by [ggplot_build][ggplot2::ggplot_build]
+#' Used by [ggplot_build][ggplot2::ggplot_build] and [ggsave][ggplot2::ggsave]
 #'
 #' @param x A ggheatmap object
 #' @param name A string of slot name in `ggheatmap` object.
 #' @keywords internal
 methods::setMethod("$", "ggheatmap", function(x, name) {
-    slot(x, name)
+    # https://github.com/tidyverse/ggplot2/issues/6002
+    if (name == "theme") {
+        slot(x, "heatmap")$theme
+    } else if (name == "plot_env") {
+        slot(x, "plot_env")
+    } else {
+        cli::cli_abort(c(
+            "`$` is just for internal usage for ggplot2 methods",
+            i = "try to use `@` method instead"
+        ))
+    }
 })
 
 #' @importFrom ggplot2 is.ggplot

R/plot-build.R

@@ -4,6 +4,12 @@ print.ggheatmap <- function(x, ...) {
     print(p)
 }
 
+#' @importFrom grid grid.draw
+#' @exportS3Method
+grid.draw.ggheat <- function(x, ...) {
+    print(x, ...)
+}
+
 #' @importFrom ggplot2 ggplot_build
 #' @export
 ggplot_build.ggheatmap <- function(plot) {