module Sinatra::UC3ServicesRoutes

# File app/lib/routes/services.rb, line 44
def self.registered(app)
  app.get '/json/ui/state' do
    get_url("#{ui_host}/state.json")
  end

  app.get '/json/ui/audit-replic' do
    get_url("#{ui_host}/state-audit-replic.json")
  end

  app.get '/json/ingest/state' do
    get_url("#{ingest_host}/state?t=json")
  end

  app.get '/json/ingest/tag' do
    get_url("#{ingest_host}/static/build.content.txt")
  end

  app.get '/json/store/state' do
    get_url("#{store_host}/state?t=json")
  end

  app.get '/json/store/tag' do
    get_url("#{store_host}/static/build.content.txt")
  end

  app.get '/json/access/state' do
    get_url("#{access_host}/state?t=json")
  end

  app.get '/json/access/tag' do
    get_url("#{access_host}/static/build.content.txt")
  end

  app.get '/json/store/nodes' do
    get_url("#{store_host}/jsonstatus")
  end

  app.get '/json/store/hostname' do
    get_url("#{store_host}/hostname")
  end

  app.get '/json/inventory/state' do
    get_url("#{inventory_host}/state?t=json")
  end

  app.post '/json/inventory/start' do
    post_url("#{inventory_host}/service/start?t=json")
  end

  app.post '/json/inventory/stop' do
    post_url("#{inventory_host}/service/stop?t=json")
  end

  app.post '/json/inventory/admin-init' do
    post_url("#{inventory_host}/admin/init")
  end

  app.get '/json/inventory/tag' do
    get_url("#{inventory_host}/static/build.content.txt")
  end

  app.get '/json/audit/state' do
    get_url("#{audit_host}/state?t=json")
  end

  app.get '/json/audit/tag' do
    get_url("#{audit_host}/static/build.content.txt")
  end

  app.get '/json/audit/nodes' do
    get_url("#{audit_host}/jsonstatus")
  end

  app.post '/json/audit/start' do
    post_url("#{audit_host}/service/start?t=json")
  end

  app.post '/json/audit/stop' do
    post_url("#{audit_host}/service/stop?t=json")
  end

  app.get '/json/replic/state' do
    get_url("#{replic_host}/state?t=json")
  end

  app.get '/json/replic/tag' do
    get_url("#{replic_host}/static/build.content.txt")
  end

  app.post '/json/replic/start' do
    post_url("#{replic_host}/service/start?t=json")
  end

  app.post '/json/replic/pause' do
    post_url("#{replic_host}/service/pause?t=json")
  end

  app.get '/json/replic/nodes' do
    get_url("#{replic_host}/jsonstatus")
  end

  app.get '/json/access/state' do
    get_url("#{access_host}/state?t=json")
  end

  app.get '/json/access/tag' do
    get_url("#{access_host}/static/build.content.txt")
  end

  app.get '/json/access/nodes' do
    get_url("#{access_host}/jsonstatus")
  end

  app.post '/stack-init' do
    content_type :json
    stack_init.to_json
  end

  app.post '/collections-init' do
    content_type :json
    collections_init.to_json
  end

  app.post '/ops/inventory/rebuild' do
    data = {
      url: manifest_url(request.params),
      responseForm: 'json'
    }

    deleteurl = "#{inventory_host}/object/#{CGI.escape(request.params['ark'])}"
    resp = delete_url_resp(deleteurl)
    if resp.code.to_i == 200
      addurl = "#{inventory_host}/add"
      resp = post_url_multipart(addurl, data)
      if resp.code.to_i == 200
        url = "/queries/repository/object-ark?ark=#{CGI.escape(request.params['ark'])}"
        {
          message: "Sucessfully re-added #{request.params['ark']}",
          redirect: url,
          modal: true
        }.to_json
      else
        content_type :json
        { uri: addurl, message: "ERROR re-adding object: #{resp.code}" }.to_json
      end
    else
      content_type :json
      { uri: deleteurl, message: "ERROR deleting object: #{resp.code}" }.to_json
    end
  rescue StandardError => e
    content_type :json
    { uri: deleteurl, message: "Exception while rebuilding inventory: #{e}" }.to_json
  end

  app.post '/ops/storage-nodes/remove-obsolete' do
    arks = UC3Query::QueryClient.client.run_query(request.path, request.params)
    nodenumber = request.params.fetch('node_number', -1)
    success = 0
    fail = 0
    arks.each do |row|
      ark = row.fetch('ark', '')
      next if ark.empty?

      deleteurl = "#{replic_host}/delete/#{nodenumber}/#{CGI.escape(ark)}"
      puts deleteurl
      dresp = delete_url_resp(deleteurl)
      if dresp.code.to_i == 200
        success += 1
      else
        fail += 1
      end
    end

    content_type :json
    { message: "Deleted objects: #{success}; failed: #{fail}" }.to_json
  end

  app.get '/ops/storage/manifest' do
    url = "#{store_host}/manifest/#{request.params['node_number']}/#{CGI.escape(request.params['ark'])}"
    puts "URL: #{url}"
    get_url(url, ctype: :xml)
  end

  app.get '/ops/storage/manifest' do
    get_url(manifest_url(request.params), ctype: :xml)
  end

  app.get '/ops/storage/manifest-yaml' do
    data = get_url_body(manifest_url(request.params))
    content_type :yaml
    ManifestToYaml.new.load_xml(data)
  end

  app.get '/ops/storage/ingest-checkm' do
    nodenum = request.params['node_number']
    ark = request.params['ark']
    ver = request.params['version_number']
    url = "#{store_host}/ingestlink/#{nodenum}/#{CGI.escape(ark)}/#{ver}?presign=true"
    get_url(url, ctype: :text)
  end

  app.post '/ops/storage/scans/cancel-all-scans' do
    post_url("#{replic_host}/scan/allow/false?t=json")
  end

  app.post '/ops/storage/scans/allow-all' do
    post_url("#{replic_host}/scan/allow/true?t=json")
  end

  app.post '/ops/storage/scans/start' do
    post_url_message("#{replic_host}/scan/start/#{request.params.fetch('node_number', 'na')}?t=json",
      message: 'Scan Started')
  end

  app.post '/ops/storage/scans/resume' do
    post_url_message("#{replic_host}/scan/restart/#{request.params.fetch('inv_scan_id', 'na')}?t=json",
      message: 'Scan Restarted')
  end

  app.post '/ops/storage/scans/cancel' do
    post_url_message("#{replic_host}/scan/cancel/#{request.params.fetch('inv_scan_id', 'na')}?t=json",
      message: 'Scan Cancelled')
  end

  app.post '/ops/storage/scans/delete' do
    delete_url_message("#{replic_host}/scandelete/#{request.params.fetch('maint_id', 0)}?t=json",
      message: 'Scan Item Removed from Cloud Storage')
  end

  app.post '/ops/storage/scans/batch-delete' do
    delete_url_message("#{replic_host}/scandelete-list/#{request.params.fetch('node_number', 0)}?t=json",
      message: 'Scan Delete Batch Initiated')
  end

  app.post '/ops/storage/scans/applycsv' do
    count = 0
    errors = 0
    CSV.parse(request.body.read).each_with_index do |row, i|
      next if i.zero?

      row[11] = '' if row[11].nil?
      row[12] = '' if row[12].nil?
      next if row[11].empty?
      next if row[9] == row[11] && row[10] == row[12]

      params = {}
      params['maint_status'] = row[11]
      params['note'] = row[12]
      params['node_number'] = row[0].to_i
      params['maint_id'] = row[1].to_i
      res = UC3Query::QueryClient.client.query_update(
        '/queries-update/storage-maints/apply-review-change',
        params
      )
      if res.fetch(:status, '') == 'OK'
        puts params
        count += 1
      else
        puts res
        errors += 1
      end
    end
    content_type :json
    { message: "Changes applied #{count}; Errors: #{errors}" }.to_json
  end

  app.get '/ops/storage/storage-config' do
    rows = []
    defprofile = case UC3::UC3Client.stack_name
                 when UC3::UC3Client::ECS_EPHEMERAL
                   'minio-ephemeral'
                 when 'docker'
                   'minio-docker'
                 else
                   ''
                 end

    nodes = ::JSON.parse(get_url_body("#{store_host}/jsonstatus"))
    nodes.fetch('NodesStatus', []).each do |node|
      row = {
        node_number: node.fetch('node', ''),
        bucket: node.fetch('bucket', ''),
        description: node.fetch('description', ''),
        profile: defprofile
      }
      row[:profile] = 'sdsc' if row[:bucket] =~ /sdsc/
      row[:profile] = 'wasabi' if row[:bucket] =~ /wasabi/
      rows << row
    end

    rows << {
      bucket: ENV.fetch('S3CONFIG_BUCKET', ''),
      description: 'Configuration Bucket. AWS CodeBuild copies configuration data from GitHub into this bucket.',
      profile: defprofile
    }
    rows << {
      bucket: ENV.fetch('S3REPORT_BUCKET', ''),
      description: 'Reporting Bucket.' \
                   'The Merritt Admin Tool builds static reports into this bucket. ' \
                   'A lifecycle policy will expire content in this bucket.',
      profile: defprofile
    }
    rows << {
      bucket: ENV.fetch('S3WORKSPACE_BUCKET', ''),
      description: 'Versioned Workspace Bucket. Merritt Ingest and Storage ' \
                   'services will use this bucket for content that is actively being ingested. ' \
                   'A lifecycle policy will function like a recycle bin for content that has been deleted ' \
                   'from this bucket.',
      profile: defprofile
    }

    table = AdminUI::FilterTable.new(
      columns: [
        AdminUI::Column.new(:description, header: 'Description'),
        AdminUI::Column.new(:node_number, header: 'Node Number'),
        AdminUI::Column.new(:bucket, header: 'Bucket'),
        AdminUI::Column.new(:profile, header: 'Profile'),
        AdminUI::Column.new(:command, header: 'Command')
      ]
    )
    rows.each do |row|
      cmd = UC3::UC3Client.stack_name == 'docker' ? 'docker compose exec -it merrittdev /bin/bash' : "session #{UC3::UC3Client.cluster_name}/merrittdev"
      cmd += "\n"
      cmd += "\n/set-credentials.sh" if %w[sdsc wasabi].include?(row[:profile])
      cmd += "\naws s3#{" --profile #{row[:profile]}" unless row[:profile].empty?} ls s3://#{row[:bucket]}/"
      row[:command] = cmd unless row[:bucket].empty?
      table.add_row(
        AdminUI::Row.make_row(
          table.columns,
          row
        )
      )
    end
    adminui_show_table(
      AdminUI::Context.new(request.path),
      table
    )
  end

  app.post '/test/load/data' do
    test_data_dir = "#{UC3::FileSystemClient::DIR}/test-data"
    `mkdir -p #{test_data_dir}`
    sample_data = 'https://raw.githubusercontent.com/CDLUC3/mrt-doc/refs/heads/main/sampleFiles/'
    %w[merritt_demo cdl_dryaddev escholarship terry_test].each do |mnemonic|
      [
        { type: 'container-batch-manifest', file: 'sampleBatchOfContainers.checkm' },
        { type: 'object-manifest', file: 'sampleBatchOfFiles.checkm' },
        { type: 'batch-manifest', file: 'sampleBatchOfManifests.checkm' },
        { type: 'container', file: 'jazzbears.zip' },
        { type: 'file', file: 'bigHunt2.jpg' }
      ].each do |subm|
        file = "#{test_data_dir}/#{subm[:file]}"
        url = "#{sample_data}/#{subm[:file]}"
        `curl -L -o #{file} #{url} 2>/dev/null` unless File.exist?(file)
        load_test_file_to_merritt(file, subm[:type], mnemonic)
        sleep 5
      end
    end
    redirect '/ops/zk/ingest/jobs-by-collection'
  end
