restrict number of timetravel versions without license

Author: williamstein

src/packages/frontend/collaborators/add-collaborators.tsx

@@ -7,7 +7,7 @@
 Add collaborators to a project
 */
 
-import { Alert, Button, Card, Input, Select } from "antd";
+import { Alert, Button, Input, Select } from "antd";
 import {
   React,
   redux,
@@ -39,9 +39,7 @@ import { alert_message } from "../alerts";
 import { useStudentProjectFunctionality } from "@cocalc/frontend/course";
 import Sandbox from "./sandbox";
 import track from "@cocalc/frontend/user-tracking";
-import { SiteLicenseInput } from "@cocalc/frontend/site-licenses/input";
-import { applyLicense } from "@cocalc/frontend/project/settings/site-license";
-import { BuyLicenseForProject } from "@cocalc/frontend/site-licenses/purchase/buy-license-for-project";
+import RequireLicense from "@cocalc/frontend/site-licenses/require-license";
 
 interface RegisteredUser {
   sort?: string;
@@ -692,28 +690,10 @@ export const AddCollaborators: React.FC<Props> = ({
       style={isFlyout ? { paddingLeft: "5px", paddingRight: "5px" } : undefined}
     >
       {limitExceeded && (
-        <Card
-          size="small"
-          title={
-            <h4>
-              <div style={{ float: "right" }}>
-                <BuyLicenseForProject project_id={project_id} />
-              </div>
-              <Icon name="key" /> Select License
-            </h4>
-          }
-          style={{ margin: "10px 0" }}
-        >
-          <SiteLicenseInput
-            requireValid
-            confirmLabel={"Add this license"}
-            onChange={(license_id) => {
-              applyLicense({ project_id, license_id });
-            }}
-            requireLicense
-            requireMessage={`A license is required to have more than ${unlicensedLimit} collaborators on this project.`}
-          />
-        </Card>
+        <RequireLicense
+          project_id={project_id}
+          message={`A license is required to have more than ${unlicensedLimit} collaborators on this project.`}
+        />
       )}
       {err && <ErrorDisplay error={err} onClose={() => set_err("")} />}
       {state == "searching" && <Loading />}

src/packages/frontend/customize.tsx

@@ -129,6 +129,7 @@ export interface CustomizeState {
   limit_free_project_uptime: number; // minutes
   require_license_to_create_project?: boolean;
   unlicensed_project_collaborator_limit?: number;
+  unlicensed_project_timetravel_limit?: number;
   onprem_quota_heading: string;
   organization_email: string;
   organization_name: string;

src/packages/frontend/frame-editors/time-travel-editor/time-travel.tsx

@@ -8,8 +8,8 @@
 import { useState } from "react";
 import { Button, Checkbox, Tooltip } from "antd";
 import { Map } from "immutable";
-import { redux } from "../../app-framework";
-import { Loading } from "../../components";
+import { redux, useTypedRedux } from "@cocalc/frontend/app-framework";
+import { Loading } from "@cocalc/frontend/components";
 import { TimeTravelActions, TimeTravelState } from "./actions";
 import { Diff } from "./diff";
 import { NavigationButtons } from "./navigation-buttons";
@@ -29,6 +29,8 @@ import { SagewsDiff } from "./sagews-diff";
 import { useAsyncEffect, useEditorRedux } from "@cocalc/frontend/app-framework";
 import { Viewer } from "./viewer";
 import type { Document } from "@cocalc/sync/editor/generic/types";
+import useLicenses from "@cocalc/frontend/site-licenses/use-licenses";
+import RequireLicense from "@cocalc/frontend/site-licenses/require-license";
 
 const HAS_SPECIAL_VIEWER = new Set([
   "tasks",
@@ -55,6 +57,11 @@ interface Props {
 export function TimeTravel(props: Props) {
   const { project_id, path } = props;
   const useEditor = useEditorRedux<TimeTravelState>({ project_id, path });
+  const unlicensedLimit = useTypedRedux(
+    "customize",
+    "unlicensed_project_timetravel_limit",
+  );
+  const licenses = useLicenses({ project_id });
   const versions = useEditor("versions");
   const gitVersions = useEditor("git_versions");
   const hasFullHistory = useEditor("has_full_history");
@@ -400,27 +407,47 @@ export function TimeTravel(props: Props) {
   if (loading) {
     return renderLoading();
   }
+
+  let body;
+  if (
+    unlicensedLimit &&
+    !gitMode &&
+    licenses != null &&
+    version != null &&
+    licenses.size == 0 &&
+    versions.size - unlicensedLimit > version
+  ) {
+    // need license to view this
+    body = (
+      <RequireLicense
+        project_id={project_id}
+        message={`A license is required to view more than ${unlicensedLimit} versions of this file.`}
+      />
+    );
+  } else if (doc != null && docpath != null && docext != null && !changesMode) {
+    body = (
+      <Viewer
+        ext={docext}
+        doc={doc}
+        textMode={textMode}
+        actions={props.actions}
+        id={props.id}
+        path={docpath ? docpath : "a.js"}
+        project_id={props.project_id}
+        font_size={props.font_size}
+        editor_settings={props.editor_settings}
+      />
+    );
+  } else {
+    body = renderDiff();
+  }
+
   return (
     <div className="smc-vfill">
       {renderControls()}
       {renderTimeSelect()}
       {gitMode && !changesMode && renderGitSubject()}
-      <>
-        {doc != null && docpath != null && docext != null && !changesMode && (
-          <Viewer
-            ext={docext}
-            doc={doc}
-            textMode={textMode}
-            actions={props.actions}
-            id={props.id}
-            path={docpath ? docpath : "a.js"}
-            project_id={props.project_id}
-            font_size={props.font_size}
-            editor_settings={props.editor_settings}
-          />
-        )}
-        {renderDiff()}
-      </>
+      {body}
     </div>
   );
 }

src/packages/frontend/site-licenses/require-license.tsx

@@ -0,0 +1,32 @@
+import { Card } from "antd";
+import { BuyLicenseForProject } from "@cocalc/frontend/site-licenses/purchase/buy-license-for-project";
+import { applyLicense } from "@cocalc/frontend/project/settings/site-license";
+import { SiteLicenseInput } from "@cocalc/frontend/site-licenses/input";
+import { Icon } from "@cocalc/frontend/components";
+
+export default function RequireLicense({ project_id, message }) {
+  return (
+    <Card
+      size="small"
+      title={
+        <h4>
+          <div style={{ float: "right" }}>
+            <BuyLicenseForProject project_id={project_id} />
+          </div>
+          <Icon name="key" /> Select License
+        </h4>
+      }
+      style={{ margin: "10px 0" }}
+    >
+      <SiteLicenseInput
+        requireValid
+        confirmLabel={"Add this license"}
+        onChange={(license_id) => {
+          applyLicense({ project_id, license_id });
+        }}
+        requireLicense
+        requireMessage={message}
+      />
+    </Card>
+  );
+}

src/packages/frontend/site-licenses/use-licenses.ts

@@ -0,0 +1,15 @@
+import { useTypedRedux } from "@cocalc/frontend/app-framework";
+import { useMemo } from "react";
+import { Map } from "immutable";
+
+export default function useLicenses({ project_id }) {
+  const project_map = useTypedRedux("projects", "project_map");
+  const project = useMemo(
+    () => project_map?.get(project_id),
+    [project_id, project_map],
+  );
+  if (project == null) {
+    return null;
+  }
+  return project.get("site_license") ?? Map();
+}

src/packages/util/db-schema/site-defaults.ts

@@ -80,6 +80,7 @@ export type SiteSettingsKeys =
   | "limit_free_project_uptime"
   | "require_license_to_create_project"
   | "unlicensed_project_collaborator_limit"
+  | "unlicensed_project_timetravel_limit"
   | "google_analytics"
   | "kucalc"
   | "dns"
@@ -602,6 +603,16 @@ export const site_settings_conf: SiteSettings = {
     to_display: (val) => `${val} users`,
     tags: ["Commercialization"],
   },
+  unlicensed_project_timetravel_limit: {
+    name: "Require License to View Unlimited TimeTravel History",
+    desc: "If this number is positive, then projects without a valid license can view at most this many past versions of a file via TimeTravel.  Set this to 200 to allow up to 200 past versions.",
+    default: "0",
+    to_val: to_int,
+    valid: only_nonneg_int,
+    show: only_cocalc_com,
+    to_display: (val) => `${val} versions`,
+    tags: ["Commercialization"],
+  },
   datastore: {
     name: "Datastore",
     desc: `Show the '${DATASTORE_TITLE}' panel in the project settings`,