platform/dev/doc-import-tool/src/index.ts

//
// Copyright © 2023 Hardcore Engineering Inc.
//
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. You may
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//
// See the License for the specific language governing permissions and
// limitations under the License.
//
import { Employee } from '@hcengineering/contact'
import documents, { DocumentSpace } from '@hcengineering/controlled-documents'
import { MeasureMetricsContext, Ref, getWorkspaceId, systemAccountEmail } from '@hcengineering/core'
import { setMetadata } from '@hcengineering/platform'
import serverClientPlugin from '@hcengineering/server-client'
import { type StorageAdapter } from '@hcengineering/server-core'
import { buildStorageFromConfig, storageConfigFromEnv } from '@hcengineering/server-storage'
import serverToken, { generateToken } from '@hcengineering/server-token'
import { program } from 'commander'

import { importDoc } from './commands'
import { Config } from './config'
import { getBackend } from './convert/convert'

/**
 * @public
 */
export function docImportTool (): void {
  const ctx = new MeasureMetricsContext('doc-import-tool', {})

  const serverSecret = process.env.SERVER_SECRET
  if (serverSecret === undefined) {
    console.error('please provide server secret')
    process.exit(1)
  }

  const accountUrl = process.env.ACCOUNTS_URL
  if (accountUrl === undefined) {
    console.error('please provide account url')
    process.exit(1)
  }

  const collaboratorUrl = process.env.COLLABORATOR_URL
  if (collaboratorUrl === undefined) {
    console.error('please provide collaborator url')
    process.exit(1)
  }

  const collaborator = process.env.COLLABORATOR

  const uploadUrl = process.env.UPLOAD_URL ?? '/files'

  const mongodbUri = process.env.MONGO_URL
  if (mongodbUri === undefined) {
    console.log('Please provide mongodb url')
    process.exit(1)
  }

  setMetadata(serverClientPlugin.metadata.Endpoint, accountUrl)
  setMetadata(serverToken.metadata.Secret, serverSecret)

  async function withStorage (mongodbUri: string, f: (storageAdapter: StorageAdapter) => Promise<any>): Promise<void> {
    const adapter = buildStorageFromConfig(storageConfigFromEnv(), mongodbUri)
    try {
      await f(adapter)
    } catch (err: any) {
      console.error(err)
    }
    await adapter.close()
  }

  program.version('0.0.1')

  program
    .command('import <doc> <workspace> <owner>')
    .description('import doc into workspace')
    .option('-s|--spec <spec>', 'Specification file')
    .option('-b|--backend <backend>', 'Conversion backend', 'pandoc')
    .option('--space <space>', 'Doc space ID', documents.space.QualityDocuments)
    .action(
      async (
        doc: string,
        workspace: string,
        owner: Ref<Employee>,
        cmd: { backend: string, space: Ref<DocumentSpace>, spec?: string }
      ) => {
        console.log(
          `Importing document '${doc}' into workspace '${workspace}', owner: ${JSON.stringify(owner)}, spec: ${
            cmd.spec
          }, space: ${cmd.space}, backend: ${cmd.backend}`
        )

        await withStorage(mongodbUri, async (storageAdapter) => {
          const workspaceId = getWorkspaceId(workspace)

          const config: Config = {
            doc,
            workspaceId,
            owner,
            backend: getBackend(cmd.backend),
            specFile: cmd.spec,
            space: cmd.space,
            uploadURL: uploadUrl,
            storageAdapter,
            collaboratorURL: collaboratorUrl,
            collaborator,
            token: generateToken(systemAccountEmail, workspaceId)
          }

          await importDoc(ctx, config)
        })
      }
    )

  program.parse(process.argv)
}