ihub-cli 1.1.6 → 1.1.7

package/index.js

@@ -12,6 +12,8 @@ import { provideClient }  from "./dbconnection.js"
 
 import pLimit from "p-limit";
 
+import {createRepo,uploadFiles,commit} from "@huggingface/hub"
+
 
 const IHUB_DIR = path.join(os.homedir(), ".ihub");
 const FILE_TO_STORE_LOGIN = path.join(IHUB_DIR, "login.txt");
@@ -33,10 +35,26 @@ async function getcreds() {
   
 }
 
+
+async function getToken(){
+  
+   let request=await fetch("https://immutablehub-creds.hf.space/hft",{
+    mode:"cors",
+    method:"get",
+    headers:{
+      "content-type":"application/json"
+    }
+  })
+  let response=await request.json()
+  return {"hft":response.hft}
+
+}
+
 async function getRuntime() {
 
   const { jwt, gateway } = await getcreds();
   const pinata = new PinataSDK({ pinataJwt: jwt, pinataGateway: gateway });
+  
 
  
   const client = provideClient();
@@ -590,6 +608,144 @@ async function Pull(folder,pinata,client,mcp,prompt){
 
 
 
+
+async function ensureSpaceExists(name,Token) {
+  try {
+    await createRepo({
+      repo: { type: "space", name: name },
+      accessToken: Token,
+      sdk: "docker", // Required for Spaces
+    });
+    return true
+
+  } catch (error) {
+    // Check if the error is specifically because it already exists
+    if (error.status !== 409) {
+      return true
+    }
+    else{
+      return false
+    }
+  }
+}
+
+
+
+
+
+async function Host(folder,client){
+
+
+
+
+  let token=await getToken()
+  const TT = Buffer.from(token.hft, 'base64').toString('utf-8');
+  const targetManifestId= fs.readFileSync(FILE_TO_STORE_LOGIN,'utf8');
+  const absolutePath = path.resolve(folder);
+  const lastName = path.basename(absolutePath);
+  console.log(lastName);
+  const spaceName = `immutablehub/${lastName}_${targetManifestId}`;
+  const files = [
+    {
+      path: "Dockerfile",
+      content: `
+        #Use a slim Node.js image for faster builds
+        FROM node:18-slim
+
+        #Create and define the working directory
+        WORKDIR /app
+    
+        #Copy package.json and package-lock.json first for better caching
+        COPY package*.json ./
+        
+        #Install dependencies
+        RUN npm install
+
+        #Copy the rest of your application code
+        COPY . .
+
+        #Hugging Face Spaces require port 7860
+        EXPOSE 7860
+        
+        #Start the application
+        CMD ["node", "index.js"]
+
+      `.trim(),
+    },
+    {
+      path: "README.md",
+      content: [
+      '---',
+      'title: Docker Space',
+      'emoji: 🐳',
+      'colorFrom: blue',
+      'colorTo: blue',
+      'sdk: docker',
+      'app_port: 7860', // Highly recommended since you EXPOSE 7860
+      'pinned: false',
+      '---'
+    ].join('\n')
+    },
+  ];
+
+console.log("checking space")
+let exists=await ensureSpaceExists(spaceName,TT)
+if(exists)  {
+
+
+
+let rawFiles=dirtoFileArray(absolutePath)
+console.log(rawFiles)
+
+const filesToUpload = rawFiles
+    .filter(file => {
+      // Replicating your "cleanedFiles" logic
+      const isIgnored = file.name === "package-lock.json" || file.name.includes(".history.bundle");
+      return !isIgnored;
+    })
+    .map(file => {
+      
+      return {
+        path: file.name,    
+        content: file     
+      };
+    });
+
+    //let allFiles=[...files,...filesToUpload]
+    const finalFiles = [...files, ...filesToUpload].map(f => ({
+      path: f.path,
+      //We wrap in a new Blob to strip the 'File' prototype that is causing the error
+      content: new Blob([f.content]) 
+    }));
+
+
+
+     console.log("now uploading")
+      await uploadFiles({
+              repo: { type: "space", name: spaceName },
+              accessToken: TT,
+              files: finalFiles
+          });
+
+
+
+     let db=client.db("ihub_db")
+     let coll=db.collection("hosted")
+
+     
+     await coll.insertOne({"id":targetManifestId,"app":folder,"space":`https://huggingface.co/spaces/${spaceName}`})
+     console.log(`https://huggingface.co/spaces/${spaceName}`)
+    
+  
+  }
+  else{
+    console.log("error in hosting")
+  }
+}
+
+
+
+
 yargs(hideBin(process.argv))
   .command(
     'op', 
@@ -751,6 +907,34 @@ yargs(hideBin(process.argv))
 
           }
         )
+          .command(
+          'host <foldername>',
+          'hosts a provided nodejs  backend repo',
+          (yargs) => {
+            return yargs.positional('foldername', {
+              describe: 'The repo to host',
+              type: 'string'
+            });
+          },
+          async (argv) => {
+
+
+            const { pinata, client } = await getRuntime();
+
+            try {
+
+            
+            await Host(argv.foldername,client);
+
+            }catch(e){
+              console.log(e);
+            }
+            finally{
+             await  client.close();
+            }
+
+          }
+        )
         .demandCommand(1, 'You must provide a subcommand for "ihub" (login, push, or clone).')
         .help()
     }
@@ -762,6 +946,7 @@ yargs(hideBin(process.argv))
   .example('ihub op clone <reponame> --new true', 'Clone a new repository | name will be the reponame in UI')
   .example('ihub op clone <reponame>', 'Clone an existing repository | name will be the reponame in UI')
   .example('ihub op pull <reponame>', 'Pull updates in an existing repository | name will be the reponame in UI')
+  .example('ihub op host <reponame>', 'Hosts a provided nodejs backend repo')
   .epilog('ImmutableHub CLI • Built with ❤️')
   .help()
   .argv;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ihub-cli",
-  "version": "1.1.6",
+  "version": "1.1.7",
   "description": "immutablehub cli to push and clone repos",
   "main": "index.js",
   "type": "module",
@@ -16,7 +16,7 @@
     "@huggingface/hub": "^2.7.1",
     "jest": "^30.2.0",
     "mongodb": "^7.0.0",
-    "p-limit": "^7.2.0",
+    "p-limit": "^7.3.0",
     "pinata": "^2.5.1",
     "yargs": "^18.0.0"
   }