Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Branch: master
Fetching contributors…

Cannot retrieve contributors at this time

220 lines (191 sloc) 8.671 kB
module Foundation where
import Prelude
import Yesod
import Yesod.Static
import Yesod.Auth
import GoogleOAuth2 (googleOAuth2, AccessToken (..), forwardR, revokeAccessToken)
import Yesod.Default.Config
import Yesod.Default.Util (addStaticContentExternal)
import Network.HTTP.Conduit (Manager, Response (..))
import qualified Settings
import Settings.Development (development)
import qualified Database.Persist.Store
import Settings.StaticFiles
import Database.Persist.GenericSql
import Settings (widgetFile, Extra (..))
import Model
import Text.Jasmine (minifym)
import Web.ClientSession (getKey)
import Text.Hamlet (hamletFile)
import Data.Text (Text)
import Data.Maybe (fromJust)
import Control.Monad (when, mplus)
import qualified Data.Map as M
import Control.Concurrent.MVar (MVar, takeMVar, putMVar)
import DocumentManager (DocumentManager, startDocumentManager)
import Control.Exception (onException)
import Data.Default (def)
import qualified Google.Oauth2.V2 as Oauth2
import qualified Google.Oauth2.V2.Types as Oauth2T
import Control.Applicative ((<$>))
-- | The site argument for your application. This can be a good place to
-- keep settings and values requiring initialization before your application
-- starts running, such as database connections. Every handler will have
-- access to the data present here.
data App = App
{ settings :: AppConfig DefaultEnv Extra
, getStatic :: Static -- ^ Settings for static file serving.
, connPool :: Database.Persist.Store.PersistConfigPool Settings.PersistConfig -- ^ Database connection pool.
, httpManager :: Manager
, persistConfig :: Settings.PersistConfig
, documentManagers :: MVar (M.Map Text DocumentManager)
}
-- Set up i18n messages. See the message folder.
mkMessage "App" "messages" "en"
-- This is where we define all of the routes in our application. For a full
-- explanation of the syntax, please see:
-- http://www.yesodweb.com/book/handler
--
-- This function does three things:
--
-- * Creates the route datatype AppRoute. Every valid URL in your
-- application can be represented as a value of this type.
-- * Creates the associated type:
-- type instance Route App = AppRoute
-- * Creates the value resourcesApp which contains information on the
-- resources declared below. This is used in Handler.hs by the call to
-- mkYesodDispatch
--
-- What this function does *not* do is create a YesodSite instance for
-- App. Creating that instance requires all of the handler functions
-- for our application to be in scope. However, the handler functions
-- usually require access to the AppRoute datatype. Therefore, we
-- split these actions into two functions and place them in separate files.
mkYesodData "App" $(parseRoutesFile "config/routes")
type Form x = Html -> MForm App App (FormResult x, Widget)
-- Please see the documentation for the Yesod typeclass. There are a number
-- of settings which can be configured by overriding methods here.
instance Yesod App where
approot = ApprootMaster $ appRoot . settings
-- Store session data on the client in encrypted cookies,
-- default session idle timeout is 120 minutes
makeSessionBackend _ = do
key <- getKey "config/client_session_key.aes"
return . Just $ clientSessionBackend key 120
defaultLayout widget = do
master <- getYesod
mmsg <- getMessage
-- We break up the default layout into two components:
-- default-layout is the contents of the body tag, and
-- default-layout-wrapper is the entire page. Since the final
-- value passed to hamletToRepHtml cannot be a widget, this allows
-- you to use normal widget features in default-layout.
mat <- maybeAccessToken
liftIO $ print mat
userIdentifier <- case mat of
Nothing -> return Nothing
Just (AccessToken at) -> do
userinfo <- fmap responseBody $ liftIO $ Oauth2.getUserinfo $ def
{ Oauth2.parameterOauth_token = Just at
, Oauth2.parameterFields = Just "email,name"
}
return $ Oauth2T.userinfoEmail userinfo `mplus` Oauth2T.userinfoName userinfo `mplus` Just "Anonymous user"
pc <- widgetToPageContent $ do
addStylesheetRemote "http://fonts.googleapis.com/css?family=Lekton|Source+Sans+Pro"
addStylesheet $ StaticR bootstrap_css_bootstrap_css
addStylesheet $ StaticR bootstrap_css_bootstrap_responsive_css
addScript $ StaticR components_jquery_jquery_js
addScript $ StaticR bootstrap_js_bootstrap_min_js
addScript $ StaticR components_underscore_underscore_min_js
addScript $ StaticR components_backbone_backbone_min_js
$(widgetFile "default-layout")
hamletToRepHtml $(hamletFile "templates/default-layout-wrapper.hamlet")
-- This is done to provide an optimization for serving static files from
-- a separate domain. Please see the staticRoot setting in Settings.hs
urlRenderOverride y (StaticR s) =
Just $ uncurry (joinPath y (Settings.staticRoot $ settings y)) $ renderRoute s
urlRenderOverride _ _ = Nothing
-- The page to be redirected to when authentication is required.
authRoute _ = Just $ AuthR LoginR
isAuthorized _ _ = return Authorized
-- This function creates static content files in the static folder
-- and names them based on a hash of their content. This allows
-- expiration dates to be set far in the future without worry of
-- users receiving stale content.
addStaticContent = addStaticContentExternal minifym base64md5 Settings.staticDir (StaticR . flip StaticRoute [])
-- Place Javascript at bottom of the body tag so the rest of the page loads first
jsLoader _ = BottomOfBody
-- What messages should be logged. The following includes all messages when
-- in development, and warnings and errors in production.
shouldLog _ _source level =
development || level == LevelWarn || level == LevelError
-- How to run database actions.
instance YesodPersist App where
type YesodPersistBackend App = SqlPersist
runDB f = do
master <- getYesod
Database.Persist.Store.runPool
(persistConfig master)
f
(connPool master)
instance YesodAuth App where
type AuthId App = UserId
-- Where to send a user after successful login
loginDest _ = WelcomeR
-- Where to send a user after logout
logoutDest _ = HomeR
getAuthId (Creds _ ident extra) = runDB $ do
x <- getBy $ UniqueGoogleId ident
case x of
Just (Entity uid _) -> do
when (not $ null extra) $ update uid
[ UserAccessToken =. fromJust (lookup "access_token" extra)
, UserRefreshToken =. fromJust (lookup "refresh_token" extra)
]
return $ Just uid
Nothing -> do
fmap Just $ insert $ User ident
(fromJust $ lookup "access_token" extra)
(fromJust $ lookup "refresh_token" extra)
authPlugins app = [googleOAuth2 oauthSettings scopes]
where
oauthSettings = extraGoogleOAuth2 $ appExtra $ settings app
scopes =
[ "https://www.googleapis.com/auth/drive.file"
, "https://www.googleapis.com/auth/drive.install"
, "https://www.googleapis.com/auth/userinfo.profile"
]
onLogout = do
mAccessToken <- maybeAccessToken
case mAccessToken of
Nothing -> return ()
Just at -> liftIO $ revokeAccessToken at
authHttpManager = httpManager
-- This instance is required to use forms. You can modify renderMessage to
-- achieve customized and internationalized form validation messages.
instance RenderMessage App FormMessage where
renderMessage _ _ = defaultFormMessage
maybeAccessToken :: GHandler sub App (Maybe AccessToken)
maybeAccessToken = fmap (AccessToken . userAccessToken . entityVal) <$> maybeAuth
requireAccessToken :: GHandler sub App AccessToken
requireAccessToken = do
Entity _ user <- requireAuth
return $ AccessToken $ userAccessToken user
-- | Get the 'Extra' value, used to hold data from the settings.yml file.
getExtra :: Handler Extra
getExtra = fmap (appExtra . settings) getYesod
getDocumentManager :: AccessToken
-> Text -- ^ File ID
-> Handler DocumentManager
getDocumentManager at fileId = do
App { documentManagers = mv, httpManager = manager } <- getYesod
liftIO $ do
docManagers <- takeMVar mv
let restore = putMVar mv docManagers
case M.lookup fileId docManagers of
Just mgr -> restore >> return mgr
Nothing -> (do
mgr <- startDocumentManager at fileId manager
let docManagers' = M.insert fileId mgr docManagers
putMVar mv docManagers'
return mgr) `onException` restore
Jump to Line
Something went wrong with that request. Please try again.