twowaysql 0.2.0

data/issues/issue-f39b907d01d7fa93df8c7a9de2e1b5e27727ee0a.yaml

@@ -0,0 +1,18 @@
+--- !ditz.rubyforge.org,2008-03-06/issue 
+title: Handle Embedded variable comment at Scanner/Grammer level
+desc: Handle Embedded variable comment at Scanner/Grammer level. Currently it is handled at Node object level, so Scanner does not know the Substitution comment is Bind variable comment or Embedded variable comment. However Node should not be responsible for that, Scanner/Grammer should do.
+type: :task
+component: twowaysql
+release: 
+reporter: takuto <takuto.wada@gmail.com>
+status: :unstarted
+disposition: 
+creation_time: 2008-09-04 19:53:35.026331 Z
+references: []
+
+id: f39b907d01d7fa93df8c7a9de2e1b5e27727ee0a
+log_events: 
+- - 2008-09-04 19:53:37.545312 Z
+  - takuto <takuto.wada@gmail.com>
+  - created
+  - ""

data/issues/issue-f64d73ed4f9854f1ded77e6496dbf59cfb3770a7.yaml

@@ -0,0 +1,18 @@
+--- !ditz.rubyforge.org,2008-03-06/issue 
+title: preserve actual comment content
+desc: preserve actual comment content
+type: :feature
+component: twowaysql
+release: 
+reporter: takuto <takuto.wada@gmail.com>
+status: :unstarted
+disposition: 
+creation_time: 2008-09-03 08:33:12.940300 Z
+references: []
+
+id: f64d73ed4f9854f1ded77e6496dbf59cfb3770a7
+log_events: 
+- - 2008-09-03 08:33:34.448031 Z
+  - takuto <takuto.wada@gmail.com>
+  - created
+  - preserve actual comment content if :preserve_comment is on.

data/issues/project.yaml

@@ -0,0 +1,16 @@
+--- !ditz.rubyforge.org,2008-03-06/project 
+name: twowaysql
+version: "0.4"
+components: 
+- !ditz.rubyforge.org,2008-03-06/component 
+  name: twowaysql
+releases: 
+- !ditz.rubyforge.org,2008-03-06/release 
+  name: 0.2 release
+  status: :unreleased
+  release_time: 
+  log_events: 
+  - - 2008-09-03 08:30:15.096853 Z
+    - takuto <takuto.wada@gmail.com>
+    - created
+    - first published version.

data/lib/twowaysql/node.rb

@@ -0,0 +1,239 @@
+unless String.method_defined?(:start_with?)
+  class String #:nodoc:
+    def start_with?(prefix)
+      prefix = prefix.to_s
+      self[0, prefix.length] == prefix
+    end
+  end
+end
+
+
+module TwoWaySQL
+
+  class Context
+
+    def initialize(data)
+      @data = data
+      @enabled = true
+      @bound_variables = []
+      @sql_fragments = []
+    end
+    attr_reader :bound_variables
+    attr_reader :sql_fragments
+    attr_reader :data
+    
+    def sql(separator="")
+      @sql_fragments.join(separator)
+    end
+
+    def fork_child
+      child = Context.new(@data)
+      child.disable!
+      child
+    end
+
+    def join_child(child_ctx)
+      @sql_fragments.concat(child_ctx.sql_fragments)
+      @bound_variables.concat(child_ctx.bound_variables)
+    end
+
+    def add_sql(sql_fragment)
+      @sql_fragments << sql_fragment
+    end
+
+    def add_value(value)
+      @sql_fragments << '?'
+      @bound_variables << value
+    end
+
+    def add_values(values)
+      @sql_fragments << Array.new(values.size, '?').join(', ')
+      @bound_variables.concat(values)
+    end
+
+    def enabled?
+      @enabled
+    end
+
+    def enable!
+      @enabled = true
+    end
+
+    protected
+    def disable!
+      @enabled = false
+    end
+  end
+
+
+
+  class Node
+    protected
+    def exec_list(nodes, ctx)
+      v = nil
+      nodes.each do |i|
+        v = i.accept(ctx)
+      end
+      v
+    end
+
+    def do_eval(ctx, exp)
+      safe_eval(ctx.data, exp)
+    end
+
+    private
+    def safe_eval(ctx, exp)
+      within_safe_level(4) { eval(exp) }
+    end
+
+    def within_safe_level(level)
+      result = nil
+      Thread.start {
+        $SAFE = level
+        result = yield
+      }.join
+      result
+    end
+  end
+
+
+  class RootNode < Node
+    def initialize(tree)
+      @tree = tree
+    end
+    def accept(ctx)
+      exec_list(@tree, ctx)
+    end
+    def children
+      @tree
+    end
+  end
+
+
+  class IfNode < Node
+    def initialize(cond, tstmt, fstmt)
+      @condition = cond
+      @tstmt = tstmt
+      @fstmt = fstmt
+    end
+    def accept(ctx)
+      if do_eval(ctx, @condition)
+        exec_list(@tstmt, ctx)
+        ctx.enable!
+      elsif @fstmt
+        exec_list(@fstmt, ctx)
+        ctx.enable!
+      end
+    end
+  end
+
+
+  class BeginNode < Node
+    def initialize(tree)
+      @tree = tree
+    end
+    def accept(ctx)
+      child_ctx = ctx.fork_child
+      exec_list(@tree, child_ctx)
+      if child_ctx.enabled?
+        ctx.join_child(child_ctx)
+      end
+    end
+  end
+
+
+  class SubStatementNode < Node
+    def initialize(prefix, tree)
+      @prefix = prefix
+      @tree = tree
+    end
+    def accept(ctx)
+      ctx.add_sql(@prefix) if ctx.enabled?
+      exec_list(@tree, ctx)
+    end
+    def each
+      yield self
+    end
+  end
+
+
+  class SubstitutionNode < Node
+    def initialize(exp)
+      @exp = exp
+    end
+    def accept(ctx)
+      if @exp.start_with?('$')
+        embed_value(ctx)
+      else
+        substitute(ctx)
+      end
+    end
+    def substitute(ctx)
+      ctx.add_value(do_eval(ctx, @exp))
+    end
+    def embed_value(ctx)
+      result = do_eval(ctx, @exp[1..-1])
+      ctx.add_sql(result) unless result.nil?
+    end
+  end
+
+
+  class QuestionNode < Node
+    def initialize(num)
+      @num = num
+    end
+    def accept(ctx)
+      ctx.add_value(ctx.data[@num])
+    end
+  end
+
+
+  class ParenSubstitutionNode < Node
+    def initialize(exp)
+      @exp = exp
+    end
+    def accept(ctx)
+      result = do_eval(ctx, @exp)
+      return if result.nil?
+      if result.respond_to?('to_ary')
+        bind_values(ctx, result.to_ary)
+      else
+        ctx.add_value(result)
+      end
+    end
+    def bind_values(ctx, ary)
+      return if ary.empty?
+      ctx.add_sql("(")
+      ctx.add_values(ary)
+      ctx.add_sql(")")
+    end
+  end
+
+
+  class LiteralNode < Node
+    def initialize(val)
+      @val = val
+    end
+    def accept(ctx)
+      ctx.add_sql(@val)
+    end
+  end
+
+
+  class CommentNode < Node
+    def initialize(val)
+      @val = val
+    end
+    def accept(ctx)
+      # nothing to do
+    end
+  end
+
+
+  class EolNode < Node
+    def accept(ctx)
+      ctx.add_sql("\n")
+    end
+  end
+
+end