dbviewer 0.3.6 → 0.3.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 151a189372c94a737108969cd768ab9dbc4e16a2abc608f07026d1a6b7ee65a5
-  data.tar.gz: a5fa3a4f4aa8e1acd606f60c2264412a261beec6b6889675ef941d05d1e593cb
+  metadata.gz: 65e50cbafa3a2fc1372ce4150c0de4d54001345af9e7d487b048ba3d59929fd2
+  data.tar.gz: 3e84c14c32a912f5c447dc2d4525231f83cd2bd272f99a3b4e19261b692ee2c3
 SHA512:
-  metadata.gz: e8345197920cf8c6a006d3faecd9a0b9f2a6ef4c972226707203ba43b121e05c3a02195eec4d38a699231ae488741802a9dff5c7317ddc576f4e27381ad897e7
-  data.tar.gz: b9a8ff1804963de25e85f9a288c198ee4aaa23cf25253ff99f0ce84f3e470e2e35a47c304ff24b2cbd962d9c913fb69b08f11d6bbd2d213af52113ce4d048e18
+  metadata.gz: 5d12f7e3bc158c902256ce416e7770d95436deefcb12f7a54673200d06706d57e5368a8ddd2d6ed374c5c64ebe753c82619fa505864b5f002cf805e6d3d11146
+  data.tar.gz: 645d438dd54e867e6f58c2d0e2f1cdc09c49652b97cfd93ce39758440dea8f58e14910a5718d677513762195a5346a225c11fa289597ef6a7b28e4cf75541e2a

data/app/controllers/concerns/dbviewer/database_operations.rb

@@ -65,6 +65,19 @@ module Dbviewer
         empty_tables: tables.select { |t| t[:record_count] == 0 }
       }
 
+      # Calculate total foreign key relationships
+      begin
+        total_relationships = 0
+        tables.each do |table|
+          metadata = fetch_table_metadata(table[:name])
+          total_relationships += metadata[:foreign_keys].size if metadata && metadata[:foreign_keys]
+        end
+        analytics[:total_relationships] = total_relationships
+      rescue => e
+        Rails.logger.error("Error calculating relationship count: #{e.message}")
+        analytics[:total_relationships] = 0
+      end
+
       # Calculate schema size if possible
       begin
         analytics[:schema_size] = calculate_schema_size
@@ -207,6 +220,130 @@ module Dbviewer
       relationships
     end
 
