mcpeasy 0.1.0 → 0.2.0

data/lib/utilities/gdrive/service.rb

@@ -0,0 +1,293 @@
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "google/apis/drive_v3"
+require "googleauth"
+require "signet/oauth_2/client"
+require "fileutils"
+require "json"
+require "time"
+require_relative "../_google/auth_server"
+require_relative "../../mcpeasy/config"
+
+module Gdrive
+  class Service
+    SCOPES = [
+      "https://www.googleapis.com/auth/calendar.readonly",
+      "https://www.googleapis.com/auth/drive.readonly"
+    ]
+    SCOPE = SCOPES.join(" ")
+
+    # MIME type mappings for Google Workspace documents
+    EXPORT_FORMATS = {
+      "application/vnd.google-apps.document" => {
+        format: "text/markdown",
+        extension: ".md"
+      },
+      "application/vnd.google-apps.spreadsheet" => {
+        format: "text/csv",
+        extension: ".csv"
+      },
+      "application/vnd.google-apps.presentation" => {
+        format: "text/plain",
+        extension: ".txt"
+      },
+      "application/vnd.google-apps.drawing" => {
+        format: "image/png",
+        extension: ".png"
+      }
+    }.freeze
+
+    def initialize(skip_auth: false)
+      ensure_env!
+      @service = Google::Apis::DriveV3::DriveService.new
+      @service.authorization = authorize unless skip_auth
+    end
+
+    def search_files(query, max_results: 10)
+      results = @service.list_files(
+        q: "fullText contains '#{query.gsub("'", "\\'")}' and trashed=false",
+        page_size: max_results,
+        fields: "files(id,name,mimeType,size,modifiedTime,webViewLink)"
+      )
+
+      files = results.files.map do |file|
+        {
+          id: file.id,
+          name: file.name,
+          mime_type: file.mime_type,
+          size: file.size&.to_i,
+          modified_time: file.modified_time,
+          web_view_link: file.web_view_link
+        }
+      end
+
+      {files: files, count: files.length}
+    rescue Google::Apis::Error => e
+      raise "Google Drive API Error: #{e.message}"
+    rescue => e
+      log_error("search_files", e)
+      raise e
+    end
+
+    def get_file_content(file_id)
+      # First get file metadata
+      file = @service.get_file(file_id, fields: "id,name,mimeType,size")
+
+      content = if EXPORT_FORMATS.key?(file.mime_type)
+        # Export Google Workspace document
+        export_format = EXPORT_FORMATS[file.mime_type][:format]
+        @service.export_file(file_id, export_format)
+      else
+        # Download regular file
+        @service.get_file(file_id, download_dest: StringIO.new)
+      end
+
+      {
+        id: file.id,
+        name: file.name,
+        mime_type: file.mime_type,
+        size: file.size&.to_i,
+        content: content.is_a?(StringIO) ? content.string : content
+      }
+    rescue Google::Apis::Error => e
+      raise "Google Drive API Error: #{e.message}"
+    rescue => e
+      log_error("get_file_content", e)
+      raise e
+    end
+
+    def list_files(max_results: 20)
+      results = @service.list_files(
+        q: "trashed=false",
+        page_size: max_results,
+        order_by: "modifiedTime desc",
+        fields: "files(id,name,mimeType,size,modifiedTime,webViewLink)"
+      )
+
+      files = results.files.map do |file|
+        {
+          id: file.id,
+          name: file.name,
+          mime_type: file.mime_type,
+          size: file.size&.to_i,
+          modified_time: file.modified_time,
+          web_view_link: file.web_view_link
+        }
+      end
+
+      {files: files, count: files.length}
+    rescue Google::Apis::Error => e
+      raise "Google Drive API Error: #{e.message}"
+    rescue => e
+      log_error("list_files", e)
+      raise e
+    end
+
+    def test_connection
+      about = @service.get_about(fields: "user,storageQuota")
+      {
+        ok: true,
+        user: about.user.display_name,
+        email: about.user.email_address,
+        storage_used: about.storage_quota&.usage,
+        storage_limit: about.storage_quota&.limit
+      }
+    rescue Google::Apis::Error => e
+      raise "Google Drive API Error: #{e.message}"
+    rescue => e
+      log_error("test_connection", e)
+      raise e
+    end
+
+    def authenticate
+      perform_auth_flow
+      {success: true}
+    rescue => e
+      {success: false, error: e.message}
+    end
+
+    def perform_auth_flow
+      client_id = Mcpeasy::Config.google_client_id
+      client_secret = Mcpeasy::Config.google_client_secret
+
+      unless client_id && client_secret
+        raise "Google credentials not found. Please save your credentials.json file using: mcpz config set_google_credentials <path_to_credentials.json>"
+      end
+
+      # Create credentials using OAuth2 flow with localhost redirect
+      redirect_uri = "http://localhost:8080"
+      client = Signet::OAuth2::Client.new(
+        client_id: client_id,
+        client_secret: client_secret,
+        scope: SCOPE,
+        redirect_uri: redirect_uri,
+        authorization_uri: "https://accounts.google.com/o/oauth2/auth",
+        token_credential_uri: "https://oauth2.googleapis.com/token"
+      )
+
+      # Generate authorization URL
+      url = client.authorization_uri.to_s
+
+      puts "DEBUG: Client ID: #{client_id[0..20]}..."
+      puts "DEBUG: Scope: #{SCOPE}"
+      puts "DEBUG: Redirect URI: #{redirect_uri}"
+      puts
+
+      # Start callback server to capture OAuth code
+      puts "Starting temporary web server to capture OAuth callback..."
+      puts "Opening authorization URL in your default browser..."
+      puts url
+      puts
+
+      # Automatically open URL in default browser on macOS/Unix
+      if system("which open > /dev/null 2>&1")
+        system("open", url)
+      else
+        puts "Could not automatically open browser. Please copy the URL above manually."
+      end
+      puts
+      puts "Waiting for OAuth callback... (will timeout in 60 seconds)"
+
+      # Wait for the authorization code with timeout
+      code = GoogleAuthServer.capture_auth_code
+
+      unless code
+        raise "Failed to receive authorization code. Please try again."
+      end
+
+      puts "✅ Authorization code received!"
+      client.code = code
+      client.fetch_access_token!
+
+      # Save credentials to config
+      credentials_data = {
+        client_id: client.client_id,
+        client_secret: client.client_secret,
+        scope: client.scope,
+        refresh_token: client.refresh_token,
+        access_token: client.access_token,
+        expires_at: client.expires_at
+      }
+
+      Mcpeasy::Config.save_google_token(credentials_data)
+      puts "✅ Authentication successful! Token saved to config"
+
+      client
+    rescue => e
+      log_error("perform_auth_flow", e)
+      raise "Authentication flow failed: #{e.message}"
+    end
+
+    private
+
+    def authorize
+      credentials_data = Mcpeasy::Config.google_token
+      unless credentials_data
+        raise <<~ERROR
+          Google Drive authentication required!
+          Run the auth command first:
+          mcpz gdrive auth
+        ERROR
+      end
+
+      client = Signet::OAuth2::Client.new(
+        client_id: credentials_data.client_id,
+        client_secret: credentials_data.client_secret,
+        scope: credentials_data.scope.respond_to?(:to_a) ? credentials_data.scope.to_a.join(" ") : credentials_data.scope.to_s,
+        refresh_token: credentials_data.refresh_token,
+        access_token: credentials_data.access_token,
+        token_credential_uri: "https://oauth2.googleapis.com/token"
+      )
+
+      # Check if token needs refresh
+      if credentials_data.expires_at
+        expires_at = if credentials_data.expires_at.is_a?(String)
+          Time.parse(credentials_data.expires_at)
+        else
+          Time.at(credentials_data.expires_at)
+        end
+
+        if Time.now >= expires_at
+          client.refresh!
+          # Update saved credentials with new access token
+          updated_data = {
+            client_id: credentials_data.client_id,
+            client_secret: credentials_data.client_secret,
+            scope: credentials_data.scope.respond_to?(:to_a) ? credentials_data.scope.to_a.join(" ") : credentials_data.scope.to_s,
+            refresh_token: credentials_data.refresh_token,
+            access_token: client.access_token,
+            expires_at: client.expires_at
+          }
+          Mcpeasy::Config.save_google_token(updated_data)
+        end
+      end
+
+      client
+    rescue JSON::ParserError
+      raise "Invalid token data. Please re-run: mcpz gdrive auth"
+    rescue => e
+      log_error("authorize", e)
+      raise "Authentication failed: #{e.message}"
+    end
+
+    def ensure_env!
+      unless Mcpeasy::Config.google_client_id && Mcpeasy::Config.google_client_secret
+        raise <<~ERROR
+          Google API credentials not configured!
+          Please save your Google credentials.json file using:
+          mcpz config set_google_credentials <path_to_credentials.json>
+        ERROR
+      end
+    end
+
+    def log_error(method, error)
+      Mcpeasy::Config.ensure_config_dirs
+      File.write(
+        Mcpeasy::Config.log_file_path("gdrive", "error"),
+        "#{Time.now}: #{method} error: #{error.class}: #{error.message}\n#{error.backtrace.join("\n")}\n",
+        mode: "a"
+      )
+    end
+  end
+end

