quickpress 0.0.1

data/lib/quickpress.rb

@@ -0,0 +1,604 @@
+require 'fileutils'
+require 'yaml'
+require 'open-uri'
+require 'tilt'
+require 'quickpress/version'
+require 'quickpress/wordpress'
+require 'quickpress/cli'
+require 'quickpress/options'
+
+class String
+  # Removes starting whitespace.
+  def remove_starting
+    dup.remove_starting!
+  end
+
+  # Removes starting whitespace (destructive).
+  def remove_starting!
+    self.gsub(/^ +/, "")
+  end
+end
+
+# Controls all operations we can make.
+#
+# This module's a mess.
+#
+# The code sucks, things are not splitted well...
+# There should be delegation up the ass but I couldn't stop
+# coding.
+#
+# Gotta study ruby guidelines and see other librarys' codes
+# to see where I can improve.
+#
+module Quickpress
+
+  # Main directory where we store everything.
+  ROOT_DIR    = File.expand_path "~/.config/quickpress"
+  CONFIG_FILE = "#{ROOT_DIR}/config.yml"
+
+  @@inited = nil
+  @@ran_first_time = false
+
+  # URL of the default site used to post.
+  @@default_site = nil
+
+  # Nice name of the default site.
+  @@default_sitename = nil
+
+  @@username   = nil
+  @@password   = nil
+  @@connection = nil
+
+  # Supported templating languages and their file extensions.
+  @@supported_markup = [["markdown"     , '.md'],
+                        ["asciidoc"     , '.adoc'],
+                        ["erb"          , '.erb'],
+                        ["string"       , '.str'],
+                        ["erubis"       , '.erubis'],
+                        ["haml"         , '.haml'],
+                        ["sass"         , '.sass'],
+                        ["scss"         , '.scss'],
+                        ["less"         , '.less'],
+                        ["builder"      , '.builder'],
+                        ["liquid"       , '.liquid'],
+                        ["markdown"     , '.md'],
+                        ["textile"      , '.textile'],
+                        ["rdoc"         , '.rdoc'],
+                        ["radius"       , '.radius'],
+                        ["markaby"      , '.mab'],
+                        ["nokogiri"     , '.nokogiri'],
+                        ["coffeescript" , '.coffee'],
+                        ["creole"       , '.creole'],
+                        ["mediawiki"    , '.mw'],
+                        ["yajl"         , '.yajl'],
+                        ["csv"          , '.rcsv']]
+
+  module_function
+
+  # Loads default site from configuration file
+  def config_init
+
+    # Reading config file if exists
+    if File.exists? CONFIG_FILE
+      CLI::with_status("Initializing...") do
+
+        raw = File.read CONFIG_FILE
+        settings = YAML.load raw
+        settings = {} if not settings
+
+        @@default_site = settings["default_site"]
+        @@default_sitename = @@default_site.gsub(/htt(p|ps):\/\//, "").gsub(/\//, '-')
+      end
+      @@inited = true
+    end
+
+    FileUtils.mkdir_p ROOT_DIR if not File.exists? ROOT_DIR
+  end
+
+  # Executes at the first time, when there's no configuration
+  # directories.
+  #
+  # Asks stuff.
+  #
+  def first_time
+    puts <<-END.remove_starting!
+      Hello!
+      It looks like this is the first time you're
+      running quickpress.
+
+      Let's connect to your Wordpress(.com/.org) site.
+    END
+    puts
+
+    Quickpress::new_site(nil)
+    @@ran_first_time = true
+  end
+
+  # Adds site with URL `addr` to quickpress.
+  # If it's `nil`, will prompt the user for it.
+  def new_site(addr=nil)
+    return if @@ran_first_time
+    address = nil
+
+    # If retrying, go back here.
+    begin
+      address = addr.dup if not addr.nil? # cannot .dup NilClass
+      address ||= CLI::get("Address:")
+
+      address.gsub!(/http:\/\//, "")
+      address.gsub!(/www\./, "")
+      address.gsub!(/\/$/, "")
+
+      # Checking if site already exists
+      if File.exists? CONFIG_FILE
+        raw = File.read CONFIG_FILE
+
+        settings = {}
+        settings.merge!(YAML.load(raw))
+
+        settings["sites"].each do |s|
+          if address == s
+            puts
+            puts "There's already a site with address '#{address}'"
+            puts "Check it with `qp list-sites`."
+
+            if @@default_site == s
+              puts
+              puts "It's your default site, by the way"
+            end
+            exit 666
+          end
+        end
+      end
+
+      Quickpress::authenticate
+
+      # Will try to connect here.
+      # Might take a while.
+      CLI::with_status("Connecting...") do
+        @@connection = Wordpress.new(address, @@username, @@password)
+      end
+
+      puts <<-END.remove_starting!
+
+        Title:    #{@@connection.title}
+        Tagline:  #{@@connection.tagline}
+        Url:      #{@@connection.url}
+      END
+
+      answer = CLI::ask "Is that right?"
+      fail "will retry" if not answer
+
+      # Last site added becomes the default
+      @@default_site = address
+
+      # Hey, this is our first site!
+      if not File.exists? CONFIG_FILE
+
+        # For a @@default_site like "http://myblog.com/this/dir"
+        #
+        # The @@default_sitename must be "myblog.com-this-dir"
+        @@default_sitename = address.gsub(/htt(p|ps):\/\//, "").gsub(/\//, '-')
+
+        # Saving to config file
+        settings = {}
+
+        settings["sites"] ||= []
+        settings["sites"] << @@default_site
+
+        settings["default_site"] = @@default_site
+
+        File.write(CONFIG_FILE, YAML.dump(settings))
+
+      # Config file exists
+      else
+
+        raw = File.read CONFIG_FILE
+
+        settings = {}
+        settings.merge!(YAML.load(raw))
+
+        settings["sites"] ||= []
+        settings["sites"] << @@default_site
+
+        settings["default_site"] = @@default_site
+
+        File.write(CONFIG_FILE, YAML.dump(settings))
+      end
+
+    rescue StandardError => e
+      retry if e.message =~ /will retry/
+
+      raise e
+    end
+  end
+
+  # Shows all saved sites.
+  def list_sites
+
+    # Hey, this is our first site!
+    if not File.exists? CONFIG_FILE
+      puts "No sites stored yet!"
+      puts
+      puts "Run `qp new-site` to create your first!"
+
+    else
+      raw = File.read CONFIG_FILE
+
+      settings = {}
+      settings.merge!(YAML.load(raw))
+
+      puts "Sites currently managed by quickpress:"
+      puts
+
+      settings["sites"].each_with_index do |site, i|
+
+        if @@default_site == site
+          puts (" %3d. %s <== default site" % [i, site])
+        else
+          puts (" %3d. %s" % [i, site])
+        end
+      end
+    end
+
+  end
+
+  def forget_site ids
+
+    # Hey, there's no sites added yet!
+    if not File.exists? CONFIG_FILE
+      puts "No sites managed with quickpress yet!"
+      puts "Add them with `qp new-site`"
+      exit 666
+    end
+
+    # Getting all sites from config file
+    raw = File.read CONFIG_FILE
+
+    settings = {}
+    settings.merge!(YAML.load(raw))
+
+    max_id = settings["sites"].size - 1
+    ids_to_delete = []
+
+    # Here we go!
+    ids.split(',').each do |id|
+
+      if not (0..max_id).include? id.to_i
+        puts "Invalid id!"
+        puts "Must be between 0 and #{max_id}."
+        next
+      end
+
+      puts "Will delete the following site:"
+      puts
+      puts settings["sites"][id.to_i]
+
+      if not $options[:force]
+        answer = CLI::ask("Is that right?")
+        if not answer
+          puts "Alright, then!"
+          next
+        end
+      end
+
+      ids_to_delete << id.to_i
+    end
+
+    # Forgetting a lot of sites at once
+    # Note: Is there a better way to do this?
+    #       Once I delete an id, all the others change!
+    #       I can't simply `each do delete` them.
+
+    ids_to_delete.each {|i| settings["sites"][i] = "will_delete" }
+
+    settings["sites"].reject! { |s| s == "will_delete" }
+
+    # Just in case we've just deleted the default site,
+    # let's grab the first one left
+    if not settings["sites"].include? @@default_site
+      if not settings["sites"].empty?
+        settings["default_site"] = settings["sites"].first
+      end
+    end
+
+    File.write(CONFIG_FILE, YAML.dump(settings))
+
+    # Ooh, boy
+    # We've just ran out of sites! Better delete that config file!
+    if settings["sites"].empty?
+      FileUtils.rm_f CONFIG_FILE
+    end
+  end
+
+  def use_site id
+    Quickpress::first_time if @@default_site.nil?
+    return if @@ran_first_time
+
+    # Hey, there's no sites added yet!
+    if not File.exists? CONFIG_FILE
+      puts "No sites managed with quickpress yet!"
+      puts ""
+      exit 666
+    end
+
+    # Getting all sites from config file
+    raw = File.read CONFIG_FILE
+
+    settings = {}
+    settings.merge!(YAML.load(raw))
+
+    max_id = settings["sites"].size - 1
+
+    if not (0..max_id).include? id.to_i
+      puts "Invalid id!"
+      puts "Must be between 0 and #{max_id}."
+      exit 666
+    end
+
+    site = settings["sites"][id]
+
+    puts "Default site: #{site}"
+    settings["default_site"] = site
+    File.write(CONFIG_FILE, YAML.dump(settings))
+  end
+
+  # Entrance for when we're creating a page or a post
+  # (`what` says so).
+  #
+  def new(what, filename=nil)
+    if filename.nil?
+
+      # Get editor to open temporary file
+      editor = ENV["EDITOR"]
+      if editor.nil?
+        editor = get("Which text editor we'll use?")
+      end
+
+      extension = nil
+
+      # No markup passed as argument
+      if $options[:markup].nil?
+        puts "Choose your templating language."
+        puts
+
+        @@supported_markup.each_with_index do |m, i|
+          puts (" %2d. %s (%s)" % [i, m[0], m[1]])
+        end
+        puts
+
+        id = CLI::get("Which one?").to_i
+
+        max_id = @@supported_markup.size - 1
+
+        if not (0..max_id).include? id
+          puts "Invalid id!"
+          puts "Must be between 0 and #{max_id}."
+          exit 666
+        end
+
+        extension = @@supported_markup[id][1]
+
+      # User specified filename to post
+      else
+        markup_id = nil
+        @@supported_markup.each_with_index do |m, i|
+
+          if m[0].casecmp($options[:markup]).zero?
+            markup_id = i
+            break
+          end
+        end
+
+        if markup_id.nil?
+          fail "Unknown markup laguage '#{$options[:markup]}'"
+        end
+
+        extension = @@supported_markup[markup_id][1]
+      end
+
+      # Create draft file
+      tempfile = Tempfile.new ['quickpress', extension]
+      tempfile.write "# Leave this file empty to cancel"
+      tempfile.flush
+
+      # Oh yeah, baby
+      `#{editor} #{tempfile.path}`
+
+      if tempfile.size.zero?
+        puts "Empty file: did nothing"
+        tempfile.close
+        exit 666
+      end
+
+      puts "File: '#{tempfile.path}'" if $options[:debug]
+
+      new_file(what, tempfile.path)
+      tempfile.close
+
+    else
+      # Post file and copy it to posted directory.
+      new_file(what, filename)
+    end
+  end
+
+  # Actually sends post/page `filename` to the blog.
+  def new_file(what, filename)
+    Quickpress::startup
+    html = Tilt.new(filename).render
+
+    # If successful, will store page id and link
+    id, link = nil, nil
+
+    if what == :post
+      title = CLI::get "Post title:"
+
+      puts "Existing blog categories:"
+      @@connection.categories.each { |c| puts "* #{c}" }
+
+      puts "Use a comma-separated list (eg. 'cat1, cat2, cat3')"
+      puts "Tab-completion works."
+      puts "(will create non-existing categories automatically)"
+
+      categories = CLI::tab_complete("Post categories:", @@connection.categories)
+
+      cats = []
+      categories.split(',').each { |c| cats << c.lstrip.strip }
+
+      CLI::with_status("Posting...") do
+
+        id, link = @@connection.post(:post_status  => 'publish',
+                                     :post_date    => Time.now,
+                                     :post_title   => title,
+                                     :post_content => html,
+                                     :terms_names  => {
+                                       :category => cats
+                                     })
+      end
+      puts "Post successful!"
+
+    elsif what == :page
+      title = CLI::get "Page title:"
+
+      CLI::with_status("Creating page...") do
+
+        id, link = @@connection.post(:post_status  => 'publish',
+                                     :post_date    => Time.now,
+                                     :post_title   => title,
+                                     :post_content => html,
+                                     :post_type    => 'page')
+      end
+      puts "Page created!"
+    end
+
+    puts <<-END.remove_starting!
+      id:   #{id}
+      link: #{link}
+    END
+  end
+
+  # Deletes comma-separated list of posts/pages with `ids`.
+  # A single number is ok too.
+  def delete(what, ids)
+    Quickpress::startup
+
+    ids.split(',').each do |id|
+
+      thing = nil
+
+      CLI::with_status("Hold on a sec...") do
+
+        if what == :post
+          thing = @@connection.get_post id.to_i
+        elsif what == :page
+          thing = @@connection.get_page id.to_i
+        end
+
+      end
+
+      if what == :post
+        puts "Will delete the following post:"
+      elsif what == :page
+        puts "Will delete the following page:"
+      end
+
+      puts <<-END.remove_starting!
+
+        ID:      #{thing["post_id"]}
+        Title:   #{thing["post_title"]}
+        Date:    #{thing["post_date"].to_time}
+        Status:  #{thing["post_status"]}
+        URL:     #{thing["link"]}
+
+      END
+
+      if not $options[:force]
+        answer = CLI::ask("Is that right?")
+        if not answer
+          puts "Alright, then!"
+          next
+        end
+      end
+
+      CLI::with_status("Deleting...") do
+        if what == :post
+          @@connection.delete_post id.to_i
+        elsif what == :page
+          @@connection.delete_page id.to_i
+        end
+      end
+    end
+  end
+
+  # Show last `ammount` of posts/pages in reverse order of
+  # publication.
+  #
+  def list(what, ammount)
+    Quickpress::startup
+
+    elements = nil
+    if what == :post
+      CLI::with_status("Retrieving posts...") do
+        elements = @@connection.get_posts ammount
+      end
+
+    elsif what == :page
+      CLI::with_status("Retrieving pages...") do
+        elements = @@connection.get_pages ammount
+      end
+    end
+
+    # Ugly as fuark :(
+    puts "+-----+---------------------------------------+-----------------------+--------+"
+    puts "|   ID|Title                                  |Date                   |Status  |"
+    puts "+-----+---------------------------------------+-----------------------+--------+"
+    elements.each do |post|
+      puts sprintf("|%5d|%-39s|%s|%-8s|", post["post_id"].to_i,
+                   post["post_title"],
+                   post["post_date"].to_time,
+                   post["post_status"])
+    end
+    puts "+-----+---------------------------------------+-----------------------+--------+"
+  end
+
+  def list_markup
+    puts "Name (file extension)"
+    puts
+
+    @@supported_markup.each do |m|
+      puts "* #{m[0]} (#{m[1]})"
+    end
+  end
+
+  private
+  module_function
+
+  # Initializes everything based on the config file or
+  # simply by asking the user.
+  def startup
+    Quickpress::first_time if @@default_site.nil?
+
+    puts "Using site '#{@@default_site}'"
+
+    Quickpress::authenticate
+
+    CLI::with_status("Connecting...") do
+      @@connection ||= Wordpress.new(@@default_site, @@username, @@password)
+    end
+  end
+
+  # Gets username and password.
+  #
+  # First, try getting from environment variables
+  # `QP_USERNAME` and `QP_PASSWORD`.
+  #
+  # If that fails, asks to the user.
+  def authenticate
+    @@username ||= ENV["QP_USERNAME"]
+    @@password ||= ENV["QP_PASSWORD"]
+
+    @@username ||= CLI::get("Username:")
+    @@password ||= CLI::get_secret("Password:")
+  end
+end
+