+    # Get mini ERD data for a specific table and its relationships
+    def fetch_mini_erd_for_table(table_name)
+      related_tables = []
+      relationships = []
+
+      # Validate the table exists
+      unless database_manager.tables.include?(table_name)
+        Rails.logger.error("[DBViewer] Table not found for mini ERD: #{table_name}")
+        return {
+          tables: [],
+          relationships: [],
+          error: "Table '#{table_name}' not found in the database"
+        }
+      end
+
+      # Add current table
+      related_tables << { name: table_name }
+
+      Rails.logger.info("[DBViewer] Generating mini ERD for table: #{table_name}")
+
+      # Get foreign keys from this table to others (outgoing relationships)
+      begin
+        metadata = fetch_table_metadata(table_name)
+        Rails.logger.debug("[DBViewer] Table metadata: #{metadata.inspect}")
+
+        if metadata && metadata[:foreign_keys].present?
+          metadata[:foreign_keys].each do |fk|
+            # Ensure all required fields are present
+            next unless fk[:to_table].present? && fk[:column].present?
+
+            # Sanitize table and column names for display
+            from_table = table_name.to_s
+            to_table = fk[:to_table].to_s
+            from_column = fk[:column].to_s
+            to_column = fk[:primary_key].to_s.presence || "id"
+            relationship_name = fk[:name].to_s.presence || "#{from_table}_to_#{to_table}"
+
+            relationship = {
+              from_table: from_table,
+              to_table: to_table,
+              from_column: from_column,
+              to_column: to_column,
+              name: relationship_name,
+              direction: "outgoing"
+            }
+
+            relationships << relationship
+            Rails.logger.debug("[DBViewer] Added outgoing relationship: #{relationship.inspect}")
+
+            # Add the related table if not already included
+            unless related_tables.any? { |t| t[:name] == to_table }
+              related_tables << { name: to_table }
+            end
+          end
+        end
+      rescue => e
+        Rails.logger.error("[DBViewer] Error fetching outgoing relationships for #{table_name}: #{e.message}")
+        Rails.logger.error(e.backtrace.join("\n"))
+      end
+
+      # Get foreign keys from other tables to this one (incoming relationships)
+      begin
+        database_manager.tables.each do |other_table_name|
+          next if other_table_name == table_name # Skip self
+
+          begin
+            other_metadata = fetch_table_metadata(other_table_name)
+            if other_metadata && other_metadata[:foreign_keys].present?
+              other_metadata[:foreign_keys].each do |fk|
+                if fk[:to_table] == table_name
+                  # Ensure all required fields are present
+                  next unless fk[:column].present?
+
+                  # Sanitize table and column names for display
+                  from_table = other_table_name.to_s
+                  to_table = table_name.to_s
+                  from_column = fk[:column].to_s
+                  to_column = fk[:primary_key].to_s.presence || "id"
+                  relationship_name = fk[:name].to_s.presence || "#{from_table}_to_#{to_table}"
+
+                  relationship = {
+                    from_table: from_table,
+                    to_table: to_table,
+                    from_column: from_column,
+                    to_column: to_column,
+                    name: relationship_name,
+                    direction: "incoming"
+                  }
+
+                  relationships << relationship
+                  Rails.logger.debug("[DBViewer] Added incoming relationship: #{relationship.inspect}")
+
+                  # Add the related table if not already included
+                  unless related_tables.any? { |t| t[:name] == from_table }
+                    related_tables << { name: from_table }
+                  end
+                end
+              end
+            end
+          rescue => e
+            Rails.logger.error("[DBViewer] Error processing relationships for table #{other_table_name}: #{e.message}")
+            # Continue to the next table
+          end
+        end
+      rescue => e
+        Rails.logger.error("[DBViewer] Error fetching incoming relationships for #{table_name}: #{e.message}")
+        Rails.logger.error(e.backtrace.join("\n"))
+      end
+
+      # If no relationships were found, make sure to still include at least the current table
+      if relationships.empty?
+        Rails.logger.info("[DBViewer] No relationships found for table: #{table_name}")
+      end
+
+      result = {
+        tables: related_tables,
+        relationships: relationships,
+        timestamp: Time.now.to_i
+      }
+
+      Rails.logger.info("[DBViewer] Mini ERD data generated: #{related_tables.length} tables, #{relationships.length} relationships")
+      result
+    end
+
     # Prepare the SQL query - either from params or default
     def prepare_query
       quoted_table = safe_quote_table_name(@table_name)

data/app/controllers/dbviewer/tables_controller.rb

@@ -27,6 +27,12 @@ module Dbviewer
       @total_pages = calculate_total_pages(@total_count, @per_page)
       @records = fetch_table_records(@table_name)
 
+      # Ensure @records is never nil to prevent template errors
+      if @records.nil?
+        column_names = fetch_table_columns(@table_name).map { |c| c[:name] }
+        @records = ActiveRecord::Result.new(column_names, [])
+      end
+
       # Fetch timestamp visualization data if the table has a created_at column
       if has_timestamp_column?(@table_name)
         @time_grouping = params[:time_group] || "daily"
@@ -46,6 +52,34 @@ module Dbviewer
       end
     end
 