end

# File app/lib/routes/services.rb, line 28
def access_host
  "http://#{ENV.fetch('SVC_ACCESS', 'access:8080/store')}"
end

# File app/lib/routes/services.rb, line 32
def audit_host
  "http://#{ENV.fetch('SVC_AUDIT', 'audit:8080/audit')}"
end

# File app/lib/routes/services.rb, line 459
def collections_init
  resp = []
  [
    { ark: 'ark:/13030/m5rn35s8', name: 'Merritt Demo', mnemonic: 'merritt_demo', public: true },
    { ark: 'ark:/13030/m5qv8jks', name: 'cdl_dryaddev', mnemonic: 'cdl_dryaddev', public: true },
    { ark: 'ark:/13030/m5154f09', name: 'escholarship', mnemonic: 'escholarship', public: true },
    { ark: 'ark:/13030/99999999', name: 'Terry Test', mnemonic: 'terry_test', public: true }
  ].each do |c|
    r = UC3S3::ConfigObjectsClient.client.add_collection(
      c[:ark], c[:name], c[:mnemonic], public: c.fetch(:public, false)
    )
    begin
      resp << ::JSON.parse(r.body)
    rescue StandardError => e
      resp << { action: "Create Collection #{c[:name]}", error: e.to_s }
    end
    begin
      params = {}
      params['ark'] = c[:ark]
      r = UC3Query::QueryClient.client.query_update(
        '/init/create-collection-nodes',
        params,
        purpose: "Create Storage Nodes for Collection #{c[:ark]}"
      )
      resp << r
    rescue StandardError => e
      resp << { action: "Create Storage Nodes for Collection #{c[:name]}", error: e.to_s }
    end
  end
  resp
