mirror of
https://github.com/cpvalente/ontime.git
synced 2026-08-20 22:49:18 +00:00
improve get state check
This commit is contained in:
@@ -11,6 +11,7 @@ import {
|
|||||||
ModalFooter,
|
ModalFooter,
|
||||||
ModalHeader,
|
ModalHeader,
|
||||||
ModalOverlay,
|
ModalOverlay,
|
||||||
|
Select,
|
||||||
} from '@chakra-ui/react';
|
} from '@chakra-ui/react';
|
||||||
import { IoArrowDownCircleOutline } from '@react-icons/all-files/io5/IoArrowDownCircleOutline';
|
import { IoArrowDownCircleOutline } from '@react-icons/all-files/io5/IoArrowDownCircleOutline';
|
||||||
import { IoArrowUpCircleOutline } from '@react-icons/all-files/io5/IoArrowUpCircleOutline';
|
import { IoArrowUpCircleOutline } from '@react-icons/all-files/io5/IoArrowUpCircleOutline';
|
||||||
@@ -49,12 +50,18 @@ export default function SheetsModal(props: SheetsModalProps) {
|
|||||||
const [userFields, setUserFields] = useState<UserFields | null>(null);
|
const [userFields, setUserFields] = useState<UserFields | null>(null);
|
||||||
const [project, setProject] = useState<ProjectData | null>(null);
|
const [project, setProject] = useState<ProjectData | null>(null);
|
||||||
|
|
||||||
const [sheetState, setSheetState] = useState<GoogleSheetState>({ auth: false, id: false, worksheet: false });
|
const [sheetState, setSheetState] = useState<GoogleSheetState>({
|
||||||
|
secret: false,
|
||||||
|
auth: false,
|
||||||
|
id: false,
|
||||||
|
worksheet: false,
|
||||||
|
worksheetOptions: [],
|
||||||
|
});
|
||||||
|
|
||||||
const queryClient = useQueryClient();
|
const queryClient = useQueryClient();
|
||||||
|
|
||||||
const sheetid = useRef<HTMLInputElement>(null);
|
const sheetid = useRef<HTMLInputElement>(null);
|
||||||
const worksheet = useRef<HTMLInputElement>(null);
|
const worksheet = useRef<HTMLSelectElement>(null);
|
||||||
|
|
||||||
const handleClose = () => {
|
const handleClose = () => {
|
||||||
setRundown(null);
|
setRundown(null);
|
||||||
@@ -82,14 +89,21 @@ export default function SheetsModal(props: SheetsModalProps) {
|
|||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
getSheetSettings().then((data) => {
|
getSheetSettings().then((data) => {
|
||||||
if (sheetid.current?.value != data.id || worksheet.current?.value != data.worksheet) {
|
if (data.id == '') {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
console.log(worksheet.current?.value);
|
||||||
|
if (
|
||||||
|
sheetid.current?.value != data.id ||
|
||||||
|
(worksheet.current?.value != data.worksheet && worksheet.current?.value)
|
||||||
|
) {
|
||||||
_onChange();
|
_onChange();
|
||||||
}
|
if (sheetid.current) {
|
||||||
if (sheetid.current) {
|
sheetid.current.value = data.id;
|
||||||
sheetid.current.value = data.id;
|
}
|
||||||
}
|
if (worksheet.current) {
|
||||||
if (worksheet.current) {
|
worksheet.current.value = data.worksheet;
|
||||||
worksheet.current.value = data.worksheet;
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
@@ -186,14 +200,20 @@ export default function SheetsModal(props: SheetsModalProps) {
|
|||||||
data-testid='file-input'
|
data-testid='file-input'
|
||||||
/>
|
/>
|
||||||
<div>Need to add some help here</div>
|
<div>Need to add some help here</div>
|
||||||
|
<br />
|
||||||
<div>
|
<div>
|
||||||
<Button onClick={handleClick}>Upload Client Secrect</Button>
|
<Button onClick={handleClick}>Upload Client Secrect</Button>
|
||||||
|
{sheetState.secret ? 'have good secret' : 'no or bad secret'}
|
||||||
</div>
|
</div>
|
||||||
<Button variant='ontime-filled' padding='0 2em' onClick={handleAuthenticate}>
|
<br />
|
||||||
Authenticate
|
<div style={sheetState.secret ? {} : { display: 'none' }}>
|
||||||
</Button>
|
<Button variant='ontime-filled' padding='0 2em' onClick={handleAuthenticate}>
|
||||||
{sheetState.auth ? <div>You are authenticated</div> : <div>You are not authenticated</div>}
|
Authenticate
|
||||||
<div>
|
</Button>
|
||||||
|
{sheetState.auth ? 'You are authenticated' : 'You are not authenticated'}
|
||||||
|
</div>
|
||||||
|
<br />
|
||||||
|
<div style={sheetState.auth ? {} : { display: 'none' }}>
|
||||||
<label htmlFor='sheetid'>Sheet ID </label>
|
<label htmlFor='sheetid'>Sheet ID </label>
|
||||||
<InputGroup size='sm'>
|
<InputGroup size='sm'>
|
||||||
<Input
|
<Input
|
||||||
@@ -208,28 +228,20 @@ export default function SheetsModal(props: SheetsModalProps) {
|
|||||||
{sheetState.id ? <IoCheckmarkCircleOutline color='green' /> : <IoCloseCircleOutline color='red' />}
|
{sheetState.id ? <IoCheckmarkCircleOutline color='green' /> : <IoCloseCircleOutline color='red' />}
|
||||||
</InputRightAddon>
|
</InputRightAddon>
|
||||||
</InputGroup>
|
</InputGroup>
|
||||||
<br />
|
|
||||||
<label htmlFor='worksheet'>Worksheet </label>
|
|
||||||
<InputGroup size='sm'>
|
|
||||||
<Input
|
|
||||||
type='text'
|
|
||||||
ref={worksheet}
|
|
||||||
id='worksheet'
|
|
||||||
width='240px'
|
|
||||||
textAlign='right'
|
|
||||||
variant='ontime-filled-on-light'
|
|
||||||
/>
|
|
||||||
<InputRightAddon>
|
|
||||||
{sheetState.worksheet ? (
|
|
||||||
<IoCheckmarkCircleOutline color='green' />
|
|
||||||
) : (
|
|
||||||
<IoCloseCircleOutline color='red' />
|
|
||||||
)}
|
|
||||||
</InputRightAddon>
|
|
||||||
</InputGroup>
|
|
||||||
</div>
|
</div>
|
||||||
<br />
|
<br />
|
||||||
<div>
|
<div style={sheetState.id ? {} : { display: 'none' }}>
|
||||||
|
<label htmlFor='worksheet'>Worksheet </label>
|
||||||
|
<Select ref={worksheet} size='sm' id='worksheet'>
|
||||||
|
{sheetState.worksheetOptions.map((value) => (
|
||||||
|
<option key={value} value={value}>
|
||||||
|
{value}
|
||||||
|
</option>
|
||||||
|
))}
|
||||||
|
</Select>
|
||||||
|
</div>
|
||||||
|
<br />
|
||||||
|
<div style={sheetState.worksheet ? {} : { display: 'none' }}>
|
||||||
<Button
|
<Button
|
||||||
disabled={!sheetState.worksheet}
|
disabled={!sheetState.worksheet}
|
||||||
variant='ontime-subtle-on-light'
|
variant='ontime-subtle-on-light'
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import { sheets, sheets_v4 } from '@googleapis/sheets';
|
import { sheets, sheets_v4 } from '@googleapis/sheets';
|
||||||
import { readFile, writeFile } from 'fs/promises';
|
import { writeFile } from 'fs/promises';
|
||||||
|
import { readFileSync } from 'fs';
|
||||||
import { OAuth2Client } from 'google-auth-library';
|
import { OAuth2Client } from 'google-auth-library';
|
||||||
import http from 'http';
|
import http from 'http';
|
||||||
import { DatabaseModel, GoogleSheetState, LogOrigin } from 'ontime-types';
|
import { DatabaseModel, GoogleSheetState, LogOrigin } from 'ontime-types';
|
||||||
@@ -20,8 +21,9 @@ type ResponseOK = {
|
|||||||
class sheet {
|
class sheet {
|
||||||
private static client: null | OAuth2Client = null;
|
private static client: null | OAuth2Client = null;
|
||||||
private readonly scope = 'https://www.googleapis.com/auth/spreadsheets';
|
private readonly scope = 'https://www.googleapis.com/auth/spreadsheets';
|
||||||
private readonly sheetsFolder;
|
private readonly sheetsFolder: string;
|
||||||
private readonly client_secret;
|
private readonly clientSecretFile: string;
|
||||||
|
private static clientSecret = null;
|
||||||
private static authUrl: null | string = null;
|
private static authUrl: null | string = null;
|
||||||
private worksheetId = 0;
|
private worksheetId = 0;
|
||||||
private sheetId = '';
|
private sheetId = '';
|
||||||
@@ -33,40 +35,77 @@ class sheet {
|
|||||||
throw new Error('Could not resolve public folder for platform');
|
throw new Error('Could not resolve public folder for platform');
|
||||||
}
|
}
|
||||||
this.sheetsFolder = join(appDataPath, 'sheets');
|
this.sheetsFolder = join(appDataPath, 'sheets');
|
||||||
this.client_secret = join(this.sheetsFolder, 'client_secret.json');
|
this.clientSecretFile = join(this.sheetsFolder, 'client_secret.json');
|
||||||
ensureDirectory(this.sheetsFolder);
|
ensureDirectory(this.sheetsFolder);
|
||||||
|
try {
|
||||||
|
sheet.clientSecret = JSON.parse(readFileSync(this.clientSecretFile, 'utf-8'));
|
||||||
|
} catch (_) {
|
||||||
|
/* empty - it is ok thet there is no clientSecret */
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public async getSheetState(): Promise<GoogleSheetState> {
|
public async getSheetState(): Promise<GoogleSheetState> {
|
||||||
|
console.log('getSheetState');
|
||||||
const ret: GoogleSheetState = {
|
const ret: GoogleSheetState = {
|
||||||
|
secret: false,
|
||||||
auth: false,
|
auth: false,
|
||||||
id: false,
|
id: false,
|
||||||
worksheet: false,
|
worksheet: false,
|
||||||
|
worksheetOptions: [],
|
||||||
};
|
};
|
||||||
|
const lastID = this.sheetId;
|
||||||
this.sheetId = '';
|
this.sheetId = '';
|
||||||
this.worksheetId = 0;
|
this.worksheetId = 0;
|
||||||
|
|
||||||
|
if (!sheet.clientSecret) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
ret.secret = true;
|
||||||
if (!sheet.client) {
|
if (!sheet.client) {
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
ret.auth = await this.refreshToken();
|
ret.auth = await this.refreshToken();
|
||||||
if (ret.auth) {
|
|
||||||
const settings = DataProvider.getGoogleSheet();
|
|
||||||
const x = await this.exist(settings.id, settings.worksheet);
|
|
||||||
if (x === true) {
|
|
||||||
ret.id = true;
|
|
||||||
this.sheetId = settings.id;
|
|
||||||
} else if (x !== false) {
|
|
||||||
ret.id = true;
|
|
||||||
ret.worksheet = true;
|
|
||||||
this.sheetId = settings.id;
|
|
||||||
this.worksheetId = x.worksheetId;
|
|
||||||
this.range = x.range;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
logger.error(LogOrigin.Server, `Google Sheet: Faild to refresh token ${err}`);
|
logger.error(LogOrigin.Server, `Google Sheet: Faild to refresh token ${err}`);
|
||||||
}
|
}
|
||||||
|
if (!ret.auth) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
const settings = DataProvider.getGoogleSheet();
|
||||||
|
if (settings.id != '' && lastID != settings.id) {
|
||||||
|
const spreadsheets = await sheets({ version: 'v4', auth: sheet.client })
|
||||||
|
.spreadsheets.get({
|
||||||
|
spreadsheetId: settings.id,
|
||||||
|
includeGridData: false,
|
||||||
|
})
|
||||||
|
.catch((err) => {
|
||||||
|
logger.error(LogOrigin.Server, `Google Sheet: faild to load sheet ${err}`);
|
||||||
|
});
|
||||||
|
if (!spreadsheets || spreadsheets.status != 200) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
ret.id = true;
|
||||||
|
ret.worksheetOptions = spreadsheets.data.sheets.map((i) => i.properties.title);
|
||||||
|
if (ret.worksheetOptions.indexOf(settings.worksheet) < 0) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
ret.worksheet = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// const x = await this.exist(settings.id, settings.worksheet);
|
||||||
|
// if (x === true) {
|
||||||
|
// ret.id = true;
|
||||||
|
// this.sheetId = settings.id;
|
||||||
|
// } else if (x !== false) {
|
||||||
|
// ret.id = true;
|
||||||
|
// ret.worksheet = true;
|
||||||
|
// this.sheetId = settings.id;
|
||||||
|
// this.worksheetId = x.worksheetId;
|
||||||
|
// this.range = x.range;
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -222,6 +261,7 @@ class sheet {
|
|||||||
public async saveClientSecrets(secrets: object) {
|
public async saveClientSecrets(secrets: object) {
|
||||||
sheet.client = null;
|
sheet.client = null;
|
||||||
sheet.authUrl = null;
|
sheet.authUrl = null;
|
||||||
|
sheet.clientSecret = null;
|
||||||
if (
|
if (
|
||||||
!('client_id' in secrets['installed']) ||
|
!('client_id' in secrets['installed']) ||
|
||||||
!('project_id' in secrets['installed']) ||
|
!('project_id' in secrets['installed']) ||
|
||||||
@@ -233,9 +273,10 @@ class sheet {
|
|||||||
) {
|
) {
|
||||||
throw new Error('Sheet slient secret is missing some keys');
|
throw new Error('Sheet slient secret is missing some keys');
|
||||||
}
|
}
|
||||||
await writeFile(this.client_secret, JSON.stringify(secrets), 'utf-8').catch((err) =>
|
await writeFile(this.clientSecretFile, JSON.stringify(secrets), 'utf-8').catch((err) =>
|
||||||
logger.error(LogOrigin.Server, `${err}`),
|
logger.error(LogOrigin.Server, `${err}`),
|
||||||
);
|
);
|
||||||
|
sheet.clientSecret = secrets;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -274,13 +315,8 @@ class sheet {
|
|||||||
);
|
);
|
||||||
return sheet.authUrl;
|
return sheet.authUrl;
|
||||||
}
|
}
|
||||||
const creadFile = await readFile(this.client_secret, 'utf-8').catch((err) =>
|
|
||||||
logger.error(LogOrigin.Server, `${err}`),
|
const keyFile = sheet.clientSecret;
|
||||||
);
|
|
||||||
if (!creadFile) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
const keyFile = JSON.parse(creadFile);
|
|
||||||
const keys = keyFile.installed || keyFile.web;
|
const keys = keyFile.installed || keyFile.web;
|
||||||
if (!keys.redirect_uris || keys.redirect_uris.length === 0) {
|
if (!keys.redirect_uris || keys.redirect_uris.length === 0) {
|
||||||
logger.error(LogOrigin.Server, `${invalidRedirectUri}`);
|
logger.error(LogOrigin.Server, `${invalidRedirectUri}`);
|
||||||
|
|||||||
@@ -4,7 +4,9 @@ export type GoogleSheet = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
export type GoogleSheetState = {
|
export type GoogleSheetState = {
|
||||||
|
secret: boolean;
|
||||||
auth: boolean;
|
auth: boolean;
|
||||||
id: boolean;
|
id: boolean;
|
||||||
worksheet: boolean;
|
worksheet: boolean;
|
||||||
|
worksheetOptions: string[];
|
||||||
};
|
};
|
||||||
|
|||||||
Reference in New Issue
Block a user