labelprint/app.rb

# /usr/bin/env ruby
require 'sinatra'
require "sinatra/reloader"
require 'tmpdir'
require 'net/http'
require_relative 'lib/receipt_builder'
require_relative 'lib/label_repository'

set :protection, :except => :frame_options
set :environment, :production
set :bind, "::"
set :port, 80

if !File.exist?('config.json') || !File.readable?('config.json')
  raise RuntimeException, 'Could not read config.json'
end

# Direct libusb bulk transfer (bypassing the kernel usblp driver, matching how
# brother_ql_web addresses its printers) times out on this Pi 3B's USB
# controller -- back to the kernel driver via a stable udev symlink instead.
PRINTER_DEVICE = '/dev/usb/lp_datamax' # see /etc/udev/rules.d/99-datamax.rules
PRINTER_DPI = 203 # DatamaxONeil E4204B Mark III native head resolution
PRINTER_DARKNESS = 8 # lower than factory default to stop thermal dot-gain from fusing barcode bars
TEST_LABEL_PDF = File.join(File.dirname(__FILE__), 'test-label.pdf')

PRINTER_DEVICE_EPSON = '/dev/usb/lp_epson' # see /etc/udev/rules.d/99-epson.rules

LABEL_REPO = LabelRepository.new(File.join(File.dirname(__FILE__), 'data', 'labels'))

BROTHER_PRINTERS = {
  'ql560'  => { port: 8560, label: 'Brother QL-560' },
  'ql700'  => { port: 8700, label: 'Brother QL-700' },
  'ql1100' => { port: 9100, label: 'Brother QL-1100' }
}.freeze

configure do
  enable :static
  set :session_secret, File.read(File.join(File.dirname(__FILE__), '.session_secret')).strip
  enable :sessions

  set :views, File.join(File.dirname(__FILE__), 'views')
  set :public_folder, File.join(File.dirname(__FILE__), 'public')
  set :files, File.join(settings.public_folder, 'files')
  set :unallowed_paths, ['.', '..']
end

configure :production do
  enable :reloader
end