end

# File app/lib/routes/services.rb, line 558
def delete_url_message(url, body: nil, message: '')
  resp = delete_url_resp(url, body: body)
  json = ::JSON.parse(resp.body)
  json['message'] = message unless message.empty?
  json.to_json
end

# File app/lib/routes/services.rb, line 545
def delete_url_resp(url, body: nil)
  uri = URI.parse(url)
  puts "Delete URI: #{url}, body: #{body}"
  req = Net::HTTP::Delete.new(uri)
  req['Content-Type'] = '*/*'
  # req['Accept'] = 'text/plain'
  req.body = body

  Net::HTTP.start(uri.hostname, uri.port, use_ssl: uri.scheme == 'https') do |http|
    http.request(req)
  end
end

# File app/lib/routes/services.rb, line 497
def get_url(url, ctype: :json)
  uri = URI.parse(url)
  response = Net::HTTP.get_response(uri)
  data = response.body
  content_type ctype
  data
rescue StandardError => e
  content_type :json
  { uri: uri, error: e.to_s }.to_json
end

# File app/lib/routes/services.rb, line 491
def get_url_body(url)
  uri = URI.parse(url)
  response = Net::HTTP.get_response(uri)
  response.body
end

# File app/lib/routes/services.rb, line 20
def ingest_host
  "http://#{ENV.fetch('SVC_INGEST', 'ingest:8080/ingest')}"
