action.ts

import * as core from '@actions/core';
import { HttpClient } from '@actions/http-client';
import { IRequestOptions } from '@actions/http-client/interfaces';
import { ICollection, RecordMap, TCollectionBlock } from '@nishans/types';
import fs from 'fs';
import { ActionUtils } from './utils';

export async function action() {
  try {
    const NOTION_TOKEN_V2 = core.getInput('token_v2');
    let id = core.getInput('database_id').replace(/-/g, '');
    const databaseId = `${id.substr(0, 8)}-${id.substr(8, 4)}-${id.substr(
      12,
      4
    )}-${id.substr(16, 4)}-${id.substr(20)}`;

    const headers: IRequestOptions['headers'] = {
      Accept: 'application/json',
      'Content-Type': 'application/json',
      cookie: `token_v2=${NOTION_TOKEN_V2}`
    };

    const http = new HttpClient(undefined, undefined, {
      headers
    });

    const collectionView = await ActionUtils.fetchData<TCollectionBlock>(
      databaseId,
      'block',
      http
    );
    core.info('Fetched database');

    const collection_id = collectionView.collection_id;
    const collection = await ActionUtils.fetchData<ICollection>(
      collection_id,
      'collection',
      http
    );

    core.info('Fetched collection');

    const response = await http.post(
      `https://www.notion.so/api/v3/queryCollection`,
      JSON.stringify({
        collection: {
          id: collection_id,
          spaceId: collectionView.space_id
        },
        collectionView: {
          id: collectionView.view_ids[0],
          spaceId: collectionView.space_id
        },
        loader: {
          type: 'reducer',
          reducers: {
            collection_group_results: {
              type: 'results'
            }
          },
          searchQuery: '',
          userTimeZone: 'Asia/Dhaka'
        }
      })
    );

    const { recordMap } = JSON.parse(await response.readBody()) as {
      recordMap: RecordMap;
    };

    core.info('Fetched rows');
    const { schema } = collection;
    const [
      categorySchemaEntry,
      colorSchemaEntry,
      ,
      base64SchemaEntry
    ] = ActionUtils.getSchemaEntries(schema);

    const rows = ActionUtils.modifyRows(recordMap, databaseId);
    const categoriesMap = ActionUtils.constructCategoriesMap(
      categorySchemaEntry[1]
    );
    ActionUtils.populateCategoriesMapItems(
      rows,
      categorySchemaEntry[0],
      categoriesMap
    );

    const README_PATH = `${process.env.GITHUB_WORKSPACE}/README.md`;
    core.info(`Reading from ${README_PATH}`);

    const readmeLines = fs.readFileSync(README_PATH, 'utf-8').split('\n');

    const [startIdx, endIdx] = ActionUtils.checkForSections(readmeLines);
    const newLines = ActionUtils.constructNewContents(
      categoriesMap,
      colorSchemaEntry[0],
      base64SchemaEntry[0]
    );

    const finalLines = [
      ...readmeLines.slice(0, startIdx + 1),
      ...newLines,
      ...readmeLines.slice(endIdx)
    ];

    core.info(`Writing to ${README_PATH}`);

    fs.writeFileSync(README_PATH, finalLines.join('\n'), 'utf-8');
    await ActionUtils.commitFile();
  } catch (err) {
    core.error(err.message);
    core.setFailed(err.message);
  }
}