brotorift 0.1.0

data/lib/runtime.rb

@@ -0,0 +1,345 @@
+
+class Def
+	attr_reader :ast
+
+	def initialize ast
+		@ast = ast
+	end
+end
+
+
+class TypeDef < Def
+	attr_reader :name
+
+	def initialize ast, name
+		super ast
+		@name = name
+	end
+end
+
+
+class BuiltinTypeDef < TypeDef
+	def initialize name
+		super nil, name
+	end
+end
+
+
+class EnumElementDef < Def
+	attr_reader :name, :value, :doc
+
+	def initialize ast
+		super ast
+		@name = ast.name
+		@value = ast.value
+		@doc = ast.doc
+	end
+end
+
+
+class EnumTypeDef < TypeDef
+	attr_reader :elements, :doc
+
+	def initialize ast
+		super ast, ast.name
+		@doc = ast.doc
+		@elements = {}
+	end
+
+	def add_element element_def
+		@elements[element_def.name] = element_def
+	end
+end
+
+
+class NodeDef < Def
+	attr_reader :name, :language, :nickname, :namespace, :doc
+
+	def initialize ast
+		super ast
+		@name = ast.name
+		@language = ast.language
+		@nickname = ast.nickname
+		@namespace = ast.namespace
+		@doc = ast.doc
+	end
+end
+
+
+class StructTypeDef < TypeDef
+	attr_reader :members, :doc
+
+	def initialize ast
+		super ast, ast.name
+		@members = []
+		@doc = ast.doc
+	end
+
+	def add_member member_def
+		@members.push member_def
+	end
+
+	def get_member name
+		@members.find { |m| m.name == name }
+	end
+end
+
+
+class TypeInstanceDef < Def
+	attr_reader :type, :params, :runtime
+
+	def initialize ast, type, params, runtime
+		super ast
+		@type = type
+		@params = params
+		@runtime = runtime
+	end
+end
+
+
+class MemberDef < Def
+	attr_reader :name, :type, :doc
+
+	def initialize ast, type
+		super ast
+		@name = ast.name
+		@type = type
+		@doc = ast.doc
+	end
+end
+
+
+class DirectionDef < Def
+	attr_reader :client, :direction, :server, :messages, :doc
+
+	def initialize ast, client, server
+		super ast
+		@client = client
+		@direction = ast.direction
+		@server = server
+		@messages = {}
+		@doc = ast.doc
+	end
+
+	def name
+		case @direction
+		when :left
+			return "#{@client.name} <- #{@server.name}"
+		when :right
+			return "#{@client.name} -> #{@server.name}"
+		end
+	end
+
+	def add_message message_def
+		@messages[message_def.name] = message_def
+	end
+end
+
+
+class MessageDef < Def
+	attr_reader :name, :id, :members, :doc
+
+	def initialize ast, id
+		super ast
+		@name = ast.name
+		@id = id
+		@members = []
+		@doc = ast.doc
+	end
+
+	def add_member member_def
+		@members.push member_def
+	end
+
+	def get_member name
+		@members.find { |m| m.name == name }
+	end
+end
+
+
+class SequenceDef < Def
+	attr_reader :name, :steps, :doc
+
+	def initialize ast
+		super ast
+		@name = ast.name
+		@steps = []
+		@doc = ast.doc
+	end
+
+	def add_step step_def
+		@steps.push step_def
+	end
+end
+
+
+class StepDef < Def
+	attr_reader :direction, :message, :doc
+
+	def initialize ast, direction, message
+		super ast
+		@direction = direction
+		@message = message
+		@doc = ast.doc
+	end
+end
+
+
+class Runtime
+	attr_reader :filename
+	attr_reader :builtins
+	attr_reader :nodes
+	attr_reader :enums
+	attr_reader :structs
+	attr_reader :directions
+	attr_reader :sequences
+	attr_reader :includes
+
+	def initialize filename
+		@filename = filename
+		@builtins = {}
+		@enums = {}
+		@nodes = {}
+		@structs = {}
+		@directions = []
+		@sequences = {}
+		@includes = []
+
+		self.init_builtins
+	end
+
+	def init_builtins
+		type_names = [
+			'Bool',
+			'Byte',
+			'Short',
+			'Int',
+			'Long',
+			'UShort',
+			'UInt',
+			'ULong',
+			'Float',
+			'Double',
+			'String',
+			'DateTime',
+			'ByteBuffer',
+			'List',
+			'Set',
+			'Map',
+			'Vector2',
+			'Vector3',
+			'Color',
+		]
+		type_names.each { |t| @builtins[t] = BuiltinTypeDef.new t }
+	end
+
+	def add_enum enum_def
+		@enums[enum_def.name] = enum_def
+	end
+
+	def add_node node_def
+		@nodes[node_def.name] = node_def
+		@nodes[node_def.nickname] = node_def if node_def.name != node_def.nickname
+	end
+
+	def add_struct struct_def
+		@structs[struct_def.name] = struct_def
+	end
+
+	def get_type name
+		type_def = @builtins[name]
+		return type_def, self if type_def != nil
+
+		type_def = @enums[name]
+		return type_def, self if type_def != nil
+
+		type_def = @structs[name]
+		return type_def, self if type_def != nil
+
+		@includes.each do |i|
+			type_def = i.get_type name
+			return type_def, i if type_def != nil
+		end
+
+		return nil, self
+	end
+
+	def add_direction direction_def
+		@directions.push direction_def
+	end
+
+	def get_direction client, direction, server
+		dir = @directions.find { |d| d.client == client and d.server == server and d.direction == direction }
+		return dir if dir != nil
+
+		@includes.each do |i|
+			dir = i.get_direction client, direction, server
+			return dir if dir != nil
+		end
+
+		return nil
+	end
+
+	def add_sequence sequence_def
+		@sequences[sequence_def.name] = sequence_def
+	end
+
+	def get_sequence name
+		sequence = @sequences[name]
+		return sequence if sequence != nil
+
+		@includes.each do |i|
+			sequence = i.get_sequence name
+			return sequence if sequence != nil
+		end
+
+		return nil
+	end
+
+	def add_include runtime
+		@includes.push runtime
+	end
+
+	def get_node_directions node, side
+		node_directions = {}
+		@directions.each do |direction|
+			next if side == :client and direction.client != node
+			next if side == :server and direction.server != node
+
+			opposite_node = nil
+			case side
+			when :client
+				opposite_node = direction.server
+			when :server
+				opposite_node = direction.client
+			end
+
+			node_direction = node_directions[opposite_node]
+			if node_direction == nil
+				node_direction = NodeDirection.new node, opposite_node
+				node_directions[opposite_node] = node_direction
+			end
+
+			if side == :client and direction.direction == :left
+				node_direction.in_direction = direction
+			elsif side == :client and direction.direction == :right
+				node_direction.out_direction = direction
+			elsif side == :server and direction.direction == :left
+				node_direction.out_direction = direction
+			elsif side == :server and direction.direction == :right
+				node_direction.in_direction = direction
+			end
+		end
+		node_directions.values
+	end
+end
+
+
+class NodeDirection
+	attr_accessor :local, :remote, :in_direction, :out_direction
+
+	def initialize local, remote
+		@local = local
+		@remote = remote
+	end
+end