data/lib/utilities/gmeet/cli.rb

@@ -2,156 +2,158 @@
 
 require "bundler/setup"
 require "thor"
-require_relative "gmeet_tool"
+require_relative "service"
 
-class GmeetCLI < Thor
-  desc "test", "Test the Google Calendar API connection"
-  def test
-    response = tool.test_connection
+module Gmeet
+  class CLI < Thor
+    desc "test", "Test the Google Calendar API connection"
+    def test
+      response = tool.test_connection
 
-    if response[:ok]
-      puts "✅ Successfully connected to Google Calendar"
-      puts "   User: #{response[:user]} (#{response[:email]})"
-    else
-      warn "❌ Connection test failed"
+      if response[:ok]
+        puts "✅ Successfully connected to Google Calendar"
+        puts "   User: #{response[:user]} (#{response[:email]})"
+      else
+        warn "❌ Connection test failed"
+      end
+    rescue RuntimeError => e
+      puts "❌ Failed to connect to Google Calendar: #{e.message}"
+      exit 1
     end
-  rescue RuntimeError => e
-    puts "❌ Failed to connect to Google Calendar: #{e.message}"
-    exit 1
-  end
 
-  desc "meetings", "List Google Meet meetings"
-  method_option :start_date, type: :string, aliases: "-s", desc: "Start date (YYYY-MM-DD)"
-  method_option :end_date, type: :string, aliases: "-e", desc: "End date (YYYY-MM-DD)"
-  method_option :max_results, type: :numeric, default: 20, aliases: "-n", desc: "Max number of meetings"
-  method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
-  def meetings
-    result = tool.list_meetings(
-      start_date: options[:start_date],
-      end_date: options[:end_date],
-      max_results: options[:max_results],
-      calendar_id: options[:calendar_id] || "primary"
-    )
-    meetings = result[:meetings]
+    desc "meetings", "List Google Meet meetings"
+    method_option :start_date, type: :string, aliases: "-s", desc: "Start date (YYYY-MM-DD)"
+    method_option :end_date, type: :string, aliases: "-e", desc: "End date (YYYY-MM-DD)"
+    method_option :max_results, type: :numeric, default: 20, aliases: "-n", desc: "Max number of meetings"
+    method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
+    def meetings
+      result = tool.list_meetings(
+        start_date: options[:start_date],
+        end_date: options[:end_date],
+        max_results: options[:max_results],
+        calendar_id: options[:calendar_id] || "primary"
+      )
+      meetings = result[:meetings]
 