end

# File app/lib/routes/services.rb, line 40
def inventory_host
  "http://#{ENV.fetch('SVC_INVENTORY', 'inventory:8080/inventory')}"
end

# File app/lib/routes/services.rb, line 406
def load_test_file_to_merritt(file, type, mnemonic)
  puts "Loading #{file} to #{mnemonic} #{ui_host}/object/update"

  puts `curl -H 'Accept: application/json' \
    -F 'file=@#{file}' \
    -F 'title=#{type}: #{file}' \
    -F 'submitter=merritt-test' \
    -F 'responseForm=xml' \
    -F 'profile=#{mnemonic}_content' \
    --user 'merritt-test:password' #{ui_host}/object/update
  `
end

# File app/lib/routes/services.rb, line 419
def manifest_url(params)
  "#{store_host}/manifest/#{params['node_number']}/#{CGI.escape(params['ark'])}"
end

# File app/lib/routes/services.rb, line 529
def post_url(url, body: nil, user: nil, password: nil)
  json = post_url_body(url, body: body, user: user, password: password)
  content_type :json
  json
rescue StandardError => e
  content_type :json
  { uri: uri, error: e.to_s }.to_json
end

# File app/lib/routes/services.rb, line 509
def post_url_body(url, body: nil, user: nil, password: nil)
  puts "URI: #{url}, body: #{body}, user: #{user}, password: #{'****' if password}"
  uri = URI.parse(url)
  req = Net::HTTP::Post.new(uri)
  if user && password
    req.basic_auth(user, password)
    # token = Base64.strict_encode64("#{user}:#{password}")
    # req['Authorization'] = "Basic #{token}"
  end
  req['Content-Type'] = 'text/plain; charset=UTF-8'
  req['Accept'] = 'text/plain'

  req.body = body

  response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: uri.scheme == 'https') do |http|
    http.request(req)
  end
  response.body
