raka 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 788df7b5953f68a917e57a8508b51aa89e1ca404b91480bcbd132b3cd7666905
-  data.tar.gz: 00123c1ab2294b5e274942e65ec507561834b259b45e4ad636e3ffe4f269e996
+  metadata.gz: f8d1e99a862ad8d34b330ee50d92afdbf3c6aa11c8111621213e5794dd18830c
+  data.tar.gz: ac8a414a0bb4e0830aea306d80901370efdf77a4a9a9ddc159d79869f0a95410
 SHA512:
-  metadata.gz: 68cac774b3aa673896b68201e6e3667075cf9fc53d943c9f48aac130c5bb801903ce15f5e68523d46c53dc98c2d9df1b23210f84a0e7c628653c40861418d2cb
-  data.tar.gz: 3216c9889ca95b84814ca69c85f42c1f972c0e3093baae37338b09c9fce325064df5ed60d2709334e0250e0eabdd357cbf2cf23088c2dadeaed50fe2ea416369
+  metadata.gz: a9163e4fb93266b5ba646fd21f88c75d85a321e20aa0cb6e55ccb331aa41f4722d3695b8af126c77f3857ffd7240966a55e7b4c16c908bd5c8471b30b9da7265
+  data.tar.gz: 4221dc18f6a1b03799d519df61c80828fd9f4b25353a22eee076f97959a25865224145f7403d5f05988394170b0d69be15720b9bcb6df1b7c3039b51b4d19d19

data/VERSION

@@ -1 +1 @@
-0.4.0
+0.5.0

data/lib/raka/lang/duckdb/impl.rb

@@ -17,10 +17,12 @@ end
 # 1. Persistent mode: operations on .db file with CREATE TABLE
 # 2. Ad-hoc mode: parquet in/out using COPY operations
 class Duckdb
-  def initialize(database: nil, params: {})
+  def initialize(database: nil, params: {}, before: nil, after: nil)
     @params = params
     @database = database
     @mode = @database ? :persistent : :adhoc
+    @before = before
+    @after = after
   end
 
   def duckdb_cmd
@@ -32,26 +34,50 @@ class Duckdb
     end
   end
 
-  def build(code, _task)
-    # Replace parameter placeholders
+  def process_params(code)
+    return code if code.nil?
+    
     processed_code = code
     (@params || {}).each do |key, value|
       processed_code = processed_code.gsub("$#{key}", "'#{value}'")
     end
+    processed_code
+  end
+
+  def build(code, _task)
+    # Process parameter placeholders for all parts
+    main_sql = process_params(code)
+    before_sql = process_params(@before)
+    after_sql = process_params(@after)
 
+    # Build SQL parts as separate statements
+    sql_parts = []
+    
+    # Add before hook if present
+    sql_parts << before_sql if before_sql
+    
+    # Add main query based on mode
     case @mode
     when :persistent
-      "DROP TABLE IF EXISTS :_name_; CREATE TABLE :_name_ AS (#{processed_code});"
+      sql_parts << "DROP TABLE IF EXISTS :_name_;"
+      sql_parts << "CREATE TABLE :_name_ AS (#{main_sql});"
     when :adhoc
-      "COPY (#{processed_code}) TO ':output:' (FORMAT PARQUET);"
+      sql_parts << "COPY (#{main_sql}) TO ':output:' (FORMAT PARQUET);"
     end
+    
+    # Add after hook if present
+    sql_parts << after_sql if after_sql
+    
+    sql_parts.join("\n")
   end
 
   def run_script(env, fname, task)
     case @mode
     when :persistent
+      # Split the SQL into separate statements and execute them individually
       bash env, %(
-      #{duckdb_cmd} -c "$(cat #{fname} | sed 's|:_name_|#{task.output_stem}|g')" | tee #{fname}.log
+      # Execute the combined SQL script with proper variable replacement
+      cat #{fname} | sed 's|:_name_|#{task.output_stem}|g' | #{duckdb_cmd} | tee #{fname}.log
       echo "#{@database}" > #{task.name}
       )
     when :adhoc

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: raka
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - yarray
@@ -123,7 +123,6 @@ files:
 - bin/raka
 - lib/raka.rb
 - lib/raka/compile.rb
-- lib/raka/interface.rbs
 - lib/raka/lang/duckdb/impl.rb
 - lib/raka/lang/psql/impl.rb
 - lib/raka/lang/python/impl.rb

data/lib/raka/interface.rbs

@@ -1,23 +0,0 @@
-class RakaTask
-  attr_reader name: String
-  attr_reader stem: String
-  attr_reader func: String?
-  attr_reader input_stem: String?
-  attr_reader scope: String?
-  attr_reader target_scope: String?
-  attr_reader scopes: Array[String]
-  attr_reader target_scope_captures: Array[String]
-  attr_reader captures: Hash[String, String]
-  attr_reader deps: Array[String]
-  attr_reader deps_str: String
-  attr_reader input: String
-  attr_reader task: Object # RakeTask
-end
-
-class RakaEnv
-end
-
-class LanguageImpl
-  def build: (String code, RakaTask task) -> String
-  def run_script: (RakaEnv env, String fname, RakaTask task) -> nil
-end