Anjing Webshells

Website : rimsha.abasa.com
: / var / canvas / app / models /
Filename : google_docs_collaboration.rb
back
# frozen_string_literal: true

#
# Copyright (C) 2011 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas 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 Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
#

class GoogleDocsCollaboration < Collaboration
  GOOGLE_DRIVE_SERVICE = "drive.google.com"

  def style_class
    "google_docs"
  end

  def service_name
    "Google Docs"
  end

  def delete_document
    if document_id && user
      google_drive_for_user.delete_doc(document_id)
    end
  end

  def initialize_document
    if !document_id && user
      name = title
      name = nil if name && name.empty?
      name ||= I18n.t("lib.google_docs.default_document_name", "Instructure Doc")

      result = google_drive_for_user.create_doc(name)
      if result
        self.document_id = result.id
        self.data = result.to_h.to_json

        self.url = result.web_view_link
      end
    end
  end

  def user_can_access_document_type?(user)
    return !!google_user_service(user) if self.user && user

    false
  end

  def authorize_user(user)
    return unless document_id

    service = google_user_service(user)
    service_user_id = service&.service_user_id
    service_user_name = service&.service_user_name
    collaborator = collaborators.where(user_id: user).first

    if collaborator
      if collaborator.authorized_service_user_id != service_user_id
        if collaborator.authorized_service_user_id
          google_drive_for_user.acl_remove(document_id, [service_user_id])
        end

        google_drive_for_user.acl_add(document_id, [service_user_name])
        collaborator.update(authorized_service_user_id: service_user_id)
      end
    else
      # no collaboration for this user, lets create it
      add_users_to_collaborators([user])
    end
  end

  def remove_users_from_document(users_to_remove)
    users_to_remove = users_to_remove.filter_map do |user|
      authorized_service_user_id_for(user)
    end

    google_drive_for_user.acl_remove(document_id, users_to_remove) if document_id
  end

  def add_users_to_document(new_users)
    if document_id
      domain = if context.root_account.feature_enabled?(:google_docs_domain_restriction)
                 context.root_account.settings[:google_docs_domain]
               else
                 nil
               end

      user_ids = new_users.filter_map do |user|
        google_user_service(user)&.service_user_name
      end

      google_drive_for_user.acl_add(document_id, user_ids, domain)
    end
  end

  def self.config
    GoogleDrive::Connection.config
  end

  def authorized_service_user_id_for(user)
    google_user_service(user)&.service_user_id
  end

  private

  def google_user_service(user)
    user.user_services.where(service: "google_drive").first
  end

  def google_drive_for_user
    @google_drive_for_user ||= begin
      refresh_token, access_token = Rails.cache.fetch(["google_drive_tokens", user].cache_key) do
        service = google_user_service(user)
        service && [service.token, service.secret]
      end
      raise GoogleDrive::NoTokenError unless refresh_token && access_token

      GoogleDrive::Connection.new(refresh_token, access_token, ApplicationController.google_drive_timeout)
    end
  end
end