Move ggtime method registration from .onAttach to .onLoad

Author: mitchelloharawild

R/zzz.R

@@ -12,6 +12,37 @@
 
   register_s3_method("ggplot2", "scale_type", "agg_vec")
 
+  # fabletools -> ggtime method migration
+  ggtime_version <- package_version(as.character(
+    suppressWarnings(
+      utils::packageDescription("ggtime", fields = "Version")
+    )
+  ))
+  if (!is.na(ggtime_version)) {
+    ggtime_ns <- getNamespace("ggtime")
+    
+    register_s3_method("ggplot2", "autoplot", "tbl_ts", utils::getS3method("autoplot", "tbl_ts", envir = ggtime_ns))
+    register_s3_method("ggplot2", "autolayer", "tbl_ts", utils::getS3method("autolayer", "tbl_ts", envir = ggtime_ns))
+    register_s3_method("ggplot2", "autoplot", "dcmp_ts", utils::getS3method("autoplot", "dcmp_ts", envir = ggtime_ns))
+    if (ggtime_version >= "0.2.0") {
+      register_s3_method("ggplot2", "autoplot", "fbl_ts", utils::getS3method("autoplot", "fbl_ts", envir = ggtime_ns))
+      register_s3_method("ggplot2", "autolayer", "fbl_ts", utils::getS3method("autolayer", "fbl_ts", envir = ggtime_ns))
+      register_s3_method("ggplot2", "fortify", "tbl_ts", utils::getS3method("fortify", "fbl_ts", envir = ggtime_ns))
+    } else {
+      register_s3_method("ggplot2", "autoplot", "fbl_ts", autoplot.fbl_ts)
+      register_s3_method("ggplot2", "autolayer", "fbl_ts", autolayer.fbl_ts)
+      register_s3_method("ggplot2", "fortify", "tbl_ts", fortify.fbl_ts)
+    }
+  } else {
+    register_s3_method("ggplot2", "autoplot", "fbl_ts", autoplot.fbl_ts)
+    register_s3_method("ggplot2", "autoplot", "tbl_ts", autoplot.tbl_ts)
+    register_s3_method("ggplot2", "autoplot", "dcmp_ts", autoplot.dcmp_ts)
+    register_s3_method("ggplot2", "autolayer", "fbl_ts", autolayer.fbl_ts)
+    register_s3_method("ggplot2", "autolayer", "tbl_ts", autolayer.fbl_ts)
+    register_s3_method("ggplot2", "fortify", "tbl_ts", fortify.fbl_ts)
+  }
+
+
   op <- options()
   op.fable <- list(
     fable.show_progress = TRUE
@@ -45,35 +76,4 @@ register_s3_method <- function(pkg, generic, class, fun = NULL) {
     }
   )
 }
-
-.onAttach <- function(...) {
-  ggtime_version <- package_version(as.character(
-    suppressWarnings(
-      utils::packageDescription("ggtime", fields = "Version")
-    )
-  ))
-  if (!is.na(ggtime_version)) {
-    ggtime_ns <- getNamespace("ggtime")
-    
-    register_s3_method("ggplot2", "autoplot", "tbl_ts", utils::getS3method("autoplot", "tbl_ts", envir = ggtime_ns))
-    register_s3_method("ggplot2", "autolayer", "tbl_ts", utils::getS3method("autolayer", "tbl_ts", envir = ggtime_ns))
-    register_s3_method("ggplot2", "autoplot", "dcmp_ts", utils::getS3method("autoplot", "dcmp_ts", envir = ggtime_ns))
-    if (ggtime_version >= "0.2.0") {
-      register_s3_method("ggplot2", "autoplot", "fbl_ts", utils::getS3method("autoplot", "fbl_ts", envir = ggtime_ns))
-      register_s3_method("ggplot2", "autolayer", "fbl_ts", utils::getS3method("autolayer", "fbl_ts", envir = ggtime_ns))
-      register_s3_method("ggplot2", "fortify", "tbl_ts", utils::getS3method("fortify", "fbl_ts", envir = ggtime_ns))
-    } else {
-      register_s3_method("ggplot2", "autoplot", "fbl_ts", autoplot.fbl_ts)
-      register_s3_method("ggplot2", "autolayer", "fbl_ts", autolayer.fbl_ts)
-      register_s3_method("ggplot2", "fortify", "tbl_ts", fortify.fbl_ts)
-    }
-  } else {
-    register_s3_method("ggplot2", "autoplot", "fbl_ts", autoplot.fbl_ts)
-    register_s3_method("ggplot2", "autoplot", "tbl_ts", autoplot.tbl_ts)
-    register_s3_method("ggplot2", "autoplot", "dcmp_ts", autoplot.dcmp_ts)
-    register_s3_method("ggplot2", "autolayer", "fbl_ts", autolayer.fbl_ts)
-    register_s3_method("ggplot2", "autolayer", "tbl_ts", autolayer.fbl_ts)
-    register_s3_method("ggplot2", "fortify", "tbl_ts", fortify.fbl_ts)
-  }
-}
 # nocov end