add the compactPoll logic
Audric Ackermann
4 years ago
parent
675da5cdb2
commit
ad1d5a3c4c
@ -0,0 +1,211 @@
|
||||
import { getV2OpenGroupRoomByRoomId } from '../../data/opengroups';
|
||||
import {
|
||||
OpenGroupRequestCommonType,
|
||||
OpenGroupV2CompactPollRequest,
|
||||
parseMessages,
|
||||
} from './ApiUtil';
|
||||
import { parseStatusCodeFromOnionRequest } from './OpenGroupAPIV2Parser';
|
||||
import _ from 'lodash';
|
||||
import { sendViaOnion } from '../../session/onions/onionSend';
|
||||
import { OpenGroupManagerV2 } from './OpenGroupManagerV2';
|
||||
import { OpenGroupMessageV2 } from './OpenGroupMessageV2';
|
||||
|
||||
const COMPACT_POLL_ENDPOINT = 'compact_poll';
|
||||
|
||||
export const compactFetchEverything = async (
|
||||
rooms: Array<OpenGroupRequestCommonType>
|
||||
): Promise<null | any> => {
|
||||
// fetch all we need
|
||||
const compactPollRequest = await getCompactPollRequest(rooms);
|
||||
if (!compactPollRequest) {
|
||||
window.log.info('Nothing found to be fetched. returning');
|
||||
return null;
|
||||
}
|
||||
|
||||
const result = await sendOpenGroupV2RequestCompactPoll(compactPollRequest);
|
||||
const statusCode = parseStatusCodeFromOnionRequest(result);
|
||||
if (statusCode !== 200) {
|
||||
return null;
|
||||
}
|
||||
return result;
|
||||
};
|
||||
|
||||
/**
|
||||
* This return body to be used to do the compactPoll
|
||||
*/
|
||||
const getCompactPollRequest = async (
|
||||
rooms: Array<OpenGroupRequestCommonType>
|
||||
): Promise<null | OpenGroupV2CompactPollRequest> => {
|
||||
// first verify the rooms we got are all from on the same server
|
||||
let firstUrl: string;
|
||||
if (rooms) {
|
||||
firstUrl = rooms[0].serverUrl;
|
||||
const anotherUrl = rooms.some(r => r.serverUrl !== firstUrl);
|
||||
if (anotherUrl) {
|
||||
throw new Error('CompactPoll is designed for a single server');
|
||||
}
|
||||
} else {
|
||||
window.log.warn('CompactPoll: No room given. nothing to do');
|
||||
return null;
|
||||
}
|
||||
|
||||
const allServerPubKeys: Array<string> = [];
|
||||
|
||||
const roomsRequestInfos = _.compact(
|
||||
await Promise.all(
|
||||
rooms.map(async ({ roomId, serverUrl }) => {
|
||||
try {
|
||||
const fetchedInfo = await getV2OpenGroupRoomByRoomId({
|
||||
serverUrl,
|
||||
roomId,
|
||||
});
|
||||
if (!fetchedInfo) {
|
||||
window.log.warn('Could not find this room getMessages');
|
||||
return null;
|
||||
}
|
||||
|
||||
const {
|
||||
lastMessageFetchedServerID,
|
||||
lastMessageDeletedServerID,
|
||||
token,
|
||||
serverPublicKey,
|
||||
} = fetchedInfo;
|
||||
allServerPubKeys.push(serverPublicKey);
|
||||
const roomRequestContent: Record<string, any> = {
|
||||
room_id: roomId,
|
||||
auth_token: token || '',
|
||||
};
|
||||
if (lastMessageDeletedServerID) {
|
||||
roomRequestContent.from_deletion_server_id = lastMessageDeletedServerID;
|
||||
}
|
||||
|
||||
if (lastMessageFetchedServerID) {
|
||||
roomRequestContent.from_message_server_id = lastMessageFetchedServerID;
|
||||
}
|
||||
|
||||
return roomRequestContent;
|
||||
} catch (e) {
|
||||
window.log.warn('failed to fetch roominfos for room', roomId);
|
||||
return null;
|
||||
}
|
||||
})
|
||||
)
|
||||
);
|
||||
if (!roomsRequestInfos?.length) {
|
||||
return null;
|
||||
}
|
||||
// double check that all those server pubkeys are the same
|
||||
let firstPubkey: string;
|
||||
if (allServerPubKeys?.length) {
|
||||
firstPubkey = allServerPubKeys[0];
|
||||
const allMatch = allServerPubKeys.every(p => p === firstPubkey);
|
||||
if (!allMatch) {
|
||||
window.log.warn('All pubkeys do not match:', allServerPubKeys);
|
||||
return null;
|
||||
}
|
||||
} else {
|
||||
window.log.warn('No pubkeys found:', allServerPubKeys);
|
||||
return null;
|
||||
}
|
||||
const body = JSON.stringify({
|
||||
requests: roomsRequestInfos,
|
||||
});
|
||||
return {
|
||||
body,
|
||||
server: firstUrl,
|
||||
serverPubKey: firstPubkey,
|
||||
endpoint: COMPACT_POLL_ENDPOINT,
|
||||
};
|
||||
};
|
||||
|
||||
/**
|
||||
* This call is separate as a lot of the logic is custom (statusCode handled separately, etc)
|
||||
*/
|
||||
async function sendOpenGroupV2RequestCompactPoll(
|
||||
request: OpenGroupV2CompactPollRequest
|
||||
): Promise<Object | null> {
|
||||
const { server, endpoint, body, serverPubKey } = request;
|
||||
// this will throw if the url is not valid
|
||||
const builtUrl = new URL(`${server}/${endpoint}`);
|
||||
|
||||
console.warn(`sending compactPoll request: ${request.body}`);
|
||||
|
||||
const res = await sendViaOnion(serverPubKey, builtUrl, {
|
||||
method: 'POST',
|
||||
body,
|
||||
});
|
||||
|
||||
const statusCode = parseStatusCodeFromOnionRequest(res);
|
||||
if (!statusCode) {
|
||||
window.log.warn(
|
||||
'sendOpenGroupV2Request Got unknown status code; res:',
|
||||
res
|
||||
);
|
||||
return res as object;
|
||||
}
|
||||
|
||||
const results = await parseCompactPollResults(res);
|
||||
|
||||
throw new Error(
|
||||
'See how we handle needs of new tokens, and save stuff to db (last deleted, ... conversation commit, etc'
|
||||
);
|
||||
|
||||
return res as object;
|
||||
}
|
||||
|
||||
type ParsedRoomCompactPollResults = {
|
||||
roomId: string;
|
||||
deletions: Array<number>;
|
||||
messages: Array<OpenGroupMessageV2>;
|
||||
moderators: Array<string>;
|
||||
};
|
||||
|
||||
const parseCompactPollResult = async (
|
||||
singleRoomResult: any
|
||||
): Promise<ParsedRoomCompactPollResults | null> => {
|
||||
const {
|
||||
room_id,
|
||||
deletions: rawDeletions,
|
||||
messages: rawMessages,
|
||||
moderators: rawMods,
|
||||
} = singleRoomResult;
|
||||
|
||||
if (
|
||||
!room_id ||
|
||||
rawDeletions === undefined ||
|
||||
rawMessages === undefined ||
|
||||
rawMods === undefined
|
||||
) {
|
||||
window.log.warn('Invalid compactPoll result', singleRoomResult);
|
||||
return null;
|
||||
}
|
||||
|
||||
const validMessages = await parseMessages(rawMessages);
|
||||
const moderators = rawMods as Array<string>;
|
||||
const deletions = rawDeletions as Array<number>;
|
||||
|
||||
return { roomId: room_id, deletions, messages: validMessages, moderators };
|
||||
};
|
||||
|
||||
const parseCompactPollResults = async (
|
||||
res: any
|
||||
): Promise<Array<ParsedRoomCompactPollResults> | null> => {
|
||||
if (
|
||||
!res ||
|
||||
!res.result ||
|
||||
!res.result.results ||
|
||||
!res.result.results.length
|
||||
) {
|
||||
return null;
|
||||
}
|
||||
const arrayOfResults = res.result.results as Array<any>;
|
||||
|
||||
const parsedResults: Array<ParsedRoomCompactPollResults> = _.compact(
|
||||
await Promise.all(arrayOfResults.map(parseCompactPollResult))
|
||||
);
|
||||
|
||||
if (!parsedResults || !parsedResults.length) {
|
||||
return null;
|
||||
}
|
||||
return parsedResults;
|
||||
};
|
||||
Loading…
Reference in New Issue