added browse_db_model function

DESCRIPTION

@@ -16,8 +16,9 @@ License: MIT + file LICENSE
 Suggests:
  yaml,
  DiagrammeR,
+ DiagrammeRsvg,
  testthat (>= 2.1.0),
  V8,
  rsvg
 LazyData: TRUE
-RoxygenNote: 6.0.1
+RoxygenNote: 6.1.1

NAMESPACE

@@ -5,6 +5,7 @@ S3method(as.data_model,list)
 S3method(print,data_model)
 S3method(print,datamodelr_graph)
 export(as.data_model)
+export(browse_db_model)
 export(dm_add_colors)
 export(dm_add_reference_)
 export(dm_add_references)

R/browse_db_model.R

@@ -0,0 +1,41 @@
+#' Make a database diagram using a connection and display to browser
+#'
+#' @param connection A database connection.
+#' @param dbType One of 'sqlserver', or 'postgres'.
+#' @param rankdir Direction of diagram default is 'BT'.
+#' @param view_type One of 'all', 'keys_only', or 'view_only'.
+#' @param focus A character vector of tables to be filtered to.
+#' @param wd The working directory to use.
+#' @param fileName The fileName.
+#' @keywords database diagram
+#' @export
+#' @examples
+#' browse_db_model()
+
+browse_db_model <-
+ function(connection = NULL,
+ dbType = 'sqlserver',
+ rankdir = 'BT',
+ view_type = 'keys_only',
+ focus = NULL,
+ wd = getwd(),
+ fileName = 'dbDiagram.svg') {
+ query <- datamodelr::dm_re_query(dbType)
+ db <- odbc::dbGetQuery(connection, query)
+
+ # convert to a data model
+ db <- datamodelr::as.data_model(db)
+
+ if (!is.null(focus)) {
+ focus <- list(tables = focus)
+ }
+
+ graph <-
+ datamodelr::dm_create_graph(db,
+ rankdir = rankdir,
+ view_type = view_type,
+ focus = focus)
+
+ rsvg::rsvg_svg(charToRaw(DiagrammeRsvg::export_svg(DiagrammeR::grViz(graph$dot_code))), fileName)
+ browseURL(paste0('file://', file.path(wd, fileName)))
+ }