-    if meetings.empty?
-      puts "🎥 No Google Meet meetings found for the specified date range"
-    else
-      puts "🎥 Found #{result[:count]} Google Meet meeting(s):"
-      meetings.each_with_index do |meeting, index|
-        puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
-        puts "      Start: #{format_datetime(meeting[:start])}"
-        puts "      End: #{format_datetime(meeting[:end])}"
-        puts "      Description: #{meeting[:description]}" if meeting[:description]
-        puts "      Location: #{meeting[:location]}" if meeting[:location]
-        puts "      Attendees: #{meeting[:attendees].join(", ")}" if meeting[:attendees]&.any?
-        puts "      Meet Link: #{meeting[:meet_link]}"
-        puts "      Calendar Link: #{meeting[:html_link]}"
-        puts
+      if meetings.empty?
+        puts "🎥 No Google Meet meetings found for the specified date range"
+      else
+        puts "🎥 Found #{result[:count]} Google Meet meeting(s):"
+        meetings.each_with_index do |meeting, index|
+          puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
+          puts "      Start: #{format_datetime(meeting[:start])}"
+          puts "      End: #{format_datetime(meeting[:end])}"
+          puts "      Description: #{meeting[:description]}" if meeting[:description]
+          puts "      Location: #{meeting[:location]}" if meeting[:location]
+          puts "      Attendees: #{meeting[:attendees].join(", ")}" if meeting[:attendees]&.any?
+          puts "      Meet Link: #{meeting[:meet_link]}"
+          puts "      Calendar Link: #{meeting[:html_link]}"
+          puts
+        end
       end
