# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from telemetry.page import page as page_module
from telemetry.page import shared_page_state

from page_sets.login_helpers import google_login


class TopPages(page_module.Page):

  def __init__(self,
               url,
               page_set,
               shared_page_state_class,
               name='',
               extra_browser_args=None):
    if name == '':
      name = url
    super(TopPages, self).__init__(
        url=url,
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class GoogleWebSearchPage(TopPages):
  """ Why: top google property; a google tab is often open """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(GoogleWebSearchPage, self).__init__(
        url='https://www.google.com/#hl=en&q=barack+obama',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(GoogleWebSearchPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForElement(text='Next')


class GoogleImageSearchPage(TopPages):
  """ Why: tough image case; top google properties """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(GoogleImageSearchPage, self).__init__(
        'https://www.google.com/search?q=cats&tbm=isch',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(GoogleImageSearchPage, self).RunNavigateSteps(action_runner)


class GmailPage(TopPages):
  """ Why: productivity, top google properties """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(GmailPage, self).__init__(
        url='https://mail.google.com/mail/',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(GmailPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForJavaScriptCondition(
        'window.gmonkey !== undefined &&'
        'document.getElementById("gb") !== null')


class GoogleCalendarPage(TopPages):
  """ Why: productivity, top google properties """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(GoogleCalendarPage, self).__init__(
        url='https://www.google.com/calendar/',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(GoogleCalendarPage, self).RunNavigateSteps(action_runner)
    action_runner.Wait(2)
    action_runner.WaitForElement('div[class~="navForward"]')
    action_runner.ExecuteJavaScript("""
        (function() {
          var elem = document.createElement('meta');
          elem.name='viewport';
          elem.content='initial-scale=1';
          document.body.appendChild(elem);
        })();""")
    action_runner.Wait(1)


class GoogleDocPage(TopPages):
  """ Why: productivity, top google properties; Sample doc in the link """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Docs_(1_open_document_tab)',
               extra_browser_args=None):
    super(GoogleDocPage, self).__init__(
        # pylint: disable=line-too-long
        url=
        'https://docs.google.com/document/d/1X-IKNjtEnx-WW5JIKRLsyhz5sbsat3mfTpAPUSX3_s4/view',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(GoogleDocPage, self).RunNavigateSteps(action_runner)
    action_runner.Wait(2)
    action_runner.WaitForJavaScriptCondition(
        'document.getElementsByClassName("kix-appview-editor").length')


class GooglePlusPage(TopPages):
  """ Why: social; top google property; Public profile; infinite scrolls """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(GooglePlusPage, self).__init__(
        url='https://plus.google.com/110031535020051778989/posts',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(GooglePlusPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForElement(text='Home')


class YoutubePage(TopPages):
  """ Why: #3 (Alexa global) """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(YoutubePage, self).__init__(
        url='http://www.youtube.com',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    google_login.LoginGoogleAccount(action_runner, 'googletest')
    super(YoutubePage, self).RunNavigateSteps(action_runner)
    action_runner.Wait(2)


class BlogspotPage(TopPages):
  """ Why: #11 (Alexa global), google property; some blogger layouts have
  infinite scroll but more interesting """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Blogger',
               extra_browser_args=None):
    super(BlogspotPage, self).__init__(
        url='http://googlewebmastercentral.blogspot.com/',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(BlogspotPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForElement(text='accessibility')


class WordpressPage(TopPages):
  """ Why: #18 (Alexa global), Picked an interesting post """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Wordpress',
               extra_browser_args=None):
    super(WordpressPage, self).__init__(
        # pylint: disable=line-too-long
        url=
        'http://en.blog.wordpress.com/2012/09/04/freshly-pressed-editors-picks-for-august-2012/',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(WordpressPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForElement(
        # pylint: disable=line-too-long
        'a[href="http://en.blog.wordpress.com/2012/08/30/new-themes-able-and-sight/"]'
    )


class FacebookPage(TopPages):
  """ Why: top social,Public profile """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Facebook',
               extra_browser_args=None):
    super(FacebookPage, self).__init__(
        url='https://www.facebook.com/barackobama',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(FacebookPage, self).RunNavigateSteps(action_runner)
    action_runner.WaitForElement(text='Videos')


class LinkedinPage(TopPages):
  """ Why: #12 (Alexa global), Public profile. """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='LinkedIn',
               extra_browser_args=None):
    super(LinkedinPage, self).__init__(
        url='http://www.linkedin.com/in/linustorvalds',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class WikipediaPage(TopPages):
  """ Why: #6 (Alexa) most visited worldwide,Picked an interesting page. """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Wikipedia_(1_tab)',
               extra_browser_args=None):
    super(WikipediaPage, self).__init__(
        url='http://en.wikipedia.org/wiki/Wikipedia',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class TwitterPage(TopPages):
  """ Why: #8 (Alexa global),Picked an interesting page """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Twitter',
               extra_browser_args=None):
    super(TwitterPage, self).__init__(
        url='https://twitter.com/katyperry',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(TwitterPage, self).RunNavigateSteps(action_runner)
    action_runner.Wait(2)


class PinterestPage(TopPages):
  """ Why: #37 (Alexa global) """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Pinterest',
               extra_browser_args=None):
    super(PinterestPage, self).__init__(
        url='http://pinterest.com',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class ESPNPage(TopPages):
  """ Why: #1 sports """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='ESPN',
               extra_browser_args=None):
    super(ESPNPage, self).__init__(
        url='http://espn.go.com',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class WeatherPage(TopPages):
  """ Why: #7 (Alexa news); #27 total time spent, picked interesting page. """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='Weather.com',
               extra_browser_args=None):
    super(WeatherPage, self).__init__(
        url='http://www.weather.com/weather/right-now/Mountain+View+CA+94043',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)


class YahooGamesPage(TopPages):
  """ Why: #1 games according to Alexa (with actual games in it) """

  def __init__(self,
               page_set,
               shared_page_state_class=shared_page_state.SharedPageState,
               name='',
               extra_browser_args=None):
    super(YahooGamesPage, self).__init__(
        url='http://games.yahoo.com',
        page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        name=name,
        extra_browser_args=extra_browser_args)

  def RunNavigateSteps(self, action_runner):
    super(YahooGamesPage, self).RunNavigateSteps(action_runner)
    action_runner.Wait(2)