end

# File app/lib/routes/services.rb, line 538
def post_url_message(url, body: nil, user: nil, password: nil, message: '')
  jsonbody = post_url_body(url, body: body, user: user, password: password)
  json = ::JSON.parse(jsonbody)
  json['message'] = message unless message.empty?
  json.to_json
end

# File app/lib/routes/services.rb, line 565
def post_url_multipart(url, params)
  puts "Multipart POST #{url} with #{params.inspect}"
  uri = URI.parse(url)
  req = Net::HTTP::Post::Multipart.new(uri, params)
  Net::HTTP.start(uri.hostname, uri.port) do |http|
    http.request(req)
  end
end

# File app/lib/routes/services.rb, line 36
def replic_host
  "http://#{ENV.fetch('SVC_REPLIC', 'replic:8080/replic')}"
end

# File app/lib/routes/services.rb, line 423
def stack_init
  UC3::FileSystemClient.client.cleanup_ingest_folders
  resp = []
  r = post_url("#{inventory_host}/admin/init")
  begin
    resp << ::JSON.parse(r)
  rescue StandardError => e
    resp << { action: 'Inventory Init', error: e.to_s }
  end
  collections_init.each do |r|
    resp << r
  end

  qc = UC3Query::QueryClient.client
  if !qc.nil? && qc.enabled
    begin
      if qc.run_query('/init/check-nodes').empty?
        qc.query_update('/init/create-nodes', purpose: 'Add Test Storage Nodes')
        resp << { action: 'Add test storage nodes', result: 'success' }
      else
        resp << { action: 'Add test storage nodes', result: 'skipped - already exists' }
      end

      if qc.run_query('/init/check-node-counts').empty?
        qc.query_update('/init/create-node-counts', purpose: 'Add Test Storage Node Counts')
        resp << { action: 'Add test storage node counts', result: 'success' }
      else
        resp << { action: 'Add test storage node counts', result: 'skipped - already exists' }
      end
    rescue StandardError => e
      resp << { action: 'Add test storage node and node counts', error: e.to_s }
    end
  end
  resp
end

# File app/lib/routes/services.rb, line 24
def store_host
  "http://#{ENV.fetch('SVC_STORE', 'store:8080/store')}"
end

# File app/lib/routes/services.rb, line 15
def ui_host
  host = ENV.fetch('SVC_UI', 'ui:8086')
  host =~ /^http/ ? host : "http://#{host}"
end