+    rescue RuntimeError => e
+      warn "❌ Failed to list meetings: #{e.message}"
+      exit 1
     end
-  rescue RuntimeError => e
-    warn "❌ Failed to list meetings: #{e.message}"
-    exit 1
-  end
 
-  desc "upcoming", "List upcoming Google Meet meetings"
-  method_option :max_results, type: :numeric, default: 10, aliases: "-n", desc: "Max number of meetings"
-  method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
-  def upcoming
-    result = tool.upcoming_meetings(
-      max_results: options[:max_results],
-      calendar_id: options[:calendar_id] || "primary"
-    )
-    meetings = result[:meetings]
+    desc "upcoming", "List upcoming Google Meet meetings"
+    method_option :max_results, type: :numeric, default: 10, aliases: "-n", desc: "Max number of meetings"
+    method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
+    def upcoming
+      result = tool.upcoming_meetings(
+        max_results: options[:max_results],
+        calendar_id: options[:calendar_id] || "primary"
+      )
+      meetings = result[:meetings]
 
-    if meetings.empty?
-      puts "🎥 No upcoming Google Meet meetings found in the next 24 hours"
-    else
-      puts "🎥 Found #{result[:count]} upcoming Google Meet meeting(s):"
-      meetings.each_with_index do |meeting, index|
-        puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
-        puts "      Start: #{format_datetime(meeting[:start])} (#{meeting[:time_until_start]})"
-        puts "      End: #{format_datetime(meeting[:end])}"
-        puts "      Description: #{meeting[:description]}" if meeting[:description]
-        puts "      Location: #{meeting[:location]}" if meeting[:location]
-        puts "      Attendees: #{meeting[:attendees].join(", ")}" if meeting[:attendees]&.any?
-        puts "      Meet Link: #{meeting[:meet_link]}"
-        puts "      Calendar Link: #{meeting[:html_link]}"
-        puts
+      if meetings.empty?
+        puts "🎥 No upcoming Google Meet meetings found in the next 24 hours"
+      else
+        puts "🎥 Found #{result[:count]} upcoming Google Meet meeting(s):"
+        meetings.each_with_index do |meeting, index|
+          puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
+          puts "      Start: #{format_datetime(meeting[:start])} (#{meeting[:time_until_start]})"
+          puts "      End: #{format_datetime(meeting[:end])}"
+          puts "      Description: #{meeting[:description]}" if meeting[:description]
+          puts "      Location: #{meeting[:location]}" if meeting[:location]
+          puts "      Attendees: #{meeting[:attendees].join(", ")}" if meeting[:attendees]&.any?
+          puts "      Meet Link: #{meeting[:meet_link]}"
+          puts "      Calendar Link: #{meeting[:html_link]}"
+          puts
+        end
       end
+    rescue RuntimeError => e
+      warn "❌ Failed to list upcoming meetings: #{e.message}"
+      exit 1
     end
-  rescue RuntimeError => e
-    warn "❌ Failed to list upcoming meetings: #{e.message}"
-    exit 1
-  end
 
-  desc "search QUERY", "Search for Google Meet meetings by text content"
-  method_option :start_date, type: :string, aliases: "-s", desc: "Start date (YYYY-MM-DD)"
-  method_option :end_date, type: :string, aliases: "-e", desc: "End date (YYYY-MM-DD)"
-  method_option :max_results, type: :numeric, default: 10, aliases: "-n", desc: "Max number of meetings"
-  def search(query)
-    result = tool.search_meetings(
-      query,
-      start_date: options[:start_date],
-      end_date: options[:end_date],
-      max_results: options[:max_results]
-    )
-    meetings = result[:meetings]
+    desc "search QUERY", "Search for Google Meet meetings by text content"
+    method_option :start_date, type: :string, aliases: "-s", desc: "Start date (YYYY-MM-DD)"
+    method_option :end_date, type: :string, aliases: "-e", desc: "End date (YYYY-MM-DD)"
+    method_option :max_results, type: :numeric, default: 10, aliases: "-n", desc: "Max number of meetings"
+    def search(query)
+      result = tool.search_meetings(
+        query,
+        start_date: options[:start_date],
+        end_date: options[:end_date],
+        max_results: options[:max_results]
+      )
+      meetings = result[:meetings]
 
