waxx 0.1.2

data/lib/waxx/patch.rb

@@ -0,0 +1,138 @@
+# Waxx Copyright (c) 2016 ePark labs Inc. & Daniel J. Fitzpatrick <dan@eparklabs.com> All rights reserved.
+# Released under the Apache Version 2 License. See LICENSE.txt.
+
+# Patches to Ruby Classes
+class Date
+  # HTML format for a date
+  def h
+    to_s.h
+  end
+  # Shortcut to format a date in strftime
+  def f(format='%d-%b-%Y')
+    strftime format
+  end
+end
+class FalseClass
+  # HTML format "false"
+  def h
+    "false"
+  end
+  # Format to a selected format (see TrueClass also)
+  #   Format Options:
+  #     yn: No
+  #     tf: False
+  #     icon: glyphicon (Bootstrap)
+  #     num: 0
+  def f(format=:yn)
+    case format.to_sym
+      when :yn
+        "No"
+      when :tf
+        "False"
+      when :icon
+        '<span class="glyphicon glyphicon-minus" aria-hidden="true"></span>'
+      when :num
+        "0"
+      else
+        raise "Unknown format in FalseClass.f: #{format}. Needs to be :yn, :tf, :icon, or :num."
+    end
+  end
+  # Show false as zero
+  def to_i
+    0
+  end
+end
+class Hash
+  # Add an symbol/string indifferent access to a hash
+  def /(k)
+    self[k.to_sym] || self[k.to_s]
+  end
+end
+class NilClass
+  # HTML format
+  def h
+    ""
+  end
+  # Format nil as and empty string (2nd param)
+  # This is mostly for nil values out of the database
+  # that have a format f() called on them.
+  def f(size=2, zero_as="", t=",", d=".")
+    zero_as
+  end
+
+  # Convert a nil to an empty symbol
+  def to_sym
+    "".to_sym
+  end
+end
+class Numeric
+  # HTML format (self -- no escaping needed)
+  def h
+    self
+  end
+  # Format a number
+  #   size: number of decimal places
+  #   zero_as: will display zero as "" (blank), "-" (dash), etc.
+  #   t: thousands seperator
+  #   d: decimal seperator
+  def f(size=2, zero_as="", t=",", d=".")
+    return zero_as if zero?
+    num_parts = to_s.split(".")
+    x = num_parts[0].gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1#{t}")
+    return x if size == 0
+    x << (d + (num_parts[1].to_s + "0000000")[0,size])
+  end
+end
+class String
+  # Escape HTML entities
+  def h
+    gsub(/[&\"<>]/, {'&'=>'&amp;', '"'=>'&quot;', '<'=>'&lt;', '>'=>'&gt;'})
+  end
+  # Convert a string to a number and format 
+  # See Numeric.f()
+  def f(size=2, zero_as="", t=",", d=".")
+    return zero_as if to_f.zero?
+    num_parts = split(".")
+    x = num_parts[0].gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1#{t}")
+    return x if size == 0
+    x << (d + (num_parts[1].to_s + "0000000")[0,size])
+  end
+end
+class Time
+  # HTML format
+  def h
+    to_s.h
+  end
+  # Format time in strftime
+  def f(format='%d-%b-%Y')
+    strftime format
+  end
+end
+class TrueClass
+  def h
+    "true"
+  end
+  # Format to a selected format (see FalseClass also)
+  #   Format Options:
+  #     yn: Yes
+  #     tf: True
+  #     icon: glyphicon (Bootstrap)
+  #     num: 1
+  def f(format=:yn)
+    case format.to_sym
+      when :text
+        "True"
+      when :icon
+        '<span class="glyphicon glyphicon-ok" aria-hidden="true"></span>'
+      when :num
+        "1"
+      else
+        raise "Unknown format in TrueClass.f: #{format}. Needs to be :yn, :tf, :icon, or :num."
+    end
+  end
+
+  # Show true as 1
+  def to_i
+    1
+  end
+end

data/lib/waxx/pdf.rb

@@ -0,0 +1,69 @@
+# Waxx Copyright (c) 2016 ePark labs Inc. & Daniel J. Fitzpatrick <dan@eparklabs.com> All rights reserved.
+# Released under the Apache Version 2 License. See LICENSE.txt.
+
+# require 'prawn' to use PDFs
+
+module Waxx::Pdf
+
+  def new_doc(margin:50, orientation: "portrait", info: {})
+    Prawn::Document.new(:margin=>margin, :orientation=>orientation, :info=>info)
+  end
+
+  def doc_info(
+    x,
+    title: "Untitled",
+    author: nil,
+    subject: "",
+    keywords: "",
+    creator: "WAXX waxx.io",
+    producer: "Prawn",
+    creation_date: Time.new
+  )
+    {
+    :Title => title,
+    :Author => author || "#{x.usr['un']}",
+    :Subject => subject,
+    :Keywords => keywords,
+    :Creator => creator,
+    :Producer => producer,
+    :CreationDate => creation_date
+    }
+  end
+
+  def file_path(x)
+    "#{Waxx::Root}/tmp/#{Time.new.strftime('%Y%m%dT%H%M%S')}-u#{x.usr['id']}.pdf"
+  end
+
+  def get
+    pdf = new_doc
+    pdf.text "Hello from WAXX. You need to implement the get method in your layout. Then call render_file & return_file"
+    render_file(x, pdf)
+    return_file(x)
+  end
+
+  def render_file(x, pdf)
+    pdf.render_file file_path(x)
+  end
+
+  def return_file(x)
+    File.open(file_path(x), "rb"){|f| x << f.read}
+  end
+
+  def show_grid(pdf)
+    existing_color = pdf.stroke_color?
+    pdf.stroke_color Color::RGB.new(230, 230, 255)
+    (0..800).step(10){|a|
+      pdf.stroke_color Color::RGB.new(130, 130, 255) if (a % 100).zero?
+      pdf.line(a,0,a,620).stroke
+      pdf.stroke_color Color::RGB.new(230, 230, 255) if (a % 100).zero?
+    }
+
+    (0..620).step(10){|b|
+      pdf.stroke_color Color::RGB.new(130, 130, 255) if (b % 100).zero?
+      pdf.line(0,b,800,b).stroke
+      pdf.stroke_color Color::RGB.new(230, 230, 255) if (b % 100).zero?
+    }
+    pdf.stroke_color existing_color
+  end
+
+end

data/lib/waxx/pg.rb

@@ -0,0 +1,246 @@
+# Waxx Copyright (c) 2016 ePark labs Inc. & Daniel J. Fitzpatrick <dan@eparklabs.com> All rights reserved.
+# Released under the Apache Version 2 License. See LICENSE.txt.
+
+##
+# The PostgreSQL Object methods 
+module Waxx::Pg
+  extend self
+
+  attr :app
+  attr :db
+  attr :table
+  attr :columns
+  attr :pkey
+  attr :joins
+  attr :orders
+
+  ##
+  # Connect to a postgresql database
+  #
+  # Set in config.yaml: 
+  #   databases:
+  #     app: postgresql://user:pass@host:port/database
+  def connect(str)
+    conn = PG.connect( str )
+    conn.type_map_for_results = PG::BasicTypeMapForResults.new conn
+    conn.type_map_for_queries = PG::BasicTypeMapForQueries.new conn
+    conn
+  end
+
+  def init(app:nil, db:"app", table:nil, pk:"id", cols:nil)
+    @app ||= (app || App.table_from_class(name)).to_sym
+    @db ||= db.to_sym
+    @table ||= (table || App.table_from_class(name)).to_sym
+    @pkey ||= pk.to_sym
+    @columns ||= {}
+    @joins ||= {}
+    @orders ||= {}
+    has(cols) if cols
+  end
+
+  def has(opts=nil)
+    init if @table.nil?
+    return @columns if opts.nil?
+    @columns = opts
+    @columns.each{|n,v|
+      v[:table] = @table
+      v[:column] = n
+      v[:views] = []
+      v[:label] ||= Waxx::Util.label(n)
+      @orders[n] = v[:order] || n
+      @orders["_#{n}".to_sym] = v[:_order] || "#{n} DESC"
+      @pkey = n if v[:pkey]
+      build_joins(n, v[:is])
+    }
+  end
+
+  def build_joins(n, col_is)
+    return if col_is.nil?
+    [col_is].flatten.each{|str| 
+      r, tc = str.split(":")
+      t, c = tc.split(".")
+      j = c =~ /\+$/ ? "LEFT" : "INNER"
+      @joins[r] = {table: @table, col: n.to_s.strip, join: j.to_s.strip, foreign_table: t.to_s.strip, foreign_col: c.to_s.strip.sub('+','')}
+    }
+  end
+
+  def [](n)
+    @columns[n.to_sym]
+  end
+
+  def /(n)
+    @columns[n.to_sym]
+  end
+
+  def get_cols(*args)
+    re = {}
+    args.flatten.map{|a| re[a] = @columns[a.to_sym]}
+    re
+  end
+
+  def runs(opts=nil)
+    init if @app.nil?
+    return App[@app] if opts.nil?
+    App[@app] = opts
+  end
+
+  def run(x, act, meth, *args)
+    App[@app][act.to_sym][meth.to_sym][x, *args]
+  end
+
+  def parse_select(select, view)
+    raise "Can not define both select and view in Waxx::Object.parse_select (#{name})." if select and view
+    return select || "*" if view.nil?
+    view.columns.map{|n,c|             
+      raise "Column #{n} not defined in #{view}" if c.nil?
+      if c[:sql_select]
+         "#{c[:sql_select]} AS #{n}"
+      elsif n != c[:column]
+        "#{c[:table]}.#{c[:column]} AS #{n}"
+      else 
+        "#{c[:table]}.#{c[:column]}"
+      end
+    }.join(", ")
+  end
+
+  def get(x, select:nil, id:nil, joins:nil,  where:nil, having:nil, order:nil, limit:nil, offset:nil, view:nil, &blk)
+    Waxx.debug "object.get"
+    select = parse_select(select, view)
+    where = ["#{@table}.#{@pkey} = $1",id] if id and where.nil?
+    # Block SQL injection in order clause. All order options must be defined in @orders.
+    if order
+      # Look in self orders
+      if not @orders/order
+        # Look in the view's columns
+        if view and view.orders/order
+          order = view.orders/order
+        else
+          Waxx.debug("ERROR: Object.get order (#{order}) not found in @orders [#{@orders.keys.join(", ")}]. Sorting by #{@pkey} instead.")
+          order = @pkey
+        end
+      else
+        order = @orders/order 
+      end
+    end
+    if joins.nil? and view
+      joins = view.joins_to_sql
+    end
+    q = {select:select, joins:joins, where:where, having:having, order:order, limit:limit, offset:offset}
+    yield q if block_given?
+    Waxx.debug "object.get.select: #{q[:select]}"
+    return [] if q[:select].empty?
+    sql = []
+    sql << "SELECT #{q[:select] || "*"}"
+    sql << "FROM #{@table} #{q[:joins]}"
+    sql << "WHERE #{q[:where][0]}" if q[:where] 
+    sql << "HAVING #{q[:having[0]]}" if q[:having] 
+    sql << "ORDER BY #{q[:order]}" if q[:order]
+    sql << "LIMIT #{q[:limit].to_i}" if q[:limit]
+    sql << "OFFSET #{q[:offset].to_i}" if q[:offset]
+    vals = []
+    vals << q[:where][1] if q[:where] and q[:where][1]
+    vals << q[:having][1] if q[:having] and q[:having][1]
+    #[sql.join(" "), vals.flatten]
+    Waxx.debug sql
+    Waxx.debug vals.join(", ")
+    x.db[@db].exec(sql.join(" "), vals.flatten)
+  end
+
+  def get_by_id(x, id, select=nil, view:nil)
+    get(x, id: id, select: select, view: view).first
+  end
+  alias by_id get_by_id
+
+  def post(x, data, cols:nil, returning:nil, view:nil, &blk)
+    if view
+      cols = view.columns.select{|n,c| c[:table] == @table}
+    else
+      cols ||= @columns
+    end
+    data = blk.call if block_given?
+    sql = "INSERT INTO #{@table} ("
+    names = []
+    vars = []
+    vals = []
+    ret = []
+    i = 1
+    cols.each{|n,v|
+      if data/n
+        names << n.to_s
+        vars << "$#{i}"
+        vals << cast(v, data/n)
+        i += 1
+      end
+      ret << n.to_s
+    }
+    sql << names.join(",")
+    sql << ") VALUES (#{vars.join(",")})"
+    sql << " RETURNING #{returning || ret.join(",")}"
+    Waxx.debug(sql)
+    Waxx.debug(vals)
+    x.db[@db].exec(sql, vals).first 
+  end
+
+  def put(x, id, data, cols:nil, returning:nil, view:nil, where:nil, &blk)
+    if view
+      cols = view.columns.select{|n,c| c[:table] == @table}
+    else
+      cols ||= @columns
+    end
+    data = blk.call if block_given?
+    sql = "UPDATE #{@table} SET "
+    set = []
+    vals = []
+    ret = []
+    i = 1
+    Waxx.debug "data: #{data}"
+    cols.each{|n,v|
+      Waxx.debug "col: #{n}: #{v.inspect}"
+      if data.has_key? n.to_s or data.has_key? n.to_sym
+        set << "#{n} = $#{i}"
+        vals << cast(v, data/n)
+        ret << n.to_s
+        i += 1
+      end
+    }
+    sql << set.join(",")
+    sql << " WHERE #{@pkey} = $#{i} #{where} RETURNING #{returning || ret.join(",")}"
+    vals << id
+    Waxx.debug(sql)
+    Waxx.debug(vals)
+    x.db[@db].exec(sql, vals).first 
+  end
+  alias patch put
+  
+  def put_post(x, id, data, cols:nil, returning:nil, view: nil)
+    q = nil
+    q = get_by_id(x, id, @pkey) if id.to_i > 0
+    return post(x, data, cols: cols, returning: returning, view: view) if q.nil?
+    put(x, id, data, cols: cols, returning: returning, view: view)
+  end
+
+  def cast(col, val)
+    case col[:type].to_sym
+    when :int
+      val.to_s.empty? ? nil : val.to_i
+    when :float, :numeric
+      val.to_s.empty? ? nil : val.to_f
+    else
+      val
+    end
+  end
+
+  def delete(x, id)
+    x.db[@db].exec("DELETE FROM #{@table} WHERE #{@pkey} = $1", [id])
+  end
+
+  def order(req_order, default_order='')
+    return default_order if req_order.nil?
+    return orders[req_order.to_sym] if orders.has_key? req_order.to_sym
+    @pkey
+  end
+
+  def debug(str, level=3)
+    Waxx.debug(str, level)
+  end
+end