app.rb

# encoding=utf-8

# Copyright 2013 Sarah Vessels
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

require 'sinatra'
require 'logger'
require 'omniauth'
require 'omniauth-rdio'
require 'json'
require './user.rb'
require 'rdio'

RDIO_API_KEY = ENV['RDIO_API_KEY']
RDIO_API_SHARED_SECRET = ENV['RDIO_API_SHARED_SECRET']

enable :sessions, :logging
set :session_secret, ENV['SESSION_KEY']

before do
  logger.level = Logger::DEBUG
end

use OmniAuth::Builder do
  provider :rdio, RDIO_API_KEY, RDIO_API_SHARED_SECRET
end

def get_client session
  client = Rdio::Client.new(RDIO_API_KEY, RDIO_API_SHARED_SECRET)
  if session[:access_token] && session[:access_secret]
    client.access_token = session[:access_token]
    client.secret = session[:access_secret]
  end
  client
end

def strip_smart_quotes str
  str.gsub(/[‘’]/, "'").gsub(/[”“]/, '"')
end

root_path = File.join(Dir.pwd, 'public')
set :public_dir, root_path
use Rack::Static, urls: ['/css', '/img', '/js'], root: root_path

get '/' do
  if session[:user]
    redirect '/index.html'
  else
    redirect '/index.html/#/login'
  end
end

get '/auth/:name/callback' do
  # See https://github.com/nixme/omniauth-rdio for description of object
  auth = request.env['omniauth.auth']
  access_token = auth['credentials']['token']
  access_secret = auth['credentials']['secret']
  session[:user] = User.new(auth)
  session[:access_token] = access_token
  session[:access_secret] = access_secret
  redirect '/index.html'
end

get '/rdio_artist_search' do
  content_type :json
  client = get_client(session)
  query = strip_smart_quotes(params[:query])
  artists = client.search(query, 'Artist')
  if artists.count > 0
    artist = artists[0]
    {id: artist.key, name: artist.name}.to_json
  else
    {id: nil, error: "Could not find artist '#{query}' on Rdio."}.to_json
  end
end

get '/rdio_track_search' do
  content_type :json
  client = get_client(session)
  query = strip_smart_quotes(params[:query])
  artist_id = params[:artist_id]
  tracks = client.getTracksForArtist(artist: artist_id, query: query)
  if tracks.count > 0
    track = tracks[0]
    {id: track.key}.to_json
  else
    {id: nil, error: "Could not find track '#{query}' on Rdio."}.to_json
  end
end

post '/rdio_playlist_create' do
  content_type :json
  json_params = JSON.parse(request.body.read)
  name = json_params['name']
  description = json_params['description']
  tracks = json_params['tracks']
  client = get_client(session)
  playlist = client.createPlaylist(name: name, description: description,
                                   tracks: tracks)
  {name: playlist.name, song_count: playlist.length, image_url: playlist.icon,
   id: playlist.key, url: playlist.shortUrl}.to_json
end