bibdesk.rb

# encoding: UTF-8

# researchr scripts relevant to BibDesk (the right one is executed from the bottom of the file)

$:.push(File.dirname($0))
require 'utility-functions'
require 'appscript'

#### utilities ####
# adds citation to json file, could be expanded to do other post-import cleanup tasks
def post_import
  cit = pbpaste
  bib = try { cit.scan(/\@(.+?)\{(.+?)\,/)[0][1] }
  fail "Could not add citation to json, citekey not found in BibTeX" unless bib

  citekey = bib

  add_to_jsonbib(citekey)
end

#### keyboard commands ####

# email selected files to Kindle, or copy through USB cable - needs some polish
# launched by ctrl+alt+cmd+K
def send_to_kindle
  pluggedin = File.exists?('/Volumes/Kindle')
  verb = pluggedin ? ["transferring", "transferred"] : ["sending", "sent"]

  require 'mail-lib' unless pluggedin

  growl "Preparing and #{verb[0]} #{Selection.size} documents to Kindle"

  successful = 0
  Selection.each do |dd|

    # prepare document
    docu = dd.cite_key.get
    title = dd.title.get.gsub(/[\{|\}]/,"")
    authors = dd.author.name.get.join(", ")
    `/usr/local/bin/pdftotext "#{PDF_path}/#{docu}.pdf" /tmp/#{docu}.txt`
    `ebook-convert /tmp/#{docu}.txt /tmp/#{docu}.mobi`
    `ebook-meta /tmp/#{docu}.mobi -t "#{title} [#{docu}]" -a "#{authors}" --category="Bibdesk"`

    next unless File.exists?("/tmp/#{docu}.mobi")

    successful += 1

    if pluggedin
      `cp /tmp/#{docu}.mobi /Volumes/Kindle/documents`
    else
      mail_file("/tmp/#{docu}.mobi")
    end
  end
  growl("#{successful} file(s) #{verb[1]} to Kindle")
end

# properly formats an author list for BibDesk with " and " as separator
# launched by ctrl+alt+cmd+P
def authorlist
  a = pbpaste.strip

  # determine whether to split on newline, space or comma
  if a.scan(";").size > 1
    splt = ";"
  elsif a.scan(".,").size > 2 # takes care of initials, like Haklev, S. H., Peter, J. H.
    splt = ".,"
  elsif a.scan(",").size > 2
    splt = ","
  end

  a = a.split(splt)

  a = a.join("||").gsubs(
    [/\.([^ ])/, '. \1'],       # fixing space between initials, Stian H.Aklev -> Stian H. Aklev
    [/ ([^ ])\|\|/, ' \1.||'],  # putting the final dot back if removed above, maybe not most elegant way
    [" and ",""],               # we've already split, hopefully correctly
    ["&",""],                   # see above
    ["||", " and "],            # key transformation, put them back together with and (hoping that the split worked)
    [/ +/," "],                 # pruning spaces
    [/\(.+?\)/, ''],            # removing affiliations Stian Haklev (University of Toronto)
    [/\d/, '']                  # remove all numbers
    )
  pbcopy(a)
end

# put on the clipboard a properly formatted list of references to the currently selected publications
# launched by ctrl+alt+cmd+C
def copy
  out = ''

  Selection.each do |dd|
    docu = dd.cite_key.get
    docu.strip!
    out << "[@#{docu}] "
  end

  pbcopy(out.strip)
  growl("#{Selection.size} citation references copied to the clipboard")
end

# attaches the last added PDF in the download directory to the currently selected Bibdesk reference
# launched by ctrl+alt+cmd+L
def linkfile
  curfile =  File.last_added("#{Downloads_path}/*.pdf")
  fail "Sorry, no PDFs found in that directory" unless curfile # no last file found

  # grab download URL before file gets moved
  a = `mdls -name kMDItemWhereFroms "#{curfile}"`
  dlurl = try {a.split('"')[1]}


  # import and autolink file
  f = MacTypes::FileURL.path(curfile)
  Selection[0].linked_files.add(f,{:to =>Selection[0]})
  Selection[0].auto_file

  growl("PDF added", "File added successfully to #{Selection[0].cite_key.get}")

  # check if OA, and add URL field if yes
  if dlurl.index('http') && check_oa(dlurl)
    Selection[0].fields["Url"].value.set(dlurl)
    Selection[0].fields["OA-URL"].value.set(dlurl)
    growl "Publication is OA, URL successfully added"
  end

end


# if cannot manage to overwrite a publication, grab PDF, delete publication, create new publication from
# bibtex import, attach PDF, and autofile
def get_bibtexnet
  require 'open-uri'

  growl "Bibtexnet", "Trying to acquire metadata for #{Selection.size} publication(s)"

  c = 0

  Selection.each do |item|
    file = try {item.linked_files.get[0].to_s}

    next unless file && File.exists?(file)

    hash = hashsum(file)
    bibtex = open("http://localhost/bibtexnet/#{hash}").read

  # ensure proper reply
    next unless bibtex.index("BIBTEX<<<")

    btxstring = bibtex.match(/BIBTEX\<\<\<(.+?)\>\>\>/m)[1]
    newpub = try {BibDesk.documents[0].import(BibDesk.documents[0], {:from => btxstring})[0]}

    f = MacTypes::FileURL.path(file)
    newpub.linked_files.add(f,{:to =>newpub})
    newpub.auto_file

    item.remove
    c += 1
  end

  growl "Successfully imported metadata for #{c} publications from bibtexnet"
end

# uses quicklook to preview the PDF of the currently selected publication (or upcoming publication to be added)
# launched by cmd+space or ctrl+alt+cmd+space
def qlook(argv="current")
  if argv == "upcoming"
    file = File.last_added("#{Downloads_path}/*.pdf")
  else
    file =  Selection[0].linked_files.get[0].to_s
  end
  if File.exists?(file)
    `qlmanage -p '#{file}'`
  else
    growl('No file available', "No file available for #{Selection[0].cite_key.get}")
  end
end

# opens a given reference as passed by bibdesk:// URL in BibDesk
# launched by bibdesk:// url in Chrome (bibdesk.app must be registered first)
def url(argv)
  arg = argv[11..-1]
  find = BibDesk.search({:for => arg})
  unless find == []
    BibDesk.activate
    find[0].select  # does not work if pub is not listed (because of a search/keyword filter etc) not sure how to "reset"
    find[0].show
  else
    growl("File not found", "Cannot find citation #{ARGV[0]}")
  end
end

# makes sure that all selected citations have corresponding wiki pages, and opens the first selected citation in Chrome
# launched by ctrl+alt+cmd+E
def open_page
  require 'wiki-lib'
  ary = Array.new
  Selection.each do |dd|
    docu = dd.cite_key.get
    ary << docu unless File.exists?("#{Wiki_path}/data/pages/ref/#{docu}.txt")
    ensure_refpage(docu)
  end
  `open http://localhost/wiki/ref:#{Selection[0].cite_key.get}`
end

#### Running the right function, depending on command line input ####
BibDesk = Appscript.app('BibDesk')
Selection = BibDesk.document.selection.get[0]
send *ARGV