Add a shebang so the script can be started directly and run rubocop -a.

Author: andreinagy

generateEntityDiagram.rb

@@ -1,3 +1,5 @@
+#!/usr/bin/env ruby
+
 require_relative 'rubyResources/logger.rb'
 require_relative 'rubyResources/swiftEntityElement.rb'
 require_relative 'rubyResources/swiftMethod.rb'

htmltemplate/js/diagram.js

@@ -113,7 +113,7 @@ function createDiagram() {
   };
 
   // tested at http://visjs.org/examples/network/physics/physicsConfiguration.html
-  const options = {
+  let options = {
     edges: {
       smooth: false
     },

rubyResources/entity.rb

@@ -1,5 +1,4 @@
 class Entity
-
   @@id = 1
 
   attr_accessor :id
@@ -37,29 +36,18 @@ def to_hash
       'typeString' => @typeString
     }
 
-    if !@properties.empty?
-      hash['properties'] = @properties
-    end
+    hash['properties'] = @properties unless @properties.empty?
 
-    if !@methods.empty?
-      hash['methods'] = @methods
-    end
+    hash['methods'] = @methods unless @methods.empty?
 
-    if !@protocols.empty?
-      hash['protocols'] = @protocols.map { |protocol|
-        protocol.id
-      }
-    end
+    hash['protocols'] = @protocols.map(&:id) unless @protocols.empty?
 
-    if !@cases.empty?
-      hash['cases'] = @cases
-    end
+    hash['cases'] = @cases unless @cases.empty?
 
-    return hash
+    hash
   end
 
-  def to_json(*args)
-    return JSON.pretty_generate(to_hash)
+  def to_json(*_args)
+    JSON.pretty_generate(to_hash)
   end
-
 end

rubyResources/entityExtension.rb

@@ -1,11 +1,9 @@
 class EntityExtension < Entity
-
   attr_accessor :extendedEntityName
 
   def initialize(inheritedEntities, extendedEntityName, contentsCodeString)
     super(inheritedEntities, 'extension', contentsCodeString)
 
     @extendedEntityName = extendedEntityName
   end
-
 end

rubyResources/entityType.rb

@@ -1,5 +1,4 @@
 class EntityType < Entity
-
   attr_accessor :name
   attr_accessor :startIndex
   attr_accessor :contentsStartIndex
@@ -21,28 +20,19 @@ def initialize(typeString, name, inheritedEntities, contentsCodeString, startInd
     @extensions = []
   end
 
-  def to_json(*args)
+  def to_json(*_args)
     hash = to_hash
 
     hash['name'] = @name
 
-    if @superClass
-      hash['superClass'] = @superClass.id
-    end
+    hash['superClass'] = @superClass.id if @superClass
 
-    if !@containedEntities.empty?
-      hash['containedEntities'] = @containedEntities.map { |containedEntity|
-        containedEntity.id
-      }
+    unless @containedEntities.empty?
+      hash['containedEntities'] = @containedEntities.map(&:id)
     end
 
-    if !@extensions.empty?
-      hash['extensions'] = @extensions.map { |extension|
-        extension.id
-      }
-    end
+    hash['extensions'] = @extensions.map(&:id) unless @extensions.empty?
 
-    return JSON.pretty_generate(hash)
+    JSON.pretty_generate(hash)
   end
-
 end