+    def mini_erd
+      @table_name = params[:id]
+
+      begin
+        @erd_data = fetch_mini_erd_for_table(@table_name)
+
+        if @erd_data[:error].present?
+          Rails.logger.error("Mini ERD error: #{@erd_data[:error]}")
+        end
+
+        respond_to do |format|
+          format.json { render json: @erd_data }
+          format.html { render layout: false }
+        end
+      rescue => e
+        Rails.logger.error("Error generating Mini ERD: #{e.message}")
+        Rails.logger.error(e.backtrace.join("\n"))
+
+        @error_message = e.message
+        @erd_data = { tables: [], relationships: [], error: @error_message }
+
+        respond_to do |format|
+          format.json { render json: { error: @error_message }, status: :internal_server_error }
+          format.html { render layout: false }
+        end
+      end
+    end
+
     def query
       @table_name = params[:id]
       @read_only_mode = true # Flag to indicate we're in read-only mode

data/app/views/dbviewer/home/index.html.erb

@@ -15,37 +15,58 @@
 
   <div class="row g-3 mb-4">
     <div class="col-md-3">
-      <div class="card h-100 text-center border-0 <%= stat_card_bg_class %>">
-        <div class="card-body">
-          <h5 class="mb-1">Tables</h5>
-          <h2 class="mb-0"><%= @analytics[:total_tables] %></h2>
+      <div class="card h-100 border-0 shadow-sm <%= stat_card_bg_class %>">
+        <div class="card-body d-flex align-items-center">
+          <div class="metric-icon me-3">
+            <i class="bi bi-table fs-4"></i>
+          </div>
+          <div class="text-start">
+            <h5 class="mb-1">Tables</h5>
+            <h2 class="mb-0"><%= @analytics[:total_tables] %></h2>
+          </div>
         </div>
       </div>
     </div>
     
     <div class="col-md-3">
-      <div class="card h-100 text-center border-0 <%= stat_card_bg_class %>">
-        <div class="card-body">
-          <h5 class="mb-1">Records</h5>
-          <h2 class="mb-0"><%= number_with_delimiter(@analytics[:total_records]) %></h2>
+      <div class="card h-100 border-0 shadow-sm <%= stat_card_bg_class %>">
+        <div class="card-body d-flex align-items-center">
+          <div class="metric-icon me-3">
+            <i class="bi bi-database fs-4"></i>
+          </div>
+          <div class="text-start">
+            <h5 class="mb-1">Records</h5>
+            <h2 class="mb-0"><%= number_with_delimiter(@analytics[:total_records]) %></h2>
+          </div>
         </div>
       </div>
     </div>
     
     <div class="col-md-3">
-      <div class="card h-100 text-center border-0 <%= stat_card_bg_class %>">
-        <div class="card-body">
-          <h5 class="mb-1">Columns</h5>
-          <h2 class="mb-0"><%= @analytics[:total_columns] %></h2>
+      <div class="card h-100 border-0 shadow-sm <%= stat_card_bg_class %>">
+        <div class="card-body d-flex align-items-center">
+          <div class="metric-icon me-3">
+            <i class="bi bi-link-45deg fs-4"></i>
+          </div>
+          <div class="text-start">
+            <h5 class="mb-1">Relationships</h5>
+            <h2 class="mb-0"><%= @analytics[:total_relationships] %></h2>
+            <small class="text-muted d-block">Foreign Key Connections</small>
+          </div>
         </div>
       </div>
     </div>
     
     <div class="col-md-3">
-      <div class="card h-100 text-center border-0 <%= stat_card_bg_class %>">
-        <div class="card-body">
-          <h5 class="mb-1">Database Size</h5>
-          <h2 class="mb-0"><%= number_to_human_size(@analytics[:schema_size]) %></h2>
+      <div class="card h-100 border-0 shadow-sm <%= stat_card_bg_class %>">
+        <div class="card-body d-flex align-items-center">
+          <div class="metric-icon me-3">
+            <i class="bi bi-hdd fs-4"></i>
+          </div>
+          <div class="text-start">
+            <h5 class="mb-1">Database Size</h5>
+            <h2 class="mb-0"><%= number_to_human_size(@analytics[:schema_size]) %></h2>
+          </div>
         </div>
       </div>
     </div>