data/lib/sequence_diagram_generator.rb

@@ -0,0 +1,47 @@
+require 'net/http'
+require 'uri'
+require 'open-uri'
+
+
+class SequenceDiagramGenerator
+	def self.generate runtime
+		index = 0
+		total = runtime.sequences.length
+		runtime.sequences.values.each do |sequence|
+			index += 1
+			puts "Generating sequence diagram [#{index}/#{total}]: #{sequence.name}.png"
+			generate_sequence sequence
+		end
+	end
+
+	def self.generate_sequence sequence
+		file = "docs/diagrams/#{sequence.name}.png"
+		text = make_code sequence
+		response = Net::HTTP.post_form(URI.parse('http://www.websequencediagrams.com/index.php'), 'style' => 'omegapple', 'message' => text)
+		if response.body =~ /img: "(.+)"/
+			url = "http://www.websequencediagrams.com/#{ $1 }"
+			url.match(/(png)=(.+)$/)
+			open url do |content|
+				File.open file, 'wb' do |file|
+					file.puts content.read
+				end
+			end
+		end
+	end
+
+	def self.make_code sequence
+		str = "title #{sequence.name} #{sequence.doc}\n"
+		sequence.steps.each do |step|
+			direction = step.direction
+			head = ''
+			case direction.direction
+			when :left
+				head = direction.server.name + ' -> ' + direction.client.name
+			when :right
+				head = direction.client.name + ' -> ' + direction.server.name
+			end
+			str += head + ': ' + step.message.name + ' ' + step.doc + "\n"
+		end
+		str
+	end
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: brotorift
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Peter Ren
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-05-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+- !ruby/object:Gem::Dependency
+  name: rltk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Brotorift generator
+email: lcdcdr2004@gmail.com
+executables:
+- brotorift
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/brotorift
+- lib/ast.rb
+- lib/brotorift.rb
+- lib/case_helper.rb
+- lib/compiler.rb
+- lib/compiler_error.rb
+- lib/generators/elixir_server_generator.ex.erb
+- lib/generators/elixir_server_generator.rb
+- lib/generators/scala_server_generator.rb
+- lib/generators/scala_server_generator.scala.erb
+- lib/generators/unity_client_generator.cs.erb
+- lib/generators/unity_client_generator.rb
+- lib/lexer.rb
+- lib/parser.rb
+- lib/runtime.rb
+- lib/sequence_diagram_generator.rb
+homepage: http://rubygems.org/gems/brotorift
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.6
+signing_key: 
+specification_version: 4
+summary: Brotorift
+test_files: []