helpers do
  def flash(message = '', level)
    case level
    when "success"
      session[:alert_success] = message
    when "danger"
      session[:alert_danger]  = message
    when "warning"
      session[:alert_warning] = message
    end
  end

  # Renders a PDF to ZPL ^GFA graphic labels at the printer's native DPI,
  # bypassing CUPS/Gutenprint's raster pipeline which blurs text and 1px-shifts
  # output compared to native ZPL rendering.
  def pdf_to_zpl(pdf_path, dpi: PRINTER_DPI)
    Dir.mktmpdir do |dir|
      pattern = File.join(dir, 'page-%03d.pbm')
      ok = system('gs', '-q', '-dNOPAUSE', '-dBATCH', '-dSAFER',
                  '-sDEVICE=pbmraw', "-r#{dpi}",
                  '-dTextAlphaBits=4', '-dGraphicsAlphaBits=4',
                  "-sOutputFile=#{pattern}", pdf_path)
      raise 'ghostscript rendering failed' unless ok

      Dir.glob(File.join(dir, 'page-*.pbm')).sort.map { |f| pbm_to_zpl_label(f) }.join
    end
  end

  def pbm_to_zpl_label(pbm_path)
    bytes = File.binread(pbm_path)
    header = /\AP4\s+(?:#[^\n]*\n\s*)*(\d+)\s+(\d+)\s/.match(bytes)
    raise "unreadable PBM header in #{pbm_path}" unless header

    width, height = header[1].to_i, header[2].to_i
    bytes_per_row = (width + 7) / 8
    total_bytes = bytes_per_row * height
    body = bytes.byteslice(header.end(0), total_bytes)
    hexdata = body.unpack1('H*').upcase

    "^XA\n^MD#{PRINTER_DARKNESS}\n^PW#{width}\n^LL#{height}\n^FO0,0^GFA,#{total_bytes},#{total_bytes},#{bytes_per_row},#{hexdata}\n^XZ\n"
  end

  def print_to_datamax(data)
    File.open(PRINTER_DEVICE, 'wb') { |f| f.write(data) }
  end

  # Builds a simple receipt (bold centered title + left-aligned body,
  # optional QR code) so callers don't need to know ESC/POS escape codes.
  def print_receipt(title:, body:, qr: nil)
    receipt = ReceiptBuilder.new
    receipt.align(:center).bold.text("#{title}\n").bold(false) unless title.to_s.empty?
    receipt.align(:left).text("#{body}\n") unless body.to_s.empty?
    unless qr.to_s.empty?
      receipt.align(:center).feed(1).qr(qr).feed(1)
    end
    receipt.feed(3).cut
    receipt.print!(PRINTER_DEVICE_EPSON)
  end

  ESCPOS_MNEMONICS = {
    'ESC' => "\x1B", 'GS' => "\x1D", 'FS' => "\x1C", 'DLE' => "\x10",
    'LF' => "\x0A", 'CR' => "\x0D", 'HT' => "\x09", 'NUL' => "\x00"
  }.freeze

  # Parses a line-based mnemonic ESC/POS notation, e.g.:
  #   ESC a 0x01     # Align: Center
  #   Hello          # literal text
  #   LF
  #   GS k 0x03 12345678 0x00
  # Each `#...` trailing comment is stripped, then each line's
  # whitespace-separated tokens become either a known control mnemonic, a
  # raw `0xNN` byte, or literal text (CP852-encoded) -- consecutive literal
  # tokens are rejoined with a single space so plain prose still reads
  # correctly, while whitespace around control/byte tokens is dropped since
  # those are just for human readability, not printer output.
  def parse_escpos_notation(text)
    bytes = "".b
    text.each_line do |line|
      prev_was_text = false
      line.sub(/#.*/, '').split.each do |token|
        if ESCPOS_MNEMONICS.key?(token)
          bytes << ESCPOS_MNEMONICS[token]
          prev_was_text = false
        elsif token =~ /\A0x([0-9A-Fa-f]{2})\z/
          bytes << [Regexp.last_match(1)].pack('H2')
          prev_was_text = false
        else
          bytes << ' ' if prev_was_text
          bytes << token.encode('CP852', invalid: :replace, undef: :replace, replace: '?')
          prev_was_text = true
        end
      end
    end
    bytes
  end

  # Prints already-formatted ESC/POS notation (see parse_escpos_notation) --
  # for content prepared elsewhere or reprinted from the label repository,
  # unlike print_receipt's title/body template.
  def print_raw_escpos(content)
    receipt = ReceiptBuilder.new
    receipt.align(:left).raw(parse_escpos_notation(content))
    receipt.feed(3).cut
    receipt.print!(PRINTER_DEVICE_EPSON)
  end

  # Renders ZPL to a PNG via Labelary's public API for an inline preview.
  # Fixed at the Datamax label size (4x6in @ 8dpmm ~= 203dpi) -- good enough
  # for a thumbnail, not meant to be exact for other sizes.
  def render_zpl_preview(zpl)
    uri = URI('https://api.labelary.com/v1/printers/8dpmm/labels/4x6/0/')
    res = Net::HTTP.post(uri, zpl, 'Content-Type' => 'application/x-www-form-urlencoded; charset=utf-8')
    return nil unless res.is_a?(Net::HTTPSuccess)

    res.body
  end

  # Transparently reverse-proxies the current request to one of the
  # brother_ql_web instances, so all three printers (web UI and API alike)
  # are reachable through this single sinatra port instead of their own
  # dedicated ports.
  def proxy_to_brother(port, prefix, path)
    uri = URI("http://127.0.0.1:#{port}#{path}")
    uri.query = request.query_string unless request.query_string.empty?

    proxied = Net::HTTP.const_get(request.request_method.capitalize).new(uri)
    request.env.each do |key, value|
      next unless key.start_with?('HTTP_') && key != 'HTTP_HOST'

      header = key.sub(/^HTTP_/, '').split('_').map(&:capitalize).join('-')
      proxied[header] = value
    end
    proxied['Content-Type'] = request.content_type if request.content_type
    proxied['X-Forwarded-Prefix'] = prefix
    proxied['X-Forwarded-Host'] = request.host
    proxied.body = request.body.read unless request.request_method == 'GET'

    res = Net::HTTP.start(uri.host, uri.port) { |http| http.request(proxied) }

    body = res.body || ''
    if res['Content-Type']&.include?('text/html')
      body = body.sub(/(<body[^>]*>)/i) { "#{Regexp.last_match(1)}\n#{erb(:_navbar, layout: false)}" }
    end

    headers = {}
    res.each_header do |key, value|
      next if %w[transfer-encoding connection content-length].include?(key.downcase)

      headers[key] = value
    end
    [res.code.to_i, headers, [body]]
  end
end

before do
  @alert_success = session.delete(:alert_success)
  @alert_danger  = session.delete(:alert_danger)
  @alert_warning = session.delete(:alert_warning)
end


get '/' do
  erb :index
end

get '/upload' do
  erb :upload
end

post '/shipping/zpl' do
  zpl = params[:zpl].to_s
  name = params[:save_as].to_s
  if zpl.strip.empty?
    flash 'ZPL text is empty', 'warning'
  elsif name.empty?
    flash 'Name is required to save', 'warning'
  else
    LABEL_REPO.save(name: name, type: 'zpl', content: zpl)
    if params[:commit] == 'save_and_print'
      begin
        print_to_datamax(zpl)
        flash "Saved '#{name}' and sent to printer", 'success'
      rescue => e
        flash "Saved '#{name}', but print failed (#{e.message})", 'danger'
      end
    else
      flash "Saved '#{name}'", 'success'
    end
  end
  redirect '/upload'
end

post '/upload' do
    if params[:file]

      tempfile = params[:file][:tempfile]
      filename = params[:file][:filename]
      filetype = File.extname filename
      filepath = File.join(settings.files, filename)

      File.open(filepath, 'wb') do |f|
        f.write tempfile.read
      end

      case filetype
      when ".pdf"
        begin
          zpl = pdf_to_zpl(filepath)
          print_to_datamax(zpl)
          flash "PDF: Upload successful (#{filename})", "success"
        rescue => e
          flash "PDF: Failed to print (#{e.message})", "danger"
        end
      when ".zpl"
        print_to_datamax(File.binread(filepath))
        flash "ZPL: Upload successful (#{filename})", "success"
      else
        flash "#{filetype} is not supported. Only .pdf and .zpl!", "false"
      end

      File.delete(filepath) if File.exist?(filepath)
    else
      flash 'You have to choose a file', "warning"
    end

    redirect '/upload'
end

post '/test-print' do
  begin
    zpl = pdf_to_zpl(TEST_LABEL_PDF)
    print_to_datamax(zpl)
    flash 'Test label sent to printer', 'success'
  rescue => e
    flash "Test print failed (#{e.message})", 'danger'
  end
  redirect '/upload'
end

get '/epson' do
  erb :epson
end

post '/epson/print' do
  begin
    print_receipt(title: params[:title].to_s, body: params[:body].to_s, qr: params[:qr].to_s)
    flash 'Sent to Epson printer', 'success'
  rescue => e
    flash "Epson print failed (#{e.message})", 'danger'
  end
  redirect '/epson'
end

post '/epson/raw' do
  content = params[:content].to_s
  name = params[:save_as].to_s
  if content.strip.empty?
    flash 'Receipt text is empty', 'warning'
  elsif name.empty?
    flash 'Name is required to save', 'warning'
  else
    LABEL_REPO.save(name: name, type: 'escpos', content: content)
    if params[:commit] == 'save_and_print'
      begin
        print_raw_escpos(content)
        flash "Saved '#{name}' and sent to printer", 'success'
      rescue => e
        flash "Saved '#{name}', but print failed (#{e.message})", 'danger'
      end
    else
      flash "Saved '#{name}'", 'success'
    end
  end
  redirect '/epson'
end

get '/labels' do
  @labels = LABEL_REPO.all
  erb :labels
end

get '/labels/:id/edit' do
  @label = LABEL_REPO.find(params[:id])
  halt 404, 'unknown label' unless @label

  erb :label_edit
end

post '/labels/:id' do
  label = LABEL_REPO.update(params[:id], name: params[:name].to_s, content: params[:content].to_s)
  halt 404, 'unknown label' unless label

  flash "Saved '#{label['name']}'", 'success'
  redirect '/labels'
end

get '/labels/:id/preview.png' do
  label = LABEL_REPO.find(params[:id])
  halt 404 unless label && label['type'] == 'zpl'

  png = render_zpl_preview(label['content'])
  halt 502, 'preview unavailable' unless png

  content_type 'image/png'
  png
end

post '/labels/:id/print' do
  label = LABEL_REPO.find(params[:id])
  halt 404, 'unknown label' unless label

  begin
    case label['type']
    when 'zpl'
      print_to_datamax(label['content'])
    when 'escpos'
      print_raw_escpos(label['content'])
    end
    flash "Printed '#{label['name']}'", 'success'
  rescue => e
    flash "Print failed (#{e.message})", 'danger'
  end
  redirect '/labels'
end

post '/labels/:id/delete' do
  LABEL_REPO.delete(params[:id])
  flash 'Label deleted', 'success'
  redirect '/labels'
end

get '/brother' do
  @printers = BROTHER_PRINTERS
  erb :brother_picker
end

%w[get post put delete].each do |verb|
  send(verb, '/brother/:printer/*') do
    printer = BROTHER_PRINTERS[params[:printer]]
    halt 404, 'unknown printer' unless printer

    proxy_to_brother(printer[:port], "/brother/#{params[:printer]}", '/' + params[:splat].first)
  end
end