-    if meetings.empty?
-      puts "🔍 No Google Meet meetings found matching '#{query}'"
-    else
-      puts "🔍 Found #{result[:count]} Google Meet meeting(s) matching '#{query}':"
-      meetings.each_with_index do |meeting, index|
-        puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
-        puts "      Start: #{format_datetime(meeting[:start])}"
-        puts "      End: #{format_datetime(meeting[:end])}"
-        puts "      Description: #{meeting[:description]}" if meeting[:description]
-        puts "      Location: #{meeting[:location]}" if meeting[:location]
-        puts "      Meet Link: #{meeting[:meet_link]}"
-        puts "      Calendar Link: #{meeting[:html_link]}"
-        puts
+      if meetings.empty?
+        puts "🔍 No Google Meet meetings found matching '#{query}'"
+      else
+        puts "🔍 Found #{result[:count]} Google Meet meeting(s) matching '#{query}':"
+        meetings.each_with_index do |meeting, index|
+          puts "   #{index + 1}. #{meeting[:summary] || "No title"}"
+          puts "      Start: #{format_datetime(meeting[:start])}"
+          puts "      End: #{format_datetime(meeting[:end])}"
+          puts "      Description: #{meeting[:description]}" if meeting[:description]
+          puts "      Location: #{meeting[:location]}" if meeting[:location]
+          puts "      Meet Link: #{meeting[:meet_link]}"
+          puts "      Calendar Link: #{meeting[:html_link]}"
+          puts
+        end
       end
+    rescue RuntimeError => e
+      warn "❌ Failed to search meetings: #{e.message}"
+      exit 1
     end
-  rescue RuntimeError => e
-    warn "❌ Failed to search meetings: #{e.message}"
-    exit 1
-  end
 
-  desc "url EVENT_ID", "Get the Google Meet URL for a specific event"
-  method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
-  def url(event_id)
-    result = tool.get_meeting_url(event_id, calendar_id: options[:calendar_id] || "primary")
+    desc "url EVENT_ID", "Get the Google Meet URL for a specific event"
+    method_option :calendar_id, type: :string, aliases: "-c", desc: "Calendar ID (default: primary)"
+    def url(event_id)
+      result = tool.get_meeting_url(event_id, calendar_id: options[:calendar_id] || "primary")
 
-    puts "🎥 #{result[:summary] || "Meeting"}"
-    puts "   Start: #{format_datetime(result[:start])}"
-    puts "   End: #{format_datetime(result[:end])}"
-    puts "   Meet Link: #{result[:meet_link]}"
-    puts "   Event ID: #{result[:event_id]}"
-  rescue RuntimeError => e
-    warn "❌ Failed to get meeting URL: #{e.message}"
-    exit 1
-  end
+      puts "🎥 #{result[:summary] || "Meeting"}"
+      puts "   Start: #{format_datetime(result[:start])}"
+      puts "   End: #{format_datetime(result[:end])}"
+      puts "   Meet Link: #{result[:meet_link]}"
+      puts "   Event ID: #{result[:event_id]}"
+    rescue RuntimeError => e
+      warn "❌ Failed to get meeting URL: #{e.message}"
+      exit 1
+    end
 
-  private
+    private
 
-  def tool
-    @tool ||= GmeetTool.new
-  end
+    def tool
+      @tool ||= Service.new
+    end
 
-  def format_datetime(datetime_info)
-    return "Unknown" unless datetime_info
+    def format_datetime(datetime_info)
+      return "Unknown" unless datetime_info
 
-    if datetime_info[:date]
-      # All-day event
-      datetime_info[:date]
-    elsif datetime_info[:date_time]
-      # Specific time event
-      time = Time.parse(datetime_info[:date_time])
-      time.strftime("%Y-%m-%d %H:%M")
-    else
-      "Unknown"
+      if datetime_info[:date]
+        # All-day event
+        datetime_info[:date]
+      elsif datetime_info[:date_time]
+        # Specific time event
+        time = Time.parse(datetime_info[:date_time])
+        time.strftime("%Y-%m-%d %H:%M")
+      else
+        "Unknown"
+      end
